ston1th/cddns
1
0
Fork 0
Code Issues Releases Activity

some improvements

Browse source
This commit is contained in:
ston1th 2017-05-20 11:31:51 +02:00
commit 4e2f824d99
1 changed files with 39 additions and 68 deletions
Download patch file Download diff file Expand all files Collapse all files

97
ddns.c
View file

@ -6,45 +6,30 @@
#include <sys/file.h>
#include <err.h>
#define SEP '='
#define MAX_QUERY_LEN 255
#define MAX_QUERY_ARGS 2
#define IP_LEN 16
#define IP_FILE "ip.txt"
#define TOKEN "<empty>"
void free_arr(void *arr) {
uintptr_t *a = arr;
while(*a) {
free((void*)*a++);
}
free(arr);
}
char **parse(const char *str) {
const char *s = str;
char *qs, *token;
char **arr = NULL;
int c = 1, argc = 0;
qs = strdup(s);
while((token = strsep(&qs, "&")) != NULL) {
if(++argc > MAX_QUERY_ARGS)
break;
c++;
if(arr == NULL) {
arr = malloc(c*sizeof(char *));
if(arr == NULL)
break;
} else {
arr = realloc(arr, c*sizeof(char *));
if(arr == NULL)
char *form_value(const char *args, const char *field) {
char *s;
char *token;
char *value;
char *ret = NULL;
size_t len;
len = strlen(field);
s = strdup(args);
while((token = strsep(&s, "&")) != NULL) {
if(strlen(token) < len)
continue;
if(strncmp(field, token, len) == 0) {
value = strchr(token, '=');
ret = strdup(value);
break;
}
arr[c-1] = NULL;
arr[c-2] = strdup(token);
}
free(qs);
return(arr);
free(s);
return(ret);
}
void header() {
@ -73,46 +58,32 @@ int ctc(const char *a, const char *b) {
return(ctb(v, 0));
}
int set_ip(char *env, int fd) {
char *ip = NULL, *token = NULL, *tmp = NULL;
int ret = -1, len = 0;
char **arg = parse(env);
char **arr = arg;
if(arg == NULL)
return(0);
for(; *arr; arr++) {
if(strncmp(*arr, "ip", 2) == 0) {
tmp = strchr(*arr, SEP);
if(++tmp == NULL)
continue;
ip = strdup(tmp);
} else if(strncmp(*arr, "token", 5) == 0) {
tmp = strchr(*arr, SEP);
if(++tmp == NULL)
continue;
token = strdup(tmp);
}
}
tmp = NULL;
free_arr(arg);
int set_ip(char *env, int wfd) {
char *ip = NULL, *token = NULL;
int ret = 1;
size_t len;
ip = form_value(env, "ip");
if(!ip)
return(ret);
token = form_value(env, "token");
if(!token)
free(ip);
return(ret);
if(ctc(token, TOKEN) == 1) {
if(ip) {
len = strlen(ip);
if(len >= IP_LEN-1)
len = IP_LEN-1;
write(fd, ip, len);
}
close(fd);
write(wfd, ip, len);
close(wfd);
ret = 0;
} else
ret = 1;
if(ip)
ret = 2;
free(ip);
if(token)
free(token);
return(ret);
}
char *err_msg[2] = {"wrong input", "wrong token"};
char *err_msg[2] = {"invalid arguments", "wrong token"};
int main(void) {
char *env = NULL;
@ -136,8 +107,8 @@ int main(void) {
puts("query longer than 255 bytes");
return(EXIT_SUCCESS);
}
if((errn = set_ip(env, wfd)) >= 0)
puts(err_msg[errn]);
if((errn = set_ip(env, wfd)) > 0)
puts(err_msg[errn-1]);
else
puts("ok");
} else {