mirror of
https://github.com/lxsang/ant-http
synced 2024-12-26 00:38:21 +01:00
using shared library for common api
This commit is contained in:
parent
8b7ce33944
commit
9943dbdcc8
1
.gitignore
vendored
1
.gitignore
vendored
@ -51,3 +51,4 @@ modules.order
|
|||||||
Module.symvers
|
Module.symvers
|
||||||
Mkfile.old
|
Mkfile.old
|
||||||
dkms.con
|
dkms.con
|
||||||
|
.DS_Store
|
||||||
|
73
Makefile
73
Makefile
@ -1,13 +1,7 @@
|
|||||||
|
USE_DB=TRUE
|
||||||
CC=gcc
|
CC=gcc
|
||||||
EXT=dylib
|
EXT=dylib
|
||||||
SERVER=plugin_manager.o \
|
|
||||||
plugins/ini.o \
|
|
||||||
http_server.o \
|
|
||||||
plugins/dictionary.o \
|
|
||||||
plugins/base64.o \
|
|
||||||
plugins/sha1.o \
|
|
||||||
plugins/utils.o
|
|
||||||
SERVERLIB=-lpthread -ldl
|
|
||||||
UNAME_S := $(shell uname -s)
|
UNAME_S := $(shell uname -s)
|
||||||
ifeq ($(UNAME_S),Linux)
|
ifeq ($(UNAME_S),Linux)
|
||||||
BUILDIRD=/opt/www
|
BUILDIRD=/opt/www
|
||||||
@ -17,46 +11,75 @@ ifeq ($(UNAME_S),Darwin)
|
|||||||
BUILDIRD=../ant-build
|
BUILDIRD=../ant-build
|
||||||
PF_FLAG= -DMACOS
|
PF_FLAG= -DMACOS
|
||||||
endif
|
endif
|
||||||
CFLAGS=-W -Wall -g -std=c99 -D DEBUG -D USE_DB $(PF_FLAG)
|
|
||||||
|
ifeq ($(USE_DB),TRUE)
|
||||||
|
DB_OBJ=plugins/dbhelper.o
|
||||||
|
DB_LIB=-lsqlite3
|
||||||
|
DB_FLAG=-D USE_DB
|
||||||
|
endif
|
||||||
|
|
||||||
|
ifeq ($(USE_DB),FALSE)
|
||||||
|
DB_OBJ=
|
||||||
|
DB_LIB=
|
||||||
|
DB_FLAG=
|
||||||
|
endif
|
||||||
|
|
||||||
|
|
||||||
|
CFLAGS= -W -Wall -g -std=c99 -D DEBUG $(DB_FLAG) $(PF_FLAG)
|
||||||
|
|
||||||
|
LIB_PATH=$(BUILDIRD)/plugins
|
||||||
|
LIB_NAME=libantd
|
||||||
|
LIB_FLAG= $(LIB_PATH$)/$(LIB_NAME).$(EXT)
|
||||||
|
SERVERLIB=-lpthread -ldl $(LIB_FLAG)
|
||||||
|
|
||||||
|
SERVER_O=plugin_manager.o \
|
||||||
|
http_server.o \
|
||||||
|
httpd.o
|
||||||
#-lsocket
|
#-lsocket
|
||||||
PLUGINS= dummy.$(EXT) fileman.$(EXT) pluginsman.$(EXT) wterm.$(EXT) nodedaemon.$(EXT) cookiex.$(EXT) wsimg.$(EXT)
|
PLUGINS= dummy.$(EXT) fileman.$(EXT) pluginsman.$(EXT) wterm.$(EXT) nodedaemon.$(EXT) cookiex.$(EXT) wsimg.$(EXT)
|
||||||
|
|
||||||
PLUGINSDEP = plugins/ini.o \
|
LIBOBJS = plugins/ini.o \
|
||||||
plugins/plugin.o \
|
plugins/handle.o \
|
||||||
plugins/dbhelper.o \
|
$(DB_OBJ) \
|
||||||
plugins/dictionary.o \
|
plugins/dictionary.o \
|
||||||
plugins/base64.o \
|
plugins/base64.o \
|
||||||
plugins/utils.o \
|
plugins/utils.o \
|
||||||
plugins/ws.o \
|
plugins/ws.o \
|
||||||
plugins/sha1.o \
|
plugins/sha1.o \
|
||||||
plugins/list.o
|
plugins/list.o
|
||||||
PLUGINLIBS = -lsqlite3
|
|
||||||
|
PLUGINSDEP = plugins/plugin.o
|
||||||
|
|
||||||
|
|
||||||
main: httpd plugins
|
main: httpd plugins
|
||||||
|
|
||||||
|
|
||||||
httpd:$(SERVER)
|
httpd: lib $(SERVER_O)
|
||||||
$(CC) $(CFLAGS) $(SERVERLIB) $(SERVER) -o $(BUILDIRD)/httpd httpd.c
|
$(CC) $(CFLAGS) $(SERVER_O) $(SERVERLIB) -o $(BUILDIRD)/httpd
|
||||||
cp antd $(BUILDIRD)
|
cp antd $(BUILDIRD)
|
||||||
|
|
||||||
|
lib: $(LIBOBJS)
|
||||||
|
$(CC) $(CFLAGS) $(DB_LIB) -shared -o $(LIB_PATH$)/$(LIB_NAME).$(EXT) $(LIBOBJS)
|
||||||
%.o: %.c
|
%.o: %.c
|
||||||
$(CC) -fPIC $(CFLAGS) -c $< -o $@
|
$(CC) -fPIC $(CFLAGS) -c $< -o $@
|
||||||
|
|
||||||
plugins: $(PLUGINS)
|
plugins: $(PLUGINS)
|
||||||
|
|
||||||
%.$(EXT): $(PLUGINSDEP)
|
%.$(EXT): $(PLUGINSDEP)
|
||||||
for file in $(wildcard plugins/$(basename $@)/*.c) ; do\
|
for file in $(wildcard plugins/$(basename $@)/*.c) ; do\
|
||||||
$(CC) -fPIC $(CFLAGS) -c $$file -o $$file.o; \
|
$(CC) -fPIC $(CFLAGS) -c $$file -o $$file.o; \
|
||||||
done
|
done
|
||||||
$(CC) $(CFLAGS) $(PLUGINLIBS) -shared -o $(BUILDIRD)/plugins/$(basename $@).$(EXT) \
|
$(CC) $(CFLAGS) $(PLUGINLIBS) $(LIB_FLAG) -shared -o $(BUILDIRD)/plugins/$(basename $@).$(EXT) \
|
||||||
$(PLUGINSDEP) plugins/$(basename $@)/*.c.o
|
$(PLUGINSDEP) plugins/$(basename $@)/*.c.o
|
||||||
|
|
||||||
|
|
||||||
clean: sclean pclean
|
clean: sclean pclean
|
||||||
|
|
||||||
sclean:
|
sclean:
|
||||||
rm -f *.o $(BUILDIRD)/httpd
|
-rm -f *.o $(BUILDIRD)/httpd
|
||||||
|
-rm *.$(EXT)
|
||||||
pclean:
|
pclean:
|
||||||
rm -rf $(BUILDIRD)/plugins/* plugins/*.o
|
-rm -rf $(BUILDIRD)/plugins/* plugins/*.o
|
||||||
-for file in plugins/* ;do \
|
-for file in plugins/* ;do \
|
||||||
if [ -d "$$file" ]; then \
|
if [ -d "$$file" ]; then \
|
||||||
rm "$$file"/*.o; \
|
rm "$$file"/*.o; \
|
||||||
|
9
antd
9
antd
@ -1 +1,8 @@
|
|||||||
LD_LIBRARY_PATH=$(dirname "$0")/plugins $(dirname "$0")/httpd $(dirname "$0")/config.ini
|
#!/bin/sh
|
||||||
|
UNAME=`uname -s`
|
||||||
|
|
||||||
|
if [ "$UNAME" == "Darwin" ]; then
|
||||||
|
DYLD_LIBRARY_PATH=$(dirname "$0")/plugins/ $(dirname "$0")/httpd $(dirname "$0")/config.ini
|
||||||
|
else
|
||||||
|
LD_LIBRARY_PATH=$(dirname "$0")/plugins/ $(dirname "$0")/httpd $(dirname "$0")/config.ini
|
||||||
|
fi
|
||||||
|
505
http_server.c
505
http_server.c
@ -381,3 +381,508 @@ void unimplemented(int client)
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Decode the HTTP POST request with URL encode
|
||||||
|
* @param client socket client
|
||||||
|
* @param len content length
|
||||||
|
* @return query string
|
||||||
|
*/
|
||||||
|
char* post_url_decode(int client,int len)
|
||||||
|
{
|
||||||
|
char *query = (char*) malloc((len+1)*sizeof(char));
|
||||||
|
for (int i = 0; i < len; i++) {
|
||||||
|
recv(client, (query+i), 1, 0);
|
||||||
|
}
|
||||||
|
query[len]='\0';
|
||||||
|
//query = url_decode(query);
|
||||||
|
LOG("POST Query %s\n", query);
|
||||||
|
return query;
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* Decode the HTTP request
|
||||||
|
* Get the cookie values
|
||||||
|
* if it is the GET request, decode the query string into a dictionary
|
||||||
|
* if it is a POST, check the content type of the request
|
||||||
|
* - if it is a POST request with URL encoded : decode the url encode
|
||||||
|
* - if it is a POST request with multipart form data: de code the multipart
|
||||||
|
* - if other - UNIMPLEMENTED
|
||||||
|
* @param client socket client
|
||||||
|
* @param method HTTP method
|
||||||
|
* @param query query string in case of GET
|
||||||
|
* @return a dictionary of key- value
|
||||||
|
*/
|
||||||
|
dictionary decode_request(int client,const char* method,const char* query)
|
||||||
|
{
|
||||||
|
dictionary request = NULL;
|
||||||
|
dictionary cookie = NULL;
|
||||||
|
dictionary xheader = dict();
|
||||||
|
char* line;
|
||||||
|
char * token;
|
||||||
|
if(strcmp(method,"GET") == 0)
|
||||||
|
{
|
||||||
|
// this for check if web socket is enabled
|
||||||
|
int ws= 0;
|
||||||
|
char* ws_key = NULL;
|
||||||
|
while((line = read_line(client)) && strcmp("\r\n",line))
|
||||||
|
{
|
||||||
|
trim(line, '\n');
|
||||||
|
trim(line, '\r');
|
||||||
|
token = strsep(&line,":");
|
||||||
|
trim(token,' ');
|
||||||
|
trim(line,' ');
|
||||||
|
dput(xheader,token,line);
|
||||||
|
if(token != NULL &&strcasecmp(token,"Cookie") == 0)
|
||||||
|
{
|
||||||
|
if(!cookie) cookie = decode_cookie(line);
|
||||||
|
}
|
||||||
|
else if(token != NULL && strcasecmp(token,"Upgrade") == 0)
|
||||||
|
{
|
||||||
|
// verify that the connection is upgrade to websocket
|
||||||
|
trim(line, ' ');
|
||||||
|
if(line != NULL && strcasecmp(line,"websocket") == 0)
|
||||||
|
ws = 1;
|
||||||
|
} else if(token != NULL && strcasecmp(token,"Sec-WebSocket-Key") == 0)
|
||||||
|
{
|
||||||
|
// get the key from the client
|
||||||
|
trim(line, ' ');
|
||||||
|
ws_key = strdup(line);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
request = decode_url_request(query);
|
||||||
|
if(ws && ws_key != NULL)
|
||||||
|
{
|
||||||
|
ws_confirm_request(client, ws_key);
|
||||||
|
free(ws_key);
|
||||||
|
// insert wsocket flag to request
|
||||||
|
// plugin should handle this ugraded connection
|
||||||
|
// not the server
|
||||||
|
if(!request) request = dict();
|
||||||
|
dput(request,"__web_socket__","1");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
char* ctype = NULL;
|
||||||
|
int clen = -1;
|
||||||
|
line = read_line(client);
|
||||||
|
while (line && strcmp("\r\n",line))
|
||||||
|
{
|
||||||
|
//printf("%s\n",line);
|
||||||
|
trim(line, '\n');
|
||||||
|
trim(line, '\r');
|
||||||
|
token = strsep(&line,":");
|
||||||
|
trim(token,' ');
|
||||||
|
trim(line, ' ');
|
||||||
|
dput(xheader,token,line);
|
||||||
|
if(token != NULL &&strcasecmp(token,"Content-Type") == 0)
|
||||||
|
{
|
||||||
|
ctype = strsep(&line,":");
|
||||||
|
trim(ctype,' ');
|
||||||
|
} else if(token != NULL &&strcasecmp(token,"Content-Length") == 0)
|
||||||
|
{
|
||||||
|
token = strsep(&line,":");
|
||||||
|
trim(token,' ');
|
||||||
|
clen = atoi(token);
|
||||||
|
}
|
||||||
|
else if(token != NULL &&strcasecmp(token,"Cookie") == 0)
|
||||||
|
{
|
||||||
|
if(!cookie) cookie = decode_cookie(line);
|
||||||
|
}
|
||||||
|
|
||||||
|
line = read_line(client);
|
||||||
|
}
|
||||||
|
free(line);
|
||||||
|
if(ctype == NULL || clen == -1)
|
||||||
|
{
|
||||||
|
LOG("Bad request\n");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
// decide what to do with the data
|
||||||
|
if(strstr(ctype,FORM_URL_ENCODE) > 0)
|
||||||
|
{
|
||||||
|
request = decode_url_request(post_url_decode(client,clen));
|
||||||
|
} else if(strstr(ctype,FORM_MULTI_PART)> 0)
|
||||||
|
{
|
||||||
|
//printf("Multi part form : %s\n", ctype);
|
||||||
|
request = decode_multi_part_request(client,ctype);
|
||||||
|
}
|
||||||
|
else if(strstr(ctype,APP_JSON) > 0)
|
||||||
|
{
|
||||||
|
char* query = json_data_decode(client,clen);
|
||||||
|
request = dict();
|
||||||
|
dput(request,"json", strdup(query));
|
||||||
|
free(query);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
LOG("Un supported yet %s\n",ctype);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//if(cookie->key == NULL) {free(cookie);cookie= NULL;}
|
||||||
|
if(!request)
|
||||||
|
request = dict();
|
||||||
|
|
||||||
|
dput(request,"cookie",cookie);
|
||||||
|
dput(request,"__xheader__",xheader);
|
||||||
|
return request;
|
||||||
|
}
|
||||||
|
void __px(const char* data,int size)
|
||||||
|
{
|
||||||
|
for (int i = 0; i < size; ++i)
|
||||||
|
printf(" %02x", data[i]);
|
||||||
|
|
||||||
|
printf("\n");
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* Send header to the client to confirm
|
||||||
|
* that the websocket is accepted by
|
||||||
|
* our server
|
||||||
|
*/
|
||||||
|
void ws_confirm_request(int client, const char* key)
|
||||||
|
{
|
||||||
|
char buf[256];
|
||||||
|
char rkey[128];
|
||||||
|
char sha_d[20];
|
||||||
|
char base64[64];
|
||||||
|
strcpy(rkey,key);
|
||||||
|
strcat(rkey,WS_MAGIC_STRING);
|
||||||
|
//printf("RESPONDKEY '%s'\n", rkey);
|
||||||
|
SHA1_CTX context;
|
||||||
|
SHA1_Init(&context);
|
||||||
|
SHA1_Update(&context, rkey, strlen(rkey));
|
||||||
|
SHA1_Final(&context, sha_d);
|
||||||
|
Base64encode(base64, sha_d, 20);
|
||||||
|
//printf("Base 64 '%s'\n", base64);
|
||||||
|
// send accept to client
|
||||||
|
sprintf(buf, "HTTP/1.1 101 Switching Protocols\r\n");
|
||||||
|
send(client, buf, strlen(buf), 0);
|
||||||
|
sprintf(buf, "Upgrade: websocket\r\n");
|
||||||
|
send(client, buf, strlen(buf), 0);
|
||||||
|
sprintf(buf, "Connection: Upgrade\r\n");
|
||||||
|
send(client, buf, strlen(buf), 0);
|
||||||
|
sprintf(buf, "Sec-WebSocket-Accept: %s\r\n",base64);
|
||||||
|
send(client, buf, strlen(buf), 0);
|
||||||
|
sprintf(buf, "\r\n");
|
||||||
|
send(client, buf, strlen(buf), 0);
|
||||||
|
|
||||||
|
LOG("%s\n", "Websocket is now enabled for plugin");
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* Decode the cookie header to a dictionary
|
||||||
|
* @param client The client socket
|
||||||
|
* @return The Dictionary socket or NULL
|
||||||
|
*/
|
||||||
|
dictionary decode_cookie(const char* line)
|
||||||
|
{
|
||||||
|
char *token,*token1;
|
||||||
|
char *cpstr = strdup(line);
|
||||||
|
trim(cpstr,' ');
|
||||||
|
trim(cpstr,'\n');
|
||||||
|
trim(cpstr,'\r');
|
||||||
|
//printf("FUCKIT %s\n",cpstr );
|
||||||
|
dictionary dic = NULL;
|
||||||
|
while((token = strsep(&cpstr,";")))
|
||||||
|
{
|
||||||
|
trim(token,' ');
|
||||||
|
token1 = strsep(&token,"=");
|
||||||
|
if(token1)
|
||||||
|
{
|
||||||
|
if(dic == NULL)
|
||||||
|
dic = dict();
|
||||||
|
//LOG("Found cookie : %s = %s\n",token1,token);
|
||||||
|
dput(dic,token1,strdup(token));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//}
|
||||||
|
return dic;
|
||||||
|
//free(cpstr);
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* Decode the multi-part form data from the POST request
|
||||||
|
* If it is a file upload, copy the file to tmp dir
|
||||||
|
* and generate the metadata for the server-side
|
||||||
|
* @param client the socket client
|
||||||
|
* @param ctype Content-Type of the request
|
||||||
|
* @param clen Content length, but not used here
|
||||||
|
* @return a dictionary of key - value
|
||||||
|
*/
|
||||||
|
dictionary decode_multi_part_request(int client,const char* ctype)
|
||||||
|
{
|
||||||
|
char * boundary;
|
||||||
|
char * boundend;
|
||||||
|
char * line;
|
||||||
|
char * str_copy = strdup(ctype);
|
||||||
|
char* token;
|
||||||
|
char* keytoken ;
|
||||||
|
char* valtoken ;
|
||||||
|
char* part_name;
|
||||||
|
char* part_file;
|
||||||
|
char* file_path;
|
||||||
|
char buf[BUFFLEN];
|
||||||
|
char* field;
|
||||||
|
dictionary dic = NULL;
|
||||||
|
FILE *fp = NULL;
|
||||||
|
boundary = strsep(&str_copy,"="); //discard first part
|
||||||
|
boundary = strsep(&str_copy,"=");
|
||||||
|
if(boundary && strlen(boundary)>0)
|
||||||
|
{
|
||||||
|
dic = dict();
|
||||||
|
trim(boundary,' ');
|
||||||
|
boundend = __s("%s--",boundary);
|
||||||
|
//find first boundary
|
||||||
|
while((line = read_line(client))&&strstr(line,boundary) <= 0);
|
||||||
|
// loop through each part separated by the boundary
|
||||||
|
while(line && strstr(line,boundary) > 0){
|
||||||
|
// search for content disposition:
|
||||||
|
while((line = read_line(client)) &&
|
||||||
|
strstr(line,"Content-Disposition:") <= 0);
|
||||||
|
if(strstr(line,"Content-Disposition:") <= 0) return NULL;
|
||||||
|
// extract parameters from header
|
||||||
|
part_name = NULL;
|
||||||
|
part_file = NULL;
|
||||||
|
while((token = strsep(&line,";")))
|
||||||
|
{
|
||||||
|
keytoken = strsep(&token,"=");
|
||||||
|
if(keytoken && strlen(keytoken)>0)
|
||||||
|
{
|
||||||
|
trim(keytoken,' ');
|
||||||
|
valtoken = strsep(&token,"=");
|
||||||
|
if(valtoken)
|
||||||
|
{
|
||||||
|
trim(valtoken,' ');
|
||||||
|
trim(valtoken,'\n');
|
||||||
|
trim(valtoken,'\r');
|
||||||
|
trim(valtoken,'\"');
|
||||||
|
if(strcmp(keytoken,"name") == 0)
|
||||||
|
{
|
||||||
|
part_name = valtoken;
|
||||||
|
} else if(strcmp(keytoken,"filename") == 0)
|
||||||
|
{
|
||||||
|
part_file = valtoken;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// get the binary data
|
||||||
|
if(part_name != NULL)
|
||||||
|
{
|
||||||
|
// go to the beginer of data bock
|
||||||
|
while((line = read_line(client)) && strcmp(line,"\r\n") != 0);
|
||||||
|
if(part_file == NULL)
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* This allow only 1024 bytes of data (max),
|
||||||
|
* out of this range, the data is cut out.
|
||||||
|
* Need an efficient way to handle this
|
||||||
|
*/
|
||||||
|
line = read_line(client);
|
||||||
|
trim(line,'\n');
|
||||||
|
trim(line,'\r');
|
||||||
|
trim(line,' ');
|
||||||
|
dput(dic,part_name,line);
|
||||||
|
// find the next boundary
|
||||||
|
while((line = read_line(client)) && strstr(line,boundary) <= 0);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
file_path = __s("%s%s.%u",server_config.tmpdir,part_file,(unsigned)time(NULL));
|
||||||
|
fp=fopen(file_path, "wb");
|
||||||
|
if(fp)
|
||||||
|
{
|
||||||
|
int totalsize=0,len=0;
|
||||||
|
//read until the next boundary
|
||||||
|
while((len = read_buf(client,buf,sizeof(buf))) > 0 && strstr(buf,boundary) <= 0)
|
||||||
|
{
|
||||||
|
fwrite(buf, len, 1, fp);
|
||||||
|
totalsize += len;
|
||||||
|
}
|
||||||
|
//remove \r\n at the end
|
||||||
|
fseek(fp,-2, SEEK_CUR);
|
||||||
|
totalsize -= 2;
|
||||||
|
fclose(fp);
|
||||||
|
line = buf;
|
||||||
|
|
||||||
|
field = __s("%s.file",part_name);
|
||||||
|
dput(dic,field,part_file);
|
||||||
|
field = __s("%s.tmp",part_name);
|
||||||
|
dput(dic,field,strdup(file_path));
|
||||||
|
field = __s("%s.size",part_name);
|
||||||
|
dput(dic,field,__s("%d",totalsize));
|
||||||
|
field = __s("%s.ext",part_name);
|
||||||
|
dput(dic,field,ext(part_file));
|
||||||
|
free(field);
|
||||||
|
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
LOG("Cannot wirte file to :%s\n", file_path );
|
||||||
|
}
|
||||||
|
free(file_path);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//printf("[Lines]:%s\n",line);
|
||||||
|
// check if end of request
|
||||||
|
if(line&&strstr(line,boundend)>0)
|
||||||
|
{
|
||||||
|
LOG("End request %s\n", boundend);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
free(str_copy);
|
||||||
|
return dic;
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* Decode a query string (GET request or POST URL encoded) to
|
||||||
|
* a dictionary of key-value
|
||||||
|
* @param query : the query string
|
||||||
|
* @return a dictionary of key-value
|
||||||
|
*/
|
||||||
|
dictionary decode_url_request(const char* query)
|
||||||
|
{
|
||||||
|
char* str_copy = strdup(query);
|
||||||
|
//str_copy = ;
|
||||||
|
char* token;
|
||||||
|
if(query == NULL) return NULL;
|
||||||
|
if(strlen(query) == 0) return NULL;
|
||||||
|
dictionary dic = dict();
|
||||||
|
while ((token = strsep(&str_copy, "&")))
|
||||||
|
{
|
||||||
|
char* key;
|
||||||
|
char* val;
|
||||||
|
if(strlen(token)>0)
|
||||||
|
{
|
||||||
|
key = strsep(&token,"=");
|
||||||
|
if(key && strlen(key)>0)
|
||||||
|
{
|
||||||
|
val = strsep(&token,"=");
|
||||||
|
dput(dic,key,url_decode(val));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
free(str_copy);
|
||||||
|
return dic;
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* Decode JSON query string to string
|
||||||
|
*/
|
||||||
|
char* json_data_decode(int client,int len)
|
||||||
|
{
|
||||||
|
char *query = (char*) malloc((len+1)*sizeof(char));
|
||||||
|
for (int i = 0; i < len; i++) {
|
||||||
|
recv(client, (query+i), 1, 0);
|
||||||
|
}
|
||||||
|
query[len]='\0';
|
||||||
|
//query = url_decode(query);
|
||||||
|
LOG("JSON Query %s\n", query);
|
||||||
|
return query;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* read the request as a string line format
|
||||||
|
* @param sock socket
|
||||||
|
* @return a request string
|
||||||
|
*/
|
||||||
|
char* read_line(int sock)
|
||||||
|
{
|
||||||
|
char buf[BUFFLEN];
|
||||||
|
read_buf(sock,buf,sizeof(buf));
|
||||||
|
return strdup(buf);
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* Read the socket request in to a buffer or size
|
||||||
|
* The data is read until the buffer is full or
|
||||||
|
* there are a carrier return character
|
||||||
|
* @param sock socket
|
||||||
|
* @param buf buffer
|
||||||
|
* @param size size of buffer
|
||||||
|
* @return number of bytes read
|
||||||
|
*/
|
||||||
|
int read_buf(int sock, char*buf,int size)
|
||||||
|
{
|
||||||
|
int i = 0;
|
||||||
|
char c = '\0';
|
||||||
|
int n;
|
||||||
|
while ((i < size - 1) && (c != '\n'))
|
||||||
|
{
|
||||||
|
n = recv(sock, &c, 1, 0);
|
||||||
|
if (n > 0)
|
||||||
|
{
|
||||||
|
buf[i] = c;
|
||||||
|
i++;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
c = '\n';
|
||||||
|
}
|
||||||
|
buf[i] = '\0';
|
||||||
|
return i;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Execute a plugin based on the http requeset
|
||||||
|
* First decode the http request header to find the correct plugin
|
||||||
|
* and the correct function on the plugin
|
||||||
|
* Second, decode all parameters necessary of the request and pass it
|
||||||
|
* to the callback function.
|
||||||
|
* Execute the callback function if sucess
|
||||||
|
* @param client soket client
|
||||||
|
* @param path request path
|
||||||
|
* @param method request method
|
||||||
|
* @param query_string GET query string
|
||||||
|
* @return -1 if failure
|
||||||
|
* 1 if sucess
|
||||||
|
*/
|
||||||
|
int execute_plugin(int client, const char *path, const char *method, const char *query_string)
|
||||||
|
{
|
||||||
|
char pname[255];
|
||||||
|
char pfunc[255];
|
||||||
|
void (*fn)(int, const char*,const char*, dictionary);
|
||||||
|
struct plugin_entry *plugin ;
|
||||||
|
int plen = strlen(path);
|
||||||
|
char * rpath = (char*) malloc((plen+1)*sizeof(char));
|
||||||
|
char *error;
|
||||||
|
memcpy(rpath,path+1,plen);
|
||||||
|
rpath[plen] = '\0';
|
||||||
|
trim(rpath,'/');
|
||||||
|
char * delim = strchr(rpath,'/');
|
||||||
|
if(delim == NULL)
|
||||||
|
{
|
||||||
|
strcpy(pname,rpath);
|
||||||
|
strcpy(pfunc,"default");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
int npos,fpos;
|
||||||
|
npos = delim - rpath;
|
||||||
|
fpos = strlen(rpath) - npos ;
|
||||||
|
memcpy(pname,rpath,npos);
|
||||||
|
pname[npos] = '\0';
|
||||||
|
memcpy(pfunc,rpath+npos+1,fpos);
|
||||||
|
pfunc[fpos-1]='\0';
|
||||||
|
}
|
||||||
|
LOG("Client %d\n",client );
|
||||||
|
LOG("Path : '%s'\n", rpath);
|
||||||
|
LOG("Method:%s\n", method);
|
||||||
|
LOG("Plugin name '%s'\n",pname);
|
||||||
|
LOG("Query path. '%s'\n", pfunc);
|
||||||
|
LOG("query :%s\n", query_string);
|
||||||
|
|
||||||
|
//load the plugin
|
||||||
|
if((plugin = plugin_lookup(pname)) == NULL)
|
||||||
|
if((plugin= plugin_load(pname)) == NULL)
|
||||||
|
return -1;
|
||||||
|
// load the function
|
||||||
|
fn = (void (*)(int, const char *, const char*, dictionary))dlsym(plugin->handle, PLUGIN_HANDLER);
|
||||||
|
if ((error = dlerror()) != NULL)
|
||||||
|
{
|
||||||
|
LOG("Problem when finding %s method from %s : %s \n", PLUGIN_HANDLER, pname,error);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
dictionary dic = decode_request(client,method,query_string);
|
||||||
|
(*fn)(client,method,pfunc,dic);
|
||||||
|
free(dic);
|
||||||
|
free(rpath);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
@ -4,8 +4,16 @@
|
|||||||
#include <sys/stat.h>
|
#include <sys/stat.h>
|
||||||
#include <pthread.h>
|
#include <pthread.h>
|
||||||
#include <signal.h>
|
#include <signal.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include "plugins/dictionary.h"
|
||||||
#include "plugin_manager.h"
|
#include "plugin_manager.h"
|
||||||
|
|
||||||
|
#define FORM_URL_ENCODE "application/x-www-form-urlencoded"
|
||||||
|
#define FORM_MULTI_PART "multipart/form-data"
|
||||||
|
#define APP_JSON "application/json"
|
||||||
|
#define PLUGIN_HANDLER "handler"
|
||||||
|
#define WS_MAGIC_STRING "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
|
||||||
|
|
||||||
|
|
||||||
#define ISspace(x) isspace((int)(x))
|
#define ISspace(x) isspace((int)(x))
|
||||||
|
|
||||||
@ -21,4 +29,17 @@ void headers(int, const char *);
|
|||||||
void not_found(int);
|
void not_found(int);
|
||||||
void serve_file(int, const char *);
|
void serve_file(int, const char *);
|
||||||
int startup(unsigned *);
|
int startup(unsigned *);
|
||||||
void unimplemented(int);
|
void unimplemented(int);
|
||||||
|
|
||||||
|
void ws_confirm_request(int, const char*);
|
||||||
|
char* post_url_decode(int client,int len);
|
||||||
|
dictionary decode_url_request(const char* query);
|
||||||
|
dictionary decode_request(int client,const char* method,const char* query);
|
||||||
|
dictionary decode_multi_part_request(int,const char*);
|
||||||
|
dictionary decode_cookie(const char*);
|
||||||
|
char* json_data_decode(int,int);
|
||||||
|
char* read_line(int sock);
|
||||||
|
int read_buf(int sock,char* buf,int i);
|
||||||
|
|
||||||
|
int execute_plugin(int client, const char *path,
|
||||||
|
const char *method, const char *query_string);
|
2
httpd.c
2
httpd.c
@ -9,7 +9,7 @@ static int config_handler(void* conf, const char* section, const char* name,
|
|||||||
const char* value)
|
const char* value)
|
||||||
{
|
{
|
||||||
config_t* pconfig = (config_t*)conf;
|
config_t* pconfig = (config_t*)conf;
|
||||||
char * ppath = NULL;
|
//char * ppath = NULL;
|
||||||
if (MATCH("SERVER", "port")) {
|
if (MATCH("SERVER", "port")) {
|
||||||
pconfig->port = atoi(value);
|
pconfig->port = atoi(value);
|
||||||
} else if (MATCH("SERVER", "plugins")) {
|
} else if (MATCH("SERVER", "plugins")) {
|
||||||
|
503
plugin_manager.c
503
plugin_manager.c
@ -4,443 +4,6 @@
|
|||||||
*/
|
*/
|
||||||
static struct plugin_entry *plugin_table[HASHSIZE];
|
static struct plugin_entry *plugin_table[HASHSIZE];
|
||||||
config_t server_config;
|
config_t server_config;
|
||||||
/**
|
|
||||||
* Decode the HTTP POST request with URL encode
|
|
||||||
* @param client socket client
|
|
||||||
* @param len content length
|
|
||||||
* @return query string
|
|
||||||
*/
|
|
||||||
char* post_url_decode(int client,int len)
|
|
||||||
{
|
|
||||||
char *query = (char*) malloc((len+1)*sizeof(char));
|
|
||||||
for (int i = 0; i < len; i++) {
|
|
||||||
recv(client, (query+i), 1, 0);
|
|
||||||
}
|
|
||||||
query[len]='\0';
|
|
||||||
//query = url_decode(query);
|
|
||||||
LOG("POST Query %s\n", query);
|
|
||||||
return query;
|
|
||||||
}
|
|
||||||
/**
|
|
||||||
* Decode the HTTP request
|
|
||||||
* Get the cookie values
|
|
||||||
* if it is the GET request, decode the query string into a dictionary
|
|
||||||
* if it is a POST, check the content type of the request
|
|
||||||
* - if it is a POST request with URL encoded : decode the url encode
|
|
||||||
* - if it is a POST request with multipart form data: de code the multipart
|
|
||||||
* - if other - UNIMPLEMENTED
|
|
||||||
* @param client socket client
|
|
||||||
* @param method HTTP method
|
|
||||||
* @param query query string in case of GET
|
|
||||||
* @return a dictionary of key- value
|
|
||||||
*/
|
|
||||||
dictionary decode_request(int client,const char* method,const char* query)
|
|
||||||
{
|
|
||||||
dictionary request = NULL;
|
|
||||||
dictionary cookie = NULL;
|
|
||||||
dictionary xheader = dict();
|
|
||||||
char* line;
|
|
||||||
char * token;
|
|
||||||
if(strcmp(method,"GET") == 0)
|
|
||||||
{
|
|
||||||
// this for check if web socket is enabled
|
|
||||||
int ws= 0;
|
|
||||||
char* ws_key = NULL;
|
|
||||||
while((line = read_line(client)) && strcmp("\r\n",line))
|
|
||||||
{
|
|
||||||
trim(line, '\n');
|
|
||||||
trim(line, '\r');
|
|
||||||
token = strsep(&line,":");
|
|
||||||
trim(token,' ');
|
|
||||||
trim(line,' ');
|
|
||||||
dput(xheader,token,line);
|
|
||||||
if(token != NULL &&strcasecmp(token,"Cookie") == 0)
|
|
||||||
{
|
|
||||||
if(!cookie) cookie = decode_cookie(line);
|
|
||||||
}
|
|
||||||
else if(token != NULL && strcasecmp(token,"Upgrade") == 0)
|
|
||||||
{
|
|
||||||
// verify that the connection is upgrade to websocket
|
|
||||||
trim(line, ' ');
|
|
||||||
if(line != NULL && strcasecmp(line,"websocket") == 0)
|
|
||||||
ws = 1;
|
|
||||||
} else if(token != NULL && strcasecmp(token,"Sec-WebSocket-Key") == 0)
|
|
||||||
{
|
|
||||||
// get the key from the client
|
|
||||||
trim(line, ' ');
|
|
||||||
ws_key = strdup(line);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
request = decode_url_request(query);
|
|
||||||
if(ws && ws_key != NULL)
|
|
||||||
{
|
|
||||||
ws_confirm_request(client, ws_key);
|
|
||||||
free(ws_key);
|
|
||||||
// insert wsocket flag to request
|
|
||||||
// plugin should handle this ugraded connection
|
|
||||||
// not the server
|
|
||||||
if(!request) request = dict();
|
|
||||||
dput(request,"__web_socket__","1");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
char* ctype = NULL;
|
|
||||||
int clen = -1;
|
|
||||||
line = read_line(client);
|
|
||||||
while (line && strcmp("\r\n",line))
|
|
||||||
{
|
|
||||||
//printf("%s\n",line);
|
|
||||||
trim(line, '\n');
|
|
||||||
trim(line, '\r');
|
|
||||||
token = strsep(&line,":");
|
|
||||||
trim(token,' ');
|
|
||||||
trim(line, ' ');
|
|
||||||
dput(xheader,token,line);
|
|
||||||
if(token != NULL &&strcasecmp(token,"Content-Type") == 0)
|
|
||||||
{
|
|
||||||
ctype = strsep(&line,":");
|
|
||||||
trim(ctype,' ');
|
|
||||||
} else if(token != NULL &&strcasecmp(token,"Content-Length") == 0)
|
|
||||||
{
|
|
||||||
token = strsep(&line,":");
|
|
||||||
trim(token,' ');
|
|
||||||
clen = atoi(token);
|
|
||||||
}
|
|
||||||
else if(token != NULL &&strcasecmp(token,"Cookie") == 0)
|
|
||||||
{
|
|
||||||
if(!cookie) cookie = decode_cookie(line);
|
|
||||||
}
|
|
||||||
|
|
||||||
line = read_line(client);
|
|
||||||
}
|
|
||||||
free(line);
|
|
||||||
if(ctype == NULL || clen == -1)
|
|
||||||
{
|
|
||||||
LOG("Bad request\n");
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
// decide what to do with the data
|
|
||||||
if(strstr(ctype,FORM_URL_ENCODE) > 0)
|
|
||||||
{
|
|
||||||
request = decode_url_request(post_url_decode(client,clen));
|
|
||||||
} else if(strstr(ctype,FORM_MULTI_PART)> 0)
|
|
||||||
{
|
|
||||||
//printf("Multi part form : %s\n", ctype);
|
|
||||||
request = decode_multi_part_request(client,ctype);
|
|
||||||
}
|
|
||||||
else if(strstr(ctype,APP_JSON) > 0)
|
|
||||||
{
|
|
||||||
char* query = json_data_decode(client,clen);
|
|
||||||
request = dict();
|
|
||||||
dput(request,"json", strdup(query));
|
|
||||||
free(query);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LOG("Un supported yet %s\n",ctype);
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
//if(cookie->key == NULL) {free(cookie);cookie= NULL;}
|
|
||||||
if(!request)
|
|
||||||
request = dict();
|
|
||||||
|
|
||||||
dput(request,"cookie",cookie);
|
|
||||||
dput(request,"__xheader__",xheader);
|
|
||||||
return request;
|
|
||||||
}
|
|
||||||
void __px(const char* data,int size)
|
|
||||||
{
|
|
||||||
for (int i = 0; i < size; ++i)
|
|
||||||
printf(" %02x", data[i]);
|
|
||||||
|
|
||||||
printf("\n");
|
|
||||||
}
|
|
||||||
/**
|
|
||||||
* Send header to the client to confirm
|
|
||||||
* that the websocket is accepted by
|
|
||||||
* our server
|
|
||||||
*/
|
|
||||||
void ws_confirm_request(int client, const char* key)
|
|
||||||
{
|
|
||||||
char buf[256];
|
|
||||||
char rkey[128];
|
|
||||||
char sha_d[20];
|
|
||||||
char base64[64];
|
|
||||||
strcpy(rkey,key);
|
|
||||||
strcat(rkey,WS_MAGIC_STRING);
|
|
||||||
//printf("RESPONDKEY '%s'\n", rkey);
|
|
||||||
SHA1_CTX context;
|
|
||||||
SHA1_Init(&context);
|
|
||||||
SHA1_Update(&context, rkey, strlen(rkey));
|
|
||||||
SHA1_Final(&context, sha_d);
|
|
||||||
Base64encode(base64, sha_d, 20);
|
|
||||||
//printf("Base 64 '%s'\n", base64);
|
|
||||||
// send accept to client
|
|
||||||
sprintf(buf, "HTTP/1.1 101 Switching Protocols\r\n");
|
|
||||||
send(client, buf, strlen(buf), 0);
|
|
||||||
sprintf(buf, "Upgrade: websocket\r\n");
|
|
||||||
send(client, buf, strlen(buf), 0);
|
|
||||||
sprintf(buf, "Connection: Upgrade\r\n");
|
|
||||||
send(client, buf, strlen(buf), 0);
|
|
||||||
sprintf(buf, "Sec-WebSocket-Accept: %s\r\n",base64);
|
|
||||||
send(client, buf, strlen(buf), 0);
|
|
||||||
sprintf(buf, "\r\n");
|
|
||||||
send(client, buf, strlen(buf), 0);
|
|
||||||
|
|
||||||
LOG("%s\n", "Websocket is now enabled for plugin");
|
|
||||||
}
|
|
||||||
/**
|
|
||||||
* Decode the cookie header to a dictionary
|
|
||||||
* @param client The client socket
|
|
||||||
* @return The Dictionary socket or NULL
|
|
||||||
*/
|
|
||||||
dictionary decode_cookie(const char* line)
|
|
||||||
{
|
|
||||||
char *token,*token1;
|
|
||||||
char *cpstr = strdup(line);
|
|
||||||
trim(cpstr,' ');
|
|
||||||
trim(cpstr,'\n');
|
|
||||||
trim(cpstr,'\r');
|
|
||||||
//printf("FUCKIT %s\n",cpstr );
|
|
||||||
dictionary dic = NULL;
|
|
||||||
while((token = strsep(&cpstr,";")))
|
|
||||||
{
|
|
||||||
trim(token,' ');
|
|
||||||
token1 = strsep(&token,"=");
|
|
||||||
if(token1)
|
|
||||||
{
|
|
||||||
if(dic == NULL)
|
|
||||||
dic = dict();
|
|
||||||
//LOG("Found cookie : %s = %s\n",token1,token);
|
|
||||||
dput(dic,token1,strdup(token));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
//}
|
|
||||||
return dic;
|
|
||||||
//free(cpstr);
|
|
||||||
}
|
|
||||||
/**
|
|
||||||
* Decode the multi-part form data from the POST request
|
|
||||||
* If it is a file upload, copy the file to tmp dir
|
|
||||||
* and generate the metadata for the server-side
|
|
||||||
* @param client the socket client
|
|
||||||
* @param ctype Content-Type of the request
|
|
||||||
* @param clen Content length, but not used here
|
|
||||||
* @return a dictionary of key - value
|
|
||||||
*/
|
|
||||||
dictionary decode_multi_part_request(int client,const char* ctype)
|
|
||||||
{
|
|
||||||
char * boundary;
|
|
||||||
char * boundend;
|
|
||||||
char * line;
|
|
||||||
char * str_copy = strdup(ctype);
|
|
||||||
char* token;
|
|
||||||
char* keytoken ;
|
|
||||||
char* valtoken ;
|
|
||||||
char* part_name;
|
|
||||||
char* part_file;
|
|
||||||
char* file_path;
|
|
||||||
char buf[BUFFLEN];
|
|
||||||
char* field;
|
|
||||||
dictionary dic = NULL;
|
|
||||||
FILE *fp = NULL;
|
|
||||||
boundary = strsep(&str_copy,"="); //discard first part
|
|
||||||
boundary = strsep(&str_copy,"=");
|
|
||||||
if(boundary && strlen(boundary)>0)
|
|
||||||
{
|
|
||||||
dic = dict();
|
|
||||||
trim(boundary,' ');
|
|
||||||
boundend = __s("%s--",boundary);
|
|
||||||
//find first boundary
|
|
||||||
while((line = read_line(client))&&strstr(line,boundary) <= 0);
|
|
||||||
// loop through each part separated by the boundary
|
|
||||||
while(line && strstr(line,boundary) > 0){
|
|
||||||
// search for content disposition:
|
|
||||||
while((line = read_line(client)) &&
|
|
||||||
strstr(line,"Content-Disposition:") <= 0);
|
|
||||||
if(strstr(line,"Content-Disposition:") <= 0) return NULL;
|
|
||||||
// extract parameters from header
|
|
||||||
part_name = NULL;
|
|
||||||
part_file = NULL;
|
|
||||||
while((token = strsep(&line,";")))
|
|
||||||
{
|
|
||||||
keytoken = strsep(&token,"=");
|
|
||||||
if(keytoken && strlen(keytoken)>0)
|
|
||||||
{
|
|
||||||
trim(keytoken,' ');
|
|
||||||
valtoken = strsep(&token,"=");
|
|
||||||
if(valtoken)
|
|
||||||
{
|
|
||||||
trim(valtoken,' ');
|
|
||||||
trim(valtoken,'\n');
|
|
||||||
trim(valtoken,'\r');
|
|
||||||
trim(valtoken,'\"');
|
|
||||||
if(strcmp(keytoken,"name") == 0)
|
|
||||||
{
|
|
||||||
part_name = valtoken;
|
|
||||||
} else if(strcmp(keytoken,"filename") == 0)
|
|
||||||
{
|
|
||||||
part_file = valtoken;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// get the binary data
|
|
||||||
if(part_name != NULL)
|
|
||||||
{
|
|
||||||
// go to the beginer of data bock
|
|
||||||
while((line = read_line(client)) && strcmp(line,"\r\n") != 0);
|
|
||||||
if(part_file == NULL)
|
|
||||||
{
|
|
||||||
/**
|
|
||||||
* This allow only 1024 bytes of data (max),
|
|
||||||
* out of this range, the data is cut out.
|
|
||||||
* Need an efficient way to handle this
|
|
||||||
*/
|
|
||||||
line = read_line(client);
|
|
||||||
trim(line,'\n');
|
|
||||||
trim(line,'\r');
|
|
||||||
trim(line,' ');
|
|
||||||
dput(dic,part_name,line);
|
|
||||||
// find the next boundary
|
|
||||||
while((line = read_line(client)) && strstr(line,boundary) <= 0);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
file_path = __s("%s%s.%u",server_config.tmpdir,part_file,(unsigned)time(NULL));
|
|
||||||
fp=fopen(file_path, "wb");
|
|
||||||
if(fp)
|
|
||||||
{
|
|
||||||
int totalsize=0,len=0;
|
|
||||||
//read until the next boundary
|
|
||||||
while((len = read_buf(client,buf,sizeof(buf))) > 0 && strstr(buf,boundary) <= 0)
|
|
||||||
{
|
|
||||||
fwrite(buf, len, 1, fp);
|
|
||||||
totalsize += len;
|
|
||||||
}
|
|
||||||
//remove \r\n at the end
|
|
||||||
fseek(fp,-2, SEEK_CUR);
|
|
||||||
totalsize -= 2;
|
|
||||||
fclose(fp);
|
|
||||||
line = buf;
|
|
||||||
|
|
||||||
field = __s("%s.file",part_name);
|
|
||||||
dput(dic,field,part_file);
|
|
||||||
field = __s("%s.tmp",part_name);
|
|
||||||
dput(dic,field,strdup(file_path));
|
|
||||||
field = __s("%s.size",part_name);
|
|
||||||
dput(dic,field,__s("%d",totalsize));
|
|
||||||
field = __s("%s.ext",part_name);
|
|
||||||
dput(dic,field,ext(part_file));
|
|
||||||
free(field);
|
|
||||||
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LOG("Cannot wirte file to :%s\n", file_path );
|
|
||||||
}
|
|
||||||
free(file_path);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
//printf("[Lines]:%s\n",line);
|
|
||||||
// check if end of request
|
|
||||||
if(line&&strstr(line,boundend)>0)
|
|
||||||
{
|
|
||||||
LOG("End request %s\n", boundend);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
free(str_copy);
|
|
||||||
return dic;
|
|
||||||
}
|
|
||||||
/**
|
|
||||||
* Decode a query string (GET request or POST URL encoded) to
|
|
||||||
* a dictionary of key-value
|
|
||||||
* @param query : the query string
|
|
||||||
* @return a dictionary of key-value
|
|
||||||
*/
|
|
||||||
dictionary decode_url_request(const char* query)
|
|
||||||
{
|
|
||||||
char* str_copy = strdup(query);
|
|
||||||
//str_copy = ;
|
|
||||||
char* token;
|
|
||||||
if(query == NULL) return NULL;
|
|
||||||
if(strlen(query) == 0) return NULL;
|
|
||||||
dictionary dic = dict();
|
|
||||||
while ((token = strsep(&str_copy, "&")))
|
|
||||||
{
|
|
||||||
char* key;
|
|
||||||
char* val;
|
|
||||||
if(strlen(token)>0)
|
|
||||||
{
|
|
||||||
key = strsep(&token,"=");
|
|
||||||
if(key && strlen(key)>0)
|
|
||||||
{
|
|
||||||
val = strsep(&token,"=");
|
|
||||||
dput(dic,key,url_decode(val));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
free(str_copy);
|
|
||||||
return dic;
|
|
||||||
}
|
|
||||||
/**
|
|
||||||
* Decode JSON query string to string
|
|
||||||
*/
|
|
||||||
char* json_data_decode(int client,int len)
|
|
||||||
{
|
|
||||||
char *query = (char*) malloc((len+1)*sizeof(char));
|
|
||||||
for (int i = 0; i < len; i++) {
|
|
||||||
recv(client, (query+i), 1, 0);
|
|
||||||
}
|
|
||||||
query[len]='\0';
|
|
||||||
//query = url_decode(query);
|
|
||||||
LOG("JSON Query %s\n", query);
|
|
||||||
return query;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* read the request as a string line format
|
|
||||||
* @param sock socket
|
|
||||||
* @return a request string
|
|
||||||
*/
|
|
||||||
char* read_line(int sock)
|
|
||||||
{
|
|
||||||
char buf[BUFFLEN];
|
|
||||||
read_buf(sock,buf,sizeof(buf));
|
|
||||||
return strdup(buf);
|
|
||||||
}
|
|
||||||
/**
|
|
||||||
* Read the socket request in to a buffer or size
|
|
||||||
* The data is read until the buffer is full or
|
|
||||||
* there are a carrier return character
|
|
||||||
* @param sock socket
|
|
||||||
* @param buf buffer
|
|
||||||
* @param size size of buffer
|
|
||||||
* @return number of bytes read
|
|
||||||
*/
|
|
||||||
int read_buf(int sock, char*buf,int size)
|
|
||||||
{
|
|
||||||
int i = 0;
|
|
||||||
char c = '\0';
|
|
||||||
int n;
|
|
||||||
while ((i < size - 1) && (c != '\n'))
|
|
||||||
{
|
|
||||||
n = recv(sock, &c, 1, 0);
|
|
||||||
if (n > 0)
|
|
||||||
{
|
|
||||||
buf[i] = c;
|
|
||||||
i++;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
c = '\n';
|
|
||||||
}
|
|
||||||
buf[i] = '\0';
|
|
||||||
return i;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Locate a plugin in the plugin table
|
* Locate a plugin in the plugin table
|
||||||
@ -570,69 +133,3 @@ void unload_all_plugin()
|
|||||||
exit(0);
|
exit(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Execute a plugin based on the http requeset
|
|
||||||
* First decode the http request header to find the correct plugin
|
|
||||||
* and the correct function on the plugin
|
|
||||||
* Second, decode all parameters necessary of the request and pass it
|
|
||||||
* to the callback function.
|
|
||||||
* Execute the callback function if sucess
|
|
||||||
* @param client soket client
|
|
||||||
* @param path request path
|
|
||||||
* @param method request method
|
|
||||||
* @param query_string GET query string
|
|
||||||
* @return -1 if failure
|
|
||||||
* 1 if sucess
|
|
||||||
*/
|
|
||||||
int execute_plugin(int client, const char *path, const char *method, const char *query_string)
|
|
||||||
{
|
|
||||||
char pname[255];
|
|
||||||
char pfunc[255];
|
|
||||||
void (*fn)(int, const char*,const char*, dictionary);
|
|
||||||
struct plugin_entry *plugin ;
|
|
||||||
int plen = strlen(path);
|
|
||||||
char * rpath = (char*) malloc((plen+1)*sizeof(char));
|
|
||||||
char *error;
|
|
||||||
memcpy(rpath,path+1,plen);
|
|
||||||
rpath[plen] = '\0';
|
|
||||||
trim(rpath,'/');
|
|
||||||
char * delim = strchr(rpath,'/');
|
|
||||||
if(delim == NULL)
|
|
||||||
{
|
|
||||||
strcpy(pname,rpath);
|
|
||||||
strcpy(pfunc,"default");
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
int npos,fpos;
|
|
||||||
npos = delim - rpath;
|
|
||||||
fpos = strlen(rpath) - npos ;
|
|
||||||
memcpy(pname,rpath,npos);
|
|
||||||
pname[npos] = '\0';
|
|
||||||
memcpy(pfunc,rpath+npos+1,fpos);
|
|
||||||
pfunc[fpos-1]='\0';
|
|
||||||
}
|
|
||||||
LOG("Client %d\n",client );
|
|
||||||
LOG("Path : '%s'\n", rpath);
|
|
||||||
LOG("Method:%s\n", method);
|
|
||||||
LOG("Plugin name '%s'\n",pname);
|
|
||||||
LOG("Query path. '%s'\n", pfunc);
|
|
||||||
LOG("query :%s\n", query_string);
|
|
||||||
|
|
||||||
//load the plugin
|
|
||||||
if((plugin = plugin_lookup(pname)) == NULL)
|
|
||||||
if((plugin= plugin_load(pname)) == NULL)
|
|
||||||
return -1;
|
|
||||||
// load the function
|
|
||||||
fn = (void (*)(int, const char *, const char*, dictionary))dlsym(plugin->handle, PLUGIN_HANDLER);
|
|
||||||
if ((error = dlerror()) != NULL)
|
|
||||||
{
|
|
||||||
LOG("Problem when finding %s method from %s : %s \n", PLUGIN_HANDLER, pname,error);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
dictionary dic = decode_request(client,method,query_string);
|
|
||||||
(*fn)(client,method,pfunc,dic);
|
|
||||||
free(dic);
|
|
||||||
free(rpath);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
@ -1,13 +1,9 @@
|
|||||||
|
#ifndef PLUGIN_MANAGER_H
|
||||||
|
#define PLUGIN_MANAGER_H
|
||||||
#include <dlfcn.h>
|
#include <dlfcn.h>
|
||||||
#include <sys/socket.h>
|
#include "plugins/utils.h"
|
||||||
#include "plugins/dictionary.h"
|
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
|
|
||||||
#define FORM_URL_ENCODE "application/x-www-form-urlencoded"
|
|
||||||
#define FORM_MULTI_PART "multipart/form-data"
|
|
||||||
#define APP_JSON "application/json"
|
|
||||||
#define PLUGIN_HANDLER "handler"
|
|
||||||
#define WS_MAGIC_STRING "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
|
|
||||||
struct plugin_entry {
|
struct plugin_entry {
|
||||||
struct plugin_entry *next;
|
struct plugin_entry *next;
|
||||||
char *pname;
|
char *pname;
|
||||||
@ -22,14 +18,5 @@ void unload_all_plugin();
|
|||||||
void unload_plugin(struct plugin_entry*);
|
void unload_plugin(struct plugin_entry*);
|
||||||
void unload_plugin_by_name(const char*);
|
void unload_plugin_by_name(const char*);
|
||||||
void * plugin_from_file(char* name);
|
void * plugin_from_file(char* name);
|
||||||
void ws_confirm_request(int, const char*);
|
|
||||||
char* post_url_decode(int client,int len);
|
#endif
|
||||||
dictionary decode_url_request(const char* query);
|
|
||||||
dictionary decode_request(int client,const char* method,const char* query);
|
|
||||||
dictionary decode_multi_part_request(int,const char*);
|
|
||||||
dictionary decode_cookie(const char*);
|
|
||||||
char* json_data_decode(int,int);
|
|
||||||
char* read_line(int sock);
|
|
||||||
int read_buf(int sock,char* buf,int i);
|
|
||||||
int execute_plugin(int client, const char *path,
|
|
||||||
const char *method, const char *query_string);
|
|
@ -18,7 +18,7 @@ void execute(int client,const char* method,dictionary rq)
|
|||||||
dictionary d = dict();
|
dictionary d = dict();
|
||||||
dput(d,"test",c);
|
dput(d,"test",c);
|
||||||
dput(d,"test1","This is another cookie");
|
dput(d,"test1","This is another cookie");
|
||||||
set_cookie(client,"text/html; charset=utf-8",d);
|
set_cookie(client,"text/html; charset=utf-8",d,__plugin__.name);
|
||||||
|
|
||||||
LOG("%s",c);
|
LOG("%s",c);
|
||||||
__t(client,"<h1>Set the cookie</h1>");
|
__t(client,"<h1>Set the cookie</h1>");
|
||||||
|
@ -1,3 +1,5 @@
|
|||||||
|
#ifndef DB_HELPER
|
||||||
|
#define DB_HELPER
|
||||||
#include <sqlite3.h>
|
#include <sqlite3.h>
|
||||||
#include "utils.h"
|
#include "utils.h"
|
||||||
|
|
||||||
@ -31,4 +33,5 @@ char* __name_value_list(const dbfield);
|
|||||||
char* value_of(const dbfield,const char*);
|
char* value_of(const dbfield,const char*);
|
||||||
void add_field(dbfield*,const char*, const char*);
|
void add_field(dbfield*,const char*, const char*);
|
||||||
void add_record(dbrecord*,dbfield);
|
void add_record(dbrecord*,dbfield);
|
||||||
void dbclose(sqlite3*);
|
void dbclose(sqlite3*);
|
||||||
|
#endif
|
217
plugins/handle.c
Normal file
217
plugins/handle.c
Normal file
@ -0,0 +1,217 @@
|
|||||||
|
#include "handle.h"
|
||||||
|
|
||||||
|
|
||||||
|
void set_status(int client,int code,const char* msg)
|
||||||
|
{
|
||||||
|
response(client, __s("HTTP/1.0 %d %s", code, msg));
|
||||||
|
response(client, __s("Server: %s ", SERVER_NAME));
|
||||||
|
}
|
||||||
|
void redirect(int client,const char*path)
|
||||||
|
{
|
||||||
|
__t(client,"<html><head><meta http-equiv=\"refresh\" content=\"0; url=%s\"></head><body></body></html>",path);
|
||||||
|
}
|
||||||
|
|
||||||
|
void html(int client)
|
||||||
|
{
|
||||||
|
ctype(client,"text/html; charset=utf-8");
|
||||||
|
}
|
||||||
|
void text(int client)
|
||||||
|
{
|
||||||
|
ctype(client,"text/plain; charset=utf-8");
|
||||||
|
}
|
||||||
|
void json(int client)
|
||||||
|
{
|
||||||
|
ctype(client,"application/json");
|
||||||
|
}
|
||||||
|
void textstream(int client)
|
||||||
|
{
|
||||||
|
ctype(client, "text/event-stream");
|
||||||
|
}
|
||||||
|
void octstream(int client, char* name)
|
||||||
|
{
|
||||||
|
set_status(client,200,"OK");
|
||||||
|
__t(client,"Content-Type: application/octet-stream");
|
||||||
|
__t(client,"Content-Disposition: attachment; filename=\"%s\"", name);
|
||||||
|
response(client,"");
|
||||||
|
//Content-Disposition: attachment; filename="fname.ext"
|
||||||
|
}
|
||||||
|
void jpeg(int client)
|
||||||
|
{
|
||||||
|
ctype(client,"image/jpeg");
|
||||||
|
}
|
||||||
|
void ctype(int client, const char* type)
|
||||||
|
{
|
||||||
|
set_status(client,200,"OK");
|
||||||
|
__t(client,"Content-Type: %s",type);
|
||||||
|
response(client,"");
|
||||||
|
}
|
||||||
|
|
||||||
|
int response(int client, const char* data)
|
||||||
|
{
|
||||||
|
char buf[BUFFLEN+3];
|
||||||
|
strcpy(buf, data);
|
||||||
|
int nbytes;
|
||||||
|
int size = strlen(data);
|
||||||
|
buf[size] = '\r';
|
||||||
|
buf[size+1] = '\n';
|
||||||
|
buf[size+2] = '\0';
|
||||||
|
nbytes = send(client, buf, strlen(buf), 0);
|
||||||
|
return (nbytes ==-1?0:1);
|
||||||
|
}
|
||||||
|
int __ti(int client,int data)
|
||||||
|
{
|
||||||
|
char str[15];
|
||||||
|
sprintf(str, "%d", data);
|
||||||
|
return response(client,str);
|
||||||
|
}
|
||||||
|
|
||||||
|
int __t(int client, const char* fstring,...)
|
||||||
|
{
|
||||||
|
int nbytes;
|
||||||
|
int dlen;
|
||||||
|
int sent = 0;
|
||||||
|
int buflen = 0;
|
||||||
|
va_list arguments;
|
||||||
|
char * data;
|
||||||
|
char* chunk;
|
||||||
|
va_start( arguments, fstring);
|
||||||
|
dlen = vsnprintf(0,0,fstring,arguments) + 1;
|
||||||
|
va_end(arguments);
|
||||||
|
if ((data = (char*)malloc(dlen*sizeof(char))) != 0)
|
||||||
|
{
|
||||||
|
va_start(arguments, fstring);
|
||||||
|
vsnprintf(data, dlen, fstring, arguments);
|
||||||
|
va_end(arguments);
|
||||||
|
|
||||||
|
if(dlen < BUFFLEN)
|
||||||
|
return response(client,data);
|
||||||
|
else
|
||||||
|
{
|
||||||
|
while(sent < dlen - 1)
|
||||||
|
{
|
||||||
|
if(dlen - sent > BUFFLEN)
|
||||||
|
buflen = BUFFLEN;
|
||||||
|
else
|
||||||
|
buflen = dlen - sent - 1;
|
||||||
|
//LOG("BUFFLEN %d\n",buflen);
|
||||||
|
chunk = (char*) malloc((buflen)*sizeof(char));
|
||||||
|
memcpy(chunk,data+sent,buflen);
|
||||||
|
//chunk[buflen-1] = '\0';
|
||||||
|
//response(client,chunk);
|
||||||
|
sent += buflen;
|
||||||
|
nbytes = send(client, chunk, buflen, 0);
|
||||||
|
free(chunk);
|
||||||
|
if(nbytes == -1) return 0;
|
||||||
|
}
|
||||||
|
chunk = "\r\n";
|
||||||
|
send(client, chunk, strlen(chunk), 0);
|
||||||
|
}
|
||||||
|
free(data);
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
//
|
||||||
|
}
|
||||||
|
int __b(int client, const unsigned char* data, int size)
|
||||||
|
{
|
||||||
|
char buf[BUFFLEN];
|
||||||
|
int sent = 0;
|
||||||
|
int buflen = 0;
|
||||||
|
int nbytes;
|
||||||
|
if(size <= BUFFLEN)
|
||||||
|
{
|
||||||
|
nbytes = send(client,data,size,0);
|
||||||
|
return (nbytes==-1?0:1);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
while(sent < size)
|
||||||
|
{
|
||||||
|
if(size - sent > BUFFLEN)
|
||||||
|
buflen = BUFFLEN;
|
||||||
|
else
|
||||||
|
buflen = size - sent;
|
||||||
|
memcpy(buf,data+sent,buflen);
|
||||||
|
nbytes = send(client,buf,buflen,0);
|
||||||
|
sent += buflen;
|
||||||
|
if(nbytes == -1) return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
int __fb(int client, const char* file)
|
||||||
|
{
|
||||||
|
printf("Open file %s\n",file );
|
||||||
|
unsigned char buffer[BUFFLEN];
|
||||||
|
FILE *ptr;
|
||||||
|
ptr = fopen(file,"rb");
|
||||||
|
if(!ptr)
|
||||||
|
{
|
||||||
|
LOG("Cannot read : %s\n", file);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
size_t size;
|
||||||
|
while(!feof(ptr))
|
||||||
|
{
|
||||||
|
size = fread(buffer,1,BUFFLEN,ptr);
|
||||||
|
if(!__b(client,buffer,size)) return 0;
|
||||||
|
}
|
||||||
|
fclose(ptr);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
int __f(int client, const char* file)
|
||||||
|
{
|
||||||
|
unsigned char buf[BUFFLEN];
|
||||||
|
FILE *ptr;
|
||||||
|
int nbytes;
|
||||||
|
ptr = fopen(file,"r");
|
||||||
|
if(!ptr)
|
||||||
|
{
|
||||||
|
LOG("Cannot read : %s\n", file);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
;
|
||||||
|
while(fgets(buf, sizeof(buf), ptr) != NULL)
|
||||||
|
{
|
||||||
|
nbytes = send(client, buf, strlen(buf), 0);
|
||||||
|
if(nbytes == -1) return 0;
|
||||||
|
//LOG("READ : %s\n", buf);
|
||||||
|
//fgets(buf, sizeof(buf), ptr);
|
||||||
|
}
|
||||||
|
fclose(ptr);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int upload(const char* tmp, const char* path)
|
||||||
|
{
|
||||||
|
return !rename(tmp, path);
|
||||||
|
}
|
||||||
|
// __plugin__.name
|
||||||
|
void set_cookie(int client,const char* type, dictionary dic, const char* name)
|
||||||
|
{
|
||||||
|
set_status(client,200,"OK");
|
||||||
|
__t(client,"Content-Type: %s",type);
|
||||||
|
association assoc;
|
||||||
|
for_each_assoc(assoc,dic){
|
||||||
|
__t(client,"Set-Cookie: %s=%s; Path=/%s",assoc->key, (char*)assoc->value, name);
|
||||||
|
}
|
||||||
|
response(client,"");
|
||||||
|
}
|
||||||
|
void clear_cookie(int client, dictionary dic)
|
||||||
|
{
|
||||||
|
set_status(client,200,"OK");
|
||||||
|
__t(client,"Content-Type: text/html; charset=utf-8");
|
||||||
|
association assoc;
|
||||||
|
for_each_assoc(assoc,dic){
|
||||||
|
__t(client,"Set-Cookie: %s=%s;expires=",assoc->key, (char*)assoc->value, server_time());
|
||||||
|
}
|
||||||
|
response(client,"");
|
||||||
|
}
|
||||||
|
void unknow(int client)
|
||||||
|
{
|
||||||
|
html(client);
|
||||||
|
__t(client,"404 API not found");
|
||||||
|
}
|
||||||
|
int ws_enable(dictionary dic)
|
||||||
|
{
|
||||||
|
return (dic != NULL && R_INT(dic,"__web_socket__") == 1);
|
||||||
|
}
|
48
plugins/handle.h
Normal file
48
plugins/handle.h
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
#ifndef HANDLE_H
|
||||||
|
#define HANDLE_H
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
#ifdef USE_DB
|
||||||
|
#include "dbhelper.h"
|
||||||
|
#endif
|
||||||
|
#include "dictionary.h"
|
||||||
|
#include "list.h"
|
||||||
|
#include "ini.h"
|
||||||
|
#include "ws.h"
|
||||||
|
|
||||||
|
#define SERVER_NAME "antd"
|
||||||
|
#define IS_POST(method) (strcmp(method,"POST")== 0)
|
||||||
|
#define IS_GET(method) (strcmp(method,"GET")== 0)
|
||||||
|
#define R_STR(d,k) ((char*)dvalue(d,k))
|
||||||
|
#define R_INT(d,k) (atoi(dvalue(d,k)))
|
||||||
|
#define R_FLOAT(d,k) ((double)atof(dvalue(d,k)))
|
||||||
|
#define R_PTR(d,k) (dvalue(d,k))
|
||||||
|
#define __RESULT__ "{\"result\":%d,\"msg\":\"%s\"}"
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
int response(int, const char*);
|
||||||
|
void ctype(int,const char*);
|
||||||
|
void redirect(int,const char*);
|
||||||
|
void html(int);
|
||||||
|
void text(int);
|
||||||
|
void json(int);
|
||||||
|
void jpeg(int);
|
||||||
|
void octstream(int, char*);
|
||||||
|
void textstream(int);
|
||||||
|
int __ti(int,int);
|
||||||
|
int __t(int, const char*,...);
|
||||||
|
int __b(int, const unsigned char*, int);
|
||||||
|
int __f(int, const char*);
|
||||||
|
int __fb(int, const char*);
|
||||||
|
int upload(const char*, const char*);
|
||||||
|
|
||||||
|
void set_cookie(int, const char*,dictionary,const char*);
|
||||||
|
void set_status(int,int,const char*);
|
||||||
|
void clear_cookie(int, dictionary);
|
||||||
|
/*Default function for plugin*/
|
||||||
|
void unknow(int);
|
||||||
|
int ws_enable(dictionary);
|
||||||
|
#endif
|
214
plugins/plugin.c
214
plugins/plugin.c
@ -30,185 +30,6 @@ sqldb getdb()
|
|||||||
return getdb(__plugin__.name);
|
return getdb(__plugin__.name);
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
void set_status(int client,int code,const char* msg)
|
|
||||||
{
|
|
||||||
response(client, __s("HTTP/1.0 %d %s", code, msg));
|
|
||||||
response(client, __s("Server: %s ", SERVER_NAME));
|
|
||||||
}
|
|
||||||
void redirect(int client,const char*path)
|
|
||||||
{
|
|
||||||
__t(client,"<html><head><meta http-equiv=\"refresh\" content=\"0; url=%s\"></head><body></body></html>",path);
|
|
||||||
}
|
|
||||||
|
|
||||||
void html(int client)
|
|
||||||
{
|
|
||||||
header(client,"text/html; charset=utf-8");
|
|
||||||
}
|
|
||||||
void text(int client)
|
|
||||||
{
|
|
||||||
header(client,"text/plain; charset=utf-8");
|
|
||||||
}
|
|
||||||
void json(int client)
|
|
||||||
{
|
|
||||||
header(client,"application/json");
|
|
||||||
}
|
|
||||||
void textstream(int client)
|
|
||||||
{
|
|
||||||
header(client, "text/event-stream");
|
|
||||||
}
|
|
||||||
void octstream(int client, char* name)
|
|
||||||
{
|
|
||||||
set_status(client,200,"OK");
|
|
||||||
__t(client,"Content-Type: application/octet-stream");
|
|
||||||
__t(client,"Content-Disposition: attachment; filename=\"%s\"", name);
|
|
||||||
response(client,"");
|
|
||||||
//Content-Disposition: attachment; filename="fname.ext"
|
|
||||||
}
|
|
||||||
void jpeg(int client)
|
|
||||||
{
|
|
||||||
header(client,"image/jpeg");
|
|
||||||
}
|
|
||||||
void header(int client, const char* type)
|
|
||||||
{
|
|
||||||
set_status(client,200,"OK");
|
|
||||||
__t(client,"Content-Type: %s",type);
|
|
||||||
response(client,"");
|
|
||||||
}
|
|
||||||
|
|
||||||
int response(int client, const char* data)
|
|
||||||
{
|
|
||||||
char buf[BUFFLEN+3];
|
|
||||||
strcpy(buf, data);
|
|
||||||
int nbytes;
|
|
||||||
int size = strlen(data);
|
|
||||||
buf[size] = '\r';
|
|
||||||
buf[size+1] = '\n';
|
|
||||||
buf[size+2] = '\0';
|
|
||||||
nbytes = send(client, buf, strlen(buf), 0);
|
|
||||||
return (nbytes ==-1?0:1);
|
|
||||||
}
|
|
||||||
int __ti(int client,int data)
|
|
||||||
{
|
|
||||||
char str[15];
|
|
||||||
sprintf(str, "%d", data);
|
|
||||||
return response(client,str);
|
|
||||||
}
|
|
||||||
|
|
||||||
int __t(int client, const char* fstring,...)
|
|
||||||
{
|
|
||||||
int nbytes;
|
|
||||||
int dlen;
|
|
||||||
int sent = 0;
|
|
||||||
int buflen = 0;
|
|
||||||
va_list arguments;
|
|
||||||
char * data;
|
|
||||||
char* chunk;
|
|
||||||
va_start( arguments, fstring);
|
|
||||||
dlen = vsnprintf(0,0,fstring,arguments) + 1;
|
|
||||||
va_end(arguments);
|
|
||||||
if ((data = (char*)malloc(dlen*sizeof(char))) != 0)
|
|
||||||
{
|
|
||||||
va_start(arguments, fstring);
|
|
||||||
vsnprintf(data, dlen, fstring, arguments);
|
|
||||||
va_end(arguments);
|
|
||||||
|
|
||||||
if(dlen < BUFFLEN)
|
|
||||||
return response(client,data);
|
|
||||||
else
|
|
||||||
{
|
|
||||||
while(sent < dlen - 1)
|
|
||||||
{
|
|
||||||
if(dlen - sent > BUFFLEN)
|
|
||||||
buflen = BUFFLEN;
|
|
||||||
else
|
|
||||||
buflen = dlen - sent - 1;
|
|
||||||
//LOG("BUFFLEN %d\n",buflen);
|
|
||||||
chunk = (char*) malloc((buflen)*sizeof(char));
|
|
||||||
memcpy(chunk,data+sent,buflen);
|
|
||||||
//chunk[buflen-1] = '\0';
|
|
||||||
//response(client,chunk);
|
|
||||||
sent += buflen;
|
|
||||||
nbytes = send(client, chunk, buflen, 0);
|
|
||||||
free(chunk);
|
|
||||||
if(nbytes == -1) return 0;
|
|
||||||
}
|
|
||||||
chunk = "\r\n";
|
|
||||||
send(client, chunk, strlen(chunk), 0);
|
|
||||||
}
|
|
||||||
free(data);
|
|
||||||
}
|
|
||||||
return 1;
|
|
||||||
//
|
|
||||||
}
|
|
||||||
int __b(int client, const unsigned char* data, int size)
|
|
||||||
{
|
|
||||||
char buf[BUFFLEN];
|
|
||||||
int sent = 0;
|
|
||||||
int buflen = 0;
|
|
||||||
int nbytes;
|
|
||||||
if(size <= BUFFLEN)
|
|
||||||
{
|
|
||||||
nbytes = send(client,data,size,0);
|
|
||||||
return (nbytes==-1?0:1);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
while(sent < size)
|
|
||||||
{
|
|
||||||
if(size - sent > BUFFLEN)
|
|
||||||
buflen = BUFFLEN;
|
|
||||||
else
|
|
||||||
buflen = size - sent;
|
|
||||||
memcpy(buf,data+sent,buflen);
|
|
||||||
nbytes = send(client,buf,buflen,0);
|
|
||||||
sent += buflen;
|
|
||||||
if(nbytes == -1) return 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
int __fb(int client, const char* file)
|
|
||||||
{
|
|
||||||
printf("Open file %s\n",file );
|
|
||||||
unsigned char buffer[BUFFLEN];
|
|
||||||
FILE *ptr;
|
|
||||||
ptr = fopen(file,"rb");
|
|
||||||
if(!ptr)
|
|
||||||
{
|
|
||||||
LOG("Cannot read : %s\n", file);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
size_t size;
|
|
||||||
while(!feof(ptr))
|
|
||||||
{
|
|
||||||
size = fread(buffer,1,BUFFLEN,ptr);
|
|
||||||
if(!__b(client,buffer,size)) return 0;
|
|
||||||
}
|
|
||||||
fclose(ptr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
int __f(int client, const char* file)
|
|
||||||
{
|
|
||||||
unsigned char buf[BUFFLEN];
|
|
||||||
FILE *ptr;
|
|
||||||
int nbytes;
|
|
||||||
ptr = fopen(file,"r");
|
|
||||||
if(!ptr)
|
|
||||||
{
|
|
||||||
LOG("Cannot read : %s\n", file);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
;
|
|
||||||
while(fgets(buf, sizeof(buf), ptr) != NULL)
|
|
||||||
{
|
|
||||||
nbytes = send(client, buf, strlen(buf), 0);
|
|
||||||
if(nbytes == -1) return 0;
|
|
||||||
//LOG("READ : %s\n", buf);
|
|
||||||
//fgets(buf, sizeof(buf), ptr);
|
|
||||||
}
|
|
||||||
fclose(ptr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
char* route(const char* repath)
|
char* route(const char* repath)
|
||||||
{
|
{
|
||||||
@ -225,6 +46,7 @@ char* route(const char* repath)
|
|||||||
}
|
}
|
||||||
return path;
|
return path;
|
||||||
}
|
}
|
||||||
|
|
||||||
char* htdocs(const char* repath)
|
char* htdocs(const char* repath)
|
||||||
{
|
{
|
||||||
if(repath != NULL)
|
if(repath != NULL)
|
||||||
@ -232,30 +54,7 @@ char* htdocs(const char* repath)
|
|||||||
else
|
else
|
||||||
return __s("%s",__plugin__.htdocs);
|
return __s("%s",__plugin__.htdocs);
|
||||||
}
|
}
|
||||||
int upload(const char* tmp, const char* path)
|
|
||||||
{
|
|
||||||
return !rename(tmp, path);
|
|
||||||
}
|
|
||||||
void set_cookie(int client,const char* type, dictionary dic)
|
|
||||||
{
|
|
||||||
set_status(client,200,"OK");
|
|
||||||
__t(client,"Content-Type: %s",type);
|
|
||||||
association assoc;
|
|
||||||
for_each_assoc(assoc,dic){
|
|
||||||
__t(client,"Set-Cookie: %s=%s; Path=/%s",assoc->key, (char*)assoc->value, __plugin__.name);
|
|
||||||
}
|
|
||||||
response(client,"");
|
|
||||||
}
|
|
||||||
void clear_cookie(int client, dictionary dic)
|
|
||||||
{
|
|
||||||
set_status(client,200,"OK");
|
|
||||||
__t(client,"Content-Type: text/html; charset=utf-8");
|
|
||||||
association assoc;
|
|
||||||
for_each_assoc(assoc,dic){
|
|
||||||
__t(client,"Set-Cookie: %s=%s;expires=",assoc->key, (char*)assoc->value, server_time());
|
|
||||||
}
|
|
||||||
response(client,"");
|
|
||||||
}
|
|
||||||
char* config_dir()
|
char* config_dir()
|
||||||
{
|
{
|
||||||
struct stat st;
|
struct stat st;
|
||||||
@ -263,13 +62,4 @@ char* config_dir()
|
|||||||
if (stat(path, &st) == -1)
|
if (stat(path, &st) == -1)
|
||||||
mkdir(path, 0755);
|
mkdir(path, 0755);
|
||||||
return path;
|
return path;
|
||||||
}
|
|
||||||
void unknow(int client)
|
|
||||||
{
|
|
||||||
html(client);
|
|
||||||
__t(client,"404 API not found");
|
|
||||||
}
|
|
||||||
int ws_enable(dictionary dic)
|
|
||||||
{
|
|
||||||
return (dic != NULL && R_INT(dic,"__web_socket__") == 1);
|
|
||||||
}
|
}
|
@ -1,29 +1,17 @@
|
|||||||
#include <sys/socket.h>
|
#ifndef PLUGIN_H
|
||||||
#include <netinet/in.h>
|
#define PLUGIN_H
|
||||||
#include <arpa/inet.h>
|
|
||||||
#ifdef USE_DB
|
#ifdef USE_DB
|
||||||
#include "dbhelper.h"
|
#include "dbhelper.h"
|
||||||
#endif
|
#endif
|
||||||
#include "dictionary.h"
|
#include "handle.h"
|
||||||
#include "list.h"
|
|
||||||
#include "ini.h"
|
|
||||||
#include "ws.h"
|
|
||||||
|
|
||||||
#define SERVER_NAME "antd"
|
|
||||||
#define IS_POST(method) (strcmp(method,"POST")== 0)
|
|
||||||
#define IS_GET(method) (strcmp(method,"GET")== 0)
|
|
||||||
#define R_STR(d,k) ((char*)dvalue(d,k))
|
|
||||||
#define R_INT(d,k) (atoi(dvalue(d,k)))
|
|
||||||
#define R_FLOAT(d,k) ((double)atof(dvalue(d,k)))
|
|
||||||
#define R_PTR(d,k) (dvalue(d,k))
|
|
||||||
#define __RESULT__ "{\"result\":%d,\"msg\":\"%s\"}"
|
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
char *name;
|
char *name;
|
||||||
char *dbpath;
|
char *dbpath;
|
||||||
char * htdocs;
|
char * htdocs;
|
||||||
char*pdir;
|
char*pdir;
|
||||||
int *sport;
|
int sport;
|
||||||
} plugin_header;
|
} plugin_header;
|
||||||
|
|
||||||
|
|
||||||
@ -35,32 +23,16 @@ typedef sqlite3* sqldb;
|
|||||||
extern plugin_header __plugin__;
|
extern plugin_header __plugin__;
|
||||||
extern call __init__;
|
extern call __init__;
|
||||||
|
|
||||||
int response(int, const char*);
|
|
||||||
void header(int,const char*);
|
|
||||||
void redirect(int,const char*);
|
|
||||||
void html(int);
|
|
||||||
void text(int);
|
|
||||||
void json(int);
|
|
||||||
void jpeg(int);
|
|
||||||
void octstream(int, char*);
|
|
||||||
void textstream(int);
|
|
||||||
int __ti(int,int);
|
|
||||||
int __t(int, const char*,...);
|
|
||||||
int __b(int, const unsigned char*, int);
|
|
||||||
int __f(int, const char*);
|
|
||||||
int __fb(int, const char*);
|
|
||||||
int upload(const char*, const char*);
|
|
||||||
char* config_dir();
|
|
||||||
char* route(const char*);
|
|
||||||
char* htdocs(const char*);
|
|
||||||
#ifdef USE_DB
|
#ifdef USE_DB
|
||||||
sqldb getdb();
|
sqldb getdb();
|
||||||
sqldb __getdb(char *name);
|
sqldb __getdb(char *name);
|
||||||
#endif
|
#endif
|
||||||
void set_cookie(int, const char*,dictionary);
|
|
||||||
void set_status(int,int,const char*);
|
char* route(const char*);
|
||||||
void clear_cookie(int, dictionary);
|
char* htdocs(const char*);
|
||||||
|
char* config_dir();
|
||||||
/*Default function for plugin*/
|
/*Default function for plugin*/
|
||||||
void handler(int, const char*,const char*,dictionary);
|
void handler(int, const char*,const char*,dictionary);
|
||||||
void unknow(int);
|
|
||||||
int ws_enable(dictionary);
|
#endif
|
||||||
|
Loading…
Reference in New Issue
Block a user