dany/antd-lua-plugin
1
0
Fork 0
mirror of https://github.com/lxsang/antd-lua-plugin synced 2025-07-13 05:34:25 +02:00
Code Issues Projects Releases Wiki Activity

Use unix domain socket to process lua handlers

Browse Source
This commit is contained in:
DanyLE
2022-08-27 16:38:20 +02:00
committed by Dany LE
parent 4259ec7eef
commit 7814611008
9 changed files with 591 additions and 819 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/asl/Makefile.am
View File

@ -19,10 +19,6 @@ json_la_LDFLAGS = -module -avoid-version -shared
json_la_SOURCES = 3rd/jsmn/jsmn.c json.c
lib_LTLIBRARIES += wurl.la
wurl_la_LDFLAGS = -module -avoid-version -shared
wurl_la_SOURCES = wurl.c
if HAS_FFI
lib_LTLIBRARIES += ffi.la
ffi_la_LDFLAGS = -module -avoid-version -shared -lffi

27
lib/asl/antd.c
View File

@ -255,6 +255,32 @@ static int l_t (lua_State *L) {
return 1; /* number of results */
}
static int l_antd_recv (lua_State *L) {
void* client = lua_touserdata (L, 1);
int len = luaL_checknumber(L,2);
if(len == 0)
{
lua_pushnil(L);
return 1;
}
if(len == 1)
{
char c = 0;
len = antd_recv(client, &c, len);
lua_pushnumber(L, c);
return 1;
}
else
{
lua_new_byte_array(L,len);
byte_array_t * arr = l_check_barray(L,-1);
len = antd_recv(client, arr->data, len);
lua_pushnumber(L,len);
return 2; /* number of results */
}
}
// TODO: add __b to LUA
//int __b(int, const unsigned char*, int);
static int l_b (lua_State *L) {
@ -697,6 +723,7 @@ static const struct luaL_Reg standard [] = {
//{"_text", l_text},
//{"_json", l_json},
//{"_jpeg", l_jpeg},
{"antd_recv", l_antd_recv},
{"_error", l_std_error},
{"_send_header", l_send_header},
{"b64encode", l_base64_encode},

103
lib/asl/handle.c
View File

@ -1,49 +1,34 @@
#include <antd/plugin.h>
#include <antd/scheduler.h>
#include <antd/dbhelper.h>
#include <sys/stat.h>
#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
#include <netinet/tcp.h>
#include <errno.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <netinet/in.h>
#include "../lualib.h"
/**
typedef struct {
plugin_header_t* __plugin__;
int fd;
} lua_thread_data_t;
* convert antd dictionary to lua table
*
*/
static void push_dict_to_lua(lua_State* L, dictionary_t d)
void* lua_handle(void* ptr)
{
lua_newtable(L);
chain_t as;
if(d)
for_each_assoc(as, d)
{
lua_pushstring(L,as->key);
//printf("KEY %s\n", as->key);
if(EQU(as->key,"COOKIE") || EQU(as->key,"REQUEST_HEADER") || EQU(as->key,"REQUEST_DATA") )
push_dict_to_lua(L, (dictionary_t)as->value);
else
{
if(strncmp(as->key,"octet-stream",12) == 0)
{
lua_pushlightuserdata(L, as->value);
}
else
{
lua_pushstring(L,as->value);
}
//printf("VALUE : %s\n",as->value );
}
lua_settable(L, -3);
}
}
void* lua_handle(void* data, void* meta)
{
antd_request_t* rq = (antd_request_t*) data;
char buf[BUFFLEN];
plugin_header_t* __plugin__ = meta;
lua_thread_data_t* data = (lua_thread_data_t**)ptr;
lua_State* L = NULL;
//char * index = __s("%s/%s",__plugin__.htdocs,"router.lua");
char* cnf = __s("%s%s%s", __plugin__->pdir,DIR_SEP, __plugin__->name);
antd_client_t cl = {0};
cl.sock = data->fd;
time(&cl.last_io);
cl.ssl = NULL;
cl.state = ANTD_CLIENT_PLUGIN_EXEC;
cl.z_status = 0;
cl.z_level = ANTD_CNONE;
cl.zstream = NULL;
//char * index = __s("%s/%s",__plugin__->htdocs,"router.lua");
char* cnf = __s("%s%s%s", data->__plugin__->pdir,DIR_SEP, data->__plugin__->name);
char * apis = __s("%s/%s",cnf,"api.lua");
L = luaL_newstate();
luaL_openlibs(L);
@ -54,25 +39,24 @@ void* lua_handle(void* data, void* meta)
// API header
lua_newtable(L);
lua_pushstring(L,"name");
lua_pushstring(L, __plugin__->name);
lua_pushstring(L, data->__plugin__->name);
lua_settable(L,-3);
lua_pushstring(L,"root");
htdocs(rq, buf);
lua_pushstring(L, buf);
lua_settable(L,-3);
//lua_pushstring(L,"root");
//htdocs(rq, buf);
//lua_pushstring(L, data->__plugin__->htdocs);
//lua_settable(L,-3);
lua_pushstring(L,"apiroot");
lua_pushstring(L, cnf);
lua_settable(L,-3);
lua_pushstring(L,"tmpdir");
tmpdir(buf);
lua_pushstring(L, buf);
lua_pushstring(L, data->__plugin__->tmpdir);
lua_settable(L,-3);
lua_pushstring(L,"dbpath");
lua_pushstring(L, __plugin__->dbpath);
lua_pushstring(L, data->__plugin__->dbpath);
lua_settable(L,-3);
lua_setglobal(L, "__api__");
@ -80,20 +64,32 @@ void* lua_handle(void* data, void* meta)
// Request
lua_newtable(L);
lua_pushstring(L,"id");
lua_pushlightuserdata(L, rq->client);
lua_pushlightuserdata(L, &cl);
//lua_pushnumber(L,client);
lua_settable(L, -3);
lua_pushstring(L,"request");
push_dict_to_lua(L,rq->request);
lua_pushstring(L,"socket");
lua_pushnumber(L, cl.sock);
//lua_pushnumber(L,client);
lua_settable(L, -3);
int flag = 1;
if (setsockopt(cl.sock, IPPROTO_TCP, TCP_NODELAY, &(int){1}, sizeof(int)) == -1)
{
ERROR("Unable to set TCP_NODELAY on %d - setsockopt: %s", cl.sock, strerror(errno));
}
//lua_pushstring(L,"request");
//push_dict_to_lua(L,rq->request);
//lua_settable(L, -3);
lua_setglobal(L, "HTTP_REQUEST");
free(ptr);
// load major apis
if(is_file(apis))
if (luaL_loadfile(L, apis) || lua_pcall(L, 0, 0, 0))
{
ERROR( "cannot start API file: [%s] %s\n", apis, lua_tostring(L, -1));
antd_error(&cl, 503, "Internal server error");
}
/*if (luaL_loadfile(L, index) || lua_pcall(L, 0, 0, 0))
@ -102,7 +98,7 @@ void* lua_handle(void* data, void* meta)
__t(client, "Cannot run router: %s", lua_tostring(L, -1));
}
free(index);*/
LOG("LUA handle exit on %d", rq->client->sock);
LOG("LUA handle exit on %d", cl.sock);
// clear request
if(L)
lua_close(L);
@ -110,6 +106,7 @@ void* lua_handle(void* data, void* meta)
free(cnf);
if(apis)
free(apis);
return antd_create_task(NULL, (void*)rq, NULL,rq->client->last_io);
(void) close(cl.sock);
return 0;
//lua_close(L);
}

80
lib/asl/ulib.c
View File

@ -381,43 +381,79 @@ static int l_file_move(lua_State* L)
static int l_send_file(lua_State* L)
{
const char* old = luaL_checkstring(L,1);
const char* new = luaL_checkstring(L,2);
int fromfd, tofd, ret;
size_t sz = 0;
char* old = NULL;
char* new = NULL;
if(lua_isnumber(L,1))
{
fromfd = (int)luaL_checknumber(L,1);
}
else
{
old = (char*)luaL_checkstring(L,1);
if((fromfd = open(old, O_RDONLY)) < 0)
{
lua_pushboolean(L,0);
goto end_send_file;
}
struct stat st;
if(stat(old, &st)!=0)
{
lua_pushboolean(L,0);
goto end_send_file;
}
sz = st.st_size;
}
if(lua_isnumber(L,2))
{
tofd = (int) luaL_checknumber(L,2);
}
else
{
new = (char*)luaL_checkstring(L,2);
if (unlink(new) < 0 && errno != ENOENT) {
lua_pushboolean(L,0);
goto end_send_file;
}
if((tofd = open(new, O_WRONLY | O_CREAT, 0600)) < 0)
{
lua_pushboolean(L,0);
goto end_send_file;
}
}
if(lua_isnumber(L,3))
{
sz = (size_t) luaL_checknumber(L,3);
}
off_t off = 0;
if (unlink(new) < 0 && errno != ENOENT) {
lua_pushboolean(L,0);
goto end_send_file;
}
if ((fromfd = open(old, O_RDONLY)) < 0 ||
(tofd = open(new, O_WRONLY | O_CREAT, 0600)) < 0) {
lua_pushboolean(L,0);
goto end_send_file;
}
struct stat st;
if(stat(old, &st)!=0)
{
lua_pushboolean(L,0);
goto end_send_file;
}
size_t sz = st.st_size;
int read = 0;
while (read != sz && (ret = sendfile(tofd, fromfd, &off, sz - read)) == 0)
while (
sz > 0 &&
read != sz &&
(
((ret = sendfile(tofd, fromfd, &off, sz - read)) >= 0) ||
(errno == EAGAIN)
)
)
{
if(ret < 0) ret = 0;
read += ret;
}
if(ret != sz)
if(read != sz)
{
lua_pushboolean(L,0);
goto end_send_file;
}
lua_pushboolean(L,1);
end_send_file:
if(fromfd >= 0)
if(fromfd >= 0 && old)
{
(void) close(fromfd);
}
if(tofd >= 0)
if(tofd >= 0 && new)
{
(void) close(tofd);
}

633
lib/asl/wurl.c
View File

@ -1,633 +0,0 @@
#include<stdio.h> //printf
#include<string.h> //memset
#include<stdlib.h> //for exit(0);
#include<errno.h> //For errno - the error number
#include<netdb.h> //hostent
#include <netinet/in.h>
#include <arpa/inet.h>
//#include <signal.h>
#include <dirent.h>
#include <errno.h>
#include <sys/socket.h>
#include <resolv.h>
#include <errno.h>
#include <ifaddrs.h>
#include <sys/stat.h>
#include <libgen.h>
#include "../lualib.h"
//#include "../../lua-api.h"
#include <antd/utils.h>
#include <sys/time.h>
#define CLIENT_NAME "wurl"
#define CLIENT_HOST "192.168.9.249"
#define CONN_TIME_OUT_S 3
#define MAX_BUFF 1024
#define REQUEST_BOUNDARY "------wURLFormBoundaryVo4QYVaSVseFNpeK"
#define GET 0
#define POST 1
typedef struct{
int type; // POST(1) or GET(0)
char* resource; // path
char* ctype; // content type, used by POST
int clen; // content length, used by POST
unsigned char* data ;
} wurl_header_t;
byte_array_t *l_check_barray (lua_State *L,int idx) {
void *ud = luaL_checkudata(L, idx, BYTEARRAY);
luaL_argcheck(L, ud != NULL, idx, "`byte array' expected");
return (byte_array_t *)ud;
}
void lua_new_byte_array(lua_State*L, int n)
{
size_t nbytes = sizeof(byte_array_t) + (n-1)*sizeof(unsigned char);
byte_array_t *a = (byte_array_t *)lua_newuserdata(L, nbytes);
luaL_getmetatable(L, BYTEARRAY);
lua_setmetatable(L, -2);
a->size = n;
}
/*get the ip by hostname*/
int wurl_ip_from_hostname(const char * hostname , char* ip)
{
struct hostent *he;
struct in_addr **addr_list;
int i;
if ( (he = gethostbyname( hostname ) ) == NULL)
{
// get the host info
herror("gethostbyname");
return -1;
}
addr_list = (struct in_addr **) he->h_addr_list;
for(i = 0; addr_list[i] != NULL; i++)
{
//Return the first one;
strcpy(ip , inet_ntoa(*addr_list[i]) );
return 0;
}
return -1;
}
/*
send a request
*/
int wurl_request_socket(const char* ip, int port)
{
int sockfd;
struct sockaddr_in dest;
// time out setting
struct timeval timeout;
timeout.tv_sec = CONN_TIME_OUT_S;
timeout.tv_usec = 0;
if ( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0 )
{
perror("Socket");
return -1;
}
if (setsockopt (sockfd, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout,sizeof(timeout)) < 0)
perror("setsockopt failed\n");
if (setsockopt (sockfd, SOL_SOCKET, SO_SNDTIMEO, (char *)&timeout,sizeof(timeout)) < 0)
perror("setsockopt failed\n");
bzero(&dest, sizeof(dest));
dest.sin_family = AF_INET;
dest.sin_port = htons(port);
if ( inet_aton(ip, &dest.sin_addr) == 0 )
{
perror(ip);
close(sockfd);
return -1;
}
if ( connect(sockfd, (struct sockaddr*)&dest, sizeof(dest)) != 0 )
{
close(sockfd);
perror("Connect");
return -1;
}
return sockfd;
}
/*
POST %s HTTP/1.0\r\n
Host: %s\r\n
User-Agent: %s\r\n
Content-Type: %s\r\n
Content-Length: %d\r\n\r\n"
maybe cookie support? this can cause security problem
GET %s HTTP/1.0\r\n
Host: %s\r\n
User-Agent: %s\r\n\r\n"
multipart
POST %s HTTP/1.1
Host: %s
User-Agent: %s
Content-Type: multipart/form-data; boundary=----------287032381131322
Content-Length: %d
------------287032381131322
Content-Disposition: form-data; name="datafile1"; filename="r.gif"
Content-Type: image/gif
GIF87a.............,...........D..;
------------287032381131322
Content-Disposition: form-data; name="datafile2"; filename="g.gif"
Content-Type: image/gif
GIF87a.............,...........D..;
------------287032381131322
Content-Disposition: form-data; name="datafile3"; filename="b.gif"
Content-Type: image/gif
GIF87a.............,...........D..;
------------287032381131322--
*/
int wurl_header(int sockfd, wurl_header_t rq)
{
char buff[MAX_BUFF];
if(sockfd < 0) return -1;
if(rq.type == GET) // GET
{
send(sockfd,"GET ",4,0);
send(sockfd,rq.resource, strlen(rq.resource),0);
send(sockfd," HTTP/1.0\r\n",11,0);
}
else
{
send(sockfd,"POST ",5, 0);
send(sockfd,rq.resource, strlen(rq.resource),0);
send(sockfd," HTTP/1.0\r\n",11,0);
sprintf(buff,"Content-Type: %s\r\n", rq.ctype);
send(sockfd,buff,strlen(buff),0);
sprintf(buff,"Content-Length: %d\r\n", rq.clen);
send(sockfd,buff,strlen(buff),0);
}
// host dont need to send the host
//sprintf(buff,"Host: %s\r\n",CLIENT_HOST);
//send(sockfd,buff,strlen(buff),0);
// user agent
sprintf(buff,"User-Agent: %s\r\n",CLIENT_NAME);
send(sockfd,buff,strlen(buff),0);
// terminate request
send(sockfd,"\r\n",2,0);
// if there is data, send out
if(rq.type == POST && rq.data)
{
send(sockfd,rq.data,rq.clen,0);
}
return 0;
}
/*
send multiple files to server
using multipart/form-data
*/
void wurl_send_files(int sockfd,char* resource, int n, char* name [], char* files[])
{
char buff[MAX_BUFF];
wurl_header_t rq;
rq.type = POST;
rq.resource = resource;
// get the total size of data
int totalsize = 0;
FILE* fd;
struct stat st;
for(int i = 0; i < n; ++i)
{
if(stat(files[i], &st) != 0) continue;
totalsize += st.st_size;
}
rq.clen = totalsize;
sprintf(buff,"%s; boundary=%s","multipart/form-data",REQUEST_BOUNDARY);
rq.ctype = buff;
// now send the header
wurl_header(sockfd,rq);
// now send the files
size_t size;
for(int i = 0; i < n; ++i)
{
fd = fopen(files[i],"rb");
if(!fd) continue;
// first send the boundary
//printf("sending file %s name:%s\n", files[i],name[i]);
sprintf(buff,"%s\r\n",REQUEST_BOUNDARY);
send(sockfd, buff, strlen(buff),0);
// content disposition
sprintf(buff,"Content-Disposition: form-data; name=\"%s\"; filename=\"%s\"\r\n",
name[i],basename(files[i]));
send(sockfd,buff, strlen(buff),0);
// content type
sprintf(buff,"Content-Type: %s\r\n\r\n",mime(files[i]));
send(sockfd,buff, strlen(buff),0);
// now send the file
while(!feof(fd))
{
size = fread(buff,1,MAX_BUFF,fd);
send(sockfd,buff,size,0);
//if(!__b(client,buffer,size)) return;
}
fclose(fd);
send(sockfd,"\r\n",2,0);
}
//end the boudary
sprintf(buff,"%s--\r\n",REQUEST_BOUNDARY);
send(sockfd,buff,strlen(buff),0);
}
/**
* 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 wurl_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;
}
/*
POST example
wurl_header_t rq;
rq.resource = path;
rq.type = POST;
rq.data = "s=a&q=b#test";//"{\"name\":\"sang\"}";
rq.clen = strlen(rq.data);
rq.ctype = "application/x-www-form-urlencoded";//"application/json";
if(wurl_request(hostname,port,&rq,1) == 0)
{
printf(rq.data);
}
else
{
printf("Cannot connect to host\n");
}
DOWNLOAD
wurl_header_t rq;
rq.resource = path;
rq.type = GET;
if(wurl_download(hostname,port,&rq,file) == 0)
{
printf("Download sucess ful\n");
}
else
{
printf("Cannot connect to host\n");
}
upload example
// send files
char * names[2];
names[0] = "zip";
names[1] = "text";
char* files[2];
files[0] = "/Users/mrsang/tmp/Archive.zip";
files[1] = "/Users/mrsang/tmp/test.py";
wurl_send_files(sock,path,2,names, files);
printf("RETURN:\n");
size = wurl_read_buf(sock,buff, MAX_BUFF);
while(size > 0)
{
printf("%s", buff);
size = wurl_read_buf(sock,buff, MAX_BUFF);
}
close(sock);
*/
void wurl_response_header(int sock, wurl_header_t* header)
{
char buff[MAX_BUFF];
int size = wurl_read_buf(sock,buff,MAX_BUFF);
char* token;
while (size > 0 && strcmp("\r\n",buff))
{
char* line = strdup(buff);
//printf("LINE %s\n", line);
token = strsep(&line,":");
trim(token,' ');
if(token != NULL &&strcasecmp(token,"Content-Type") == 0)
{
header->ctype = strsep(&line,":");
trim(header->ctype,' ');
trim(header->ctype,'\n');
trim(header->ctype,'\r');
} else if(token != NULL &&strcasecmp(token,"Content-Length") == 0)
{
token = strsep(&line,":");
trim(token,' ');
header->clen = atoi(token);
}
//if(line) free(line);
size = wurl_read_buf(sock,buff,MAX_BUFF);
}
}
int wurl_read_data(int sock,char** din)
{
// read line by line, ignore content length
int total_length = 0;
char* tmp = NULL;
int CHUNK = 512;
char buff[MAX_BUFF];
char * data = ( char*) malloc(CHUNK);
int cursize = CHUNK;
int size = wurl_read_buf(sock,buff,MAX_BUFF);
while(size > 0)
{
if(total_length+size > cursize)
{
tmp = (char*) realloc(data,total_length + size+ CHUNK );
if(!tmp)
{
if(data) free(data);
break;
}
cursize = total_length + size+ CHUNK;
data = tmp;
}
memcpy(data+total_length,buff,size);
total_length += size;
size = wurl_read_buf(sock,buff,MAX_BUFF);
}
data[total_length] = '\0';
close(sock);
*din = data;
return total_length;
}
/*
hostname
port
header for request and respond
lazy : if 1, all data is read to the header
if 0, user has the responsibility to handler it
*/
int wurl_request(const char* hostname, int port, wurl_header_t* header, int lazy)
{
char ip[100];
wurl_ip_from_hostname(hostname ,ip);
int sock = wurl_request_socket(ip, port);
if(sock <= 0) return -1;
// send header
wurl_header(sock,*header);
// read respond header
wurl_response_header(sock,header);
if(header->ctype == NULL || header->clen == -1)
{
LOG("Bad data\n");
return -1;
}
// read data if lazy
if(lazy)
{
// read line by line, ignore content length
header->clen = wurl_read_data(sock,(char**)&header->data);
return 0;
}
return sock;
}
int wurl_download(const char* hostname, int port, wurl_header_t* h, const char* to)
{
// we will handler the data reading
int sock = wurl_request(hostname, port,h,0);
char buff[MAX_BUFF];
if(sock < 0) return -1;
FILE* fp = fopen(to,"wb");
int size;
if(fp)
{
while((size = wurl_read_buf(sock,buff, MAX_BUFF)) > 0)
{
fwrite(buff, size, 1, fp);
}
fclose(fp);
}
else
{
close(sock);
return -1;
}
close(sock);
return 0;
}
/*
int main (int argc, char const *argv[])
{
if(argc < 4)
{
printf("wurl [host] [port] [path]\n");
exit(1);
}
char *hostname = argv[1];
char* path = argv[3];
int port = atoi(argv[2]);
char*file = argv[4];
wurl_header_t rq;
rq.resource = path;
rq.type = POST;
rq.data = "s=a&q=b#test";//"{\"name\":\"sang\"}";
rq.clen = strlen(rq.data);
rq.ctype = "application/x-www-form-urlencoded";//"application/json";
if(wurl_download(hostname,port,&rq,file) == 0)
{
printf("Download sucess ful\n");
}
else
{
printf("Cannot connect to host\n");
}
return 0;
}*/
static int l_get(lua_State *L)
{
const char* host = luaL_checkstring(L,1);
int port = luaL_checknumber(L,2);
const char* resource = luaL_checkstring(L,3);
wurl_header_t rq;
rq.resource = (char*)resource;
rq.type = GET;
if(wurl_request(host,port,&rq,1) == 0)
{
//printf("Content type is %s\n", rq.ctype);
//mime_t m = mime_from_type(rq.ctype);
lua_newtable(L);
lua_pushstring(L,"contentType");
lua_pushstring(L,rq.ctype);
lua_settable(L,-3);
/*
lua_pushstring(L,"binary");
lua_pushboolean(L,m.bin);
lua_settable(L,-3);*/
lua_pushstring(L,"data");
// byte array, that can be convert to string later
lua_new_byte_array(L,rq.clen);
byte_array_t* arr = l_check_barray(L,-1);
memcpy(arr->data,rq.data,rq.clen);
free(rq.data); // be careful
lua_settable(L,-3);
return 1;
}
else
{
lua_pushnil(L);
return 1;
}
}
static int l_post(lua_State *L)
{
const char* host = luaL_checkstring(L,1); // host
int port = luaL_checknumber(L,2); // port
const char* res = luaL_checkstring(L,3); // resource
const char* ctype = luaL_checkstring(L,4); // content type
const char* data = luaL_checkstring(L,5); // post data
wurl_header_t rq;
rq.resource = (char*)res;
rq.type = POST;
rq.ctype = (char*)ctype;
rq.clen = strlen(data);
rq.data = (unsigned char*)data;
if(wurl_request(host,port,&rq,1) == 0)
{
//printf("Content type is %s\n", rq.ctype);
mime_t m = mime_from_type(rq.ctype);
lua_newtable(L);
lua_pushstring(L,"contentType");
lua_pushstring(L,rq.ctype);
lua_settable(L,-3);
/*lua_pushstring(L,"binary");
lua_pushboolean(L,m.bin);
lua_settable(L,-3);
*/
lua_pushstring(L,"data");
lua_new_byte_array(L,rq.clen);
byte_array_t* arr = l_check_barray(L,-1);
memcpy(arr->data,rq.data,rq.clen);
free(rq.data); // be careful
lua_settable(L,-3);
return 1;
}
else
{
lua_pushnil(L);
return 1;
}
}
static int l_download(lua_State* L)
{
// we user only GET for down load and POST for upload
const char* host = luaL_checkstring(L,1); // host
int port = luaL_checknumber(L,2); // port
const char* res = luaL_checkstring(L,3); // resource
const char* file = luaL_checkstring(L,4); // file
wurl_header_t rq;
rq.resource = (char*)res;
rq.type = GET;
if(wurl_download(host,port,&rq,file) == 0)
lua_pushboolean(L, true);
else
lua_pushboolean(L,false);
return 1;
}
static int l_upload(lua_State* L)
{
const char* host = luaL_checkstring(L,1);
int port = luaL_checknumber(L,2);
const char* resource = luaL_checkstring(L,3);
const char* name = luaL_checkstring(L,4);
const char* file = luaL_checkstring(L,5);
if(!_exist(file)) goto fail;
// request socket
char ip[100];
wurl_ip_from_hostname(host ,ip);
int sock = wurl_request_socket(ip, port);
if(sock <= 0) goto fail;
char * names[2];
names[0] = (char*)name;
char* files[2];
//files[0] = "/Users/mrsang/tmp/Archive.zip";
files[0] = (char*)file;
//printf("SENDIND DILE\n");
wurl_send_files(sock, (char*)resource,1,names,files);
wurl_header_t header;
//printf("READ HEADER\n");
wurl_response_header(sock, &header);
//printf("read data\n");
wurl_read_data(sock,(char**)&header.data);
if(header.ctype != NULL && header.data)
{
lua_pushstring(L,(const char*)header.data);
free(header.data);
return 1;
}
fail:
lua_pushnil(L);
return 1;
}
static const struct luaL_Reg _lib [] = {
{"_get",l_get},
{"_post",l_post},
{"_download",l_download},
{"_upload",l_upload},
{NULL,NULL}
};
int luaopen_wurl(lua_State *L)
{
luaL_newlib(L, _lib);
return 1;
}