sq_libcurl.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780
  1. #ifdef SQ_USE_EASYCURL
  2. #ifdef __cplusplus
  3. extern "C" {
  4. #endif
  5. #include "squirrel.h"
  6. #include "sqstdblobimpl.h"
  7. #include <string.h>
  8. #include <stdio.h>
  9. #include <stdlib.h> /* for malloc */
  10. #include <assert.h> /* for a few sanity tests */
  11. #include "curl/curl.h"
  12. ////////
  13. #include "dynamic_library.h"
  14. /*SquiLu
  15. local library_functions = [
  16. ["CURL *", "curl_easy_init", "void"],
  17. ["CURLcode", "curl_easy_setopt", "CURL *curl, CURLoption option, ..."],
  18. ["CURLcode", "curl_easy_getinfo", "CURL *curl, CURLINFO info, ..."],
  19. ["CURLcode", "curl_easy_perform", "CURL *curl"],
  20. ["CURLcode", "curl_easy_pause", "CURL *handle, int bitmask"],
  21. ["CURL *", "curl_easy_duphandle", "CURL *curl"],
  22. ["void", "curl_easy_reset", "CURL *curl"],
  23. ["CURLcode", "curl_easy_recv", "CURL *curl, void *buffer, size_t buflen, size_t *n"],
  24. ["CURLcode", "curl_easy_send", "CURL *curl, const void *buffer, size_t buflen, size_t *n"],
  25. ["const char *", "curl_easy_escape", "CURL *handle, const char *string, int length"],
  26. ["const char *", "curl_easy_unescape", "CURL *handle, const char *string, int length"],
  27. ["const char *", "curl_easy_strerror", "CURLcode errornum"],
  28. ["struct curl_slist *", "curl_slist_append", "struct curl_slist *, const char *"],
  29. ["void", "curl_slist_free_all", "struct curl_slist *"],
  30. //next entry should be the last one
  31. //to make valid the test made on load_library function
  32. ["void", "curl_easy_cleanup", "CURL *curl"],
  33. ];
  34. function write_library_functions_declaration(){
  35. foreach(k,v in library_functions) {
  36. putsnl("typedef " + v[0] + " (*" + v[1] + "_t)(" + v[2] + ");");
  37. putsnl("static " + v[1] + "_t dl" + v[1] + " = 0;");
  38. }
  39. }
  40. function write_library_functions_load(){
  41. foreach(k,v in library_functions){
  42. putsnl("dl" + v[1] + " = (" + v[1] + "_t) libdyn.dlsym(\"" + v[1] + "\");");
  43. putsnl("if(!dl" + v[1] + ") return false;");
  44. }
  45. }
  46. SquiLu*/
  47. static DynamicLibrary libdyn;
  48. //@write_library_functions_declaration();
  49. // generated-code:begin
  50. typedef CURL * (*curl_easy_init_t)(void);
  51. static curl_easy_init_t dlcurl_easy_init = 0;
  52. typedef CURLcode (*curl_easy_setopt_t)(CURL *curl, CURLoption option, ...);
  53. static curl_easy_setopt_t dlcurl_easy_setopt = 0;
  54. typedef CURLcode (*curl_easy_getinfo_t)(CURL *curl, CURLINFO info, ...);
  55. static curl_easy_getinfo_t dlcurl_easy_getinfo = 0;
  56. typedef CURLcode (*curl_easy_perform_t)(CURL *curl);
  57. static curl_easy_perform_t dlcurl_easy_perform = 0;
  58. typedef CURLcode (*curl_easy_pause_t)(CURL *handle, int bitmask);
  59. static curl_easy_pause_t dlcurl_easy_pause = 0;
  60. typedef CURL * (*curl_easy_duphandle_t)(CURL *curl);
  61. static curl_easy_duphandle_t dlcurl_easy_duphandle = 0;
  62. typedef void (*curl_easy_reset_t)(CURL *curl);
  63. static curl_easy_reset_t dlcurl_easy_reset = 0;
  64. typedef CURLcode (*curl_easy_recv_t)(CURL *curl, void *buffer, size_t buflen, size_t *n);
  65. static curl_easy_recv_t dlcurl_easy_recv = 0;
  66. typedef CURLcode (*curl_easy_send_t)(CURL *curl, const void *buffer, size_t buflen, size_t *n);
  67. static curl_easy_send_t dlcurl_easy_send = 0;
  68. typedef const char * (*curl_easy_escape_t)(CURL *handle, const char *string, int length);
  69. static curl_easy_escape_t dlcurl_easy_escape = 0;
  70. typedef const char * (*curl_easy_unescape_t)(CURL *handle, const char *string, int length);
  71. static curl_easy_unescape_t dlcurl_easy_unescape = 0;
  72. typedef const char * (*curl_easy_strerror_t)(CURLcode errornum);
  73. static curl_easy_strerror_t dlcurl_easy_strerror = 0;
  74. typedef struct curl_slist * (*curl_slist_append_t)(struct curl_slist *, const char *);
  75. static curl_slist_append_t dlcurl_slist_append = 0;
  76. typedef void (*curl_slist_free_all_t)(struct curl_slist *);
  77. static curl_slist_free_all_t dlcurl_slist_free_all = 0;
  78. typedef void (*curl_easy_cleanup_t)(CURL *curl);
  79. static curl_easy_cleanup_t dlcurl_easy_cleanup = 0;
  80. // generated-code:end
  81. static const char *dynamicLibName = DYNLIB_FOR_OS(libcurl);
  82. static bool load_library(const char *libname)
  83. {
  84. if(dlcurl_easy_cleanup) return true;
  85. if(libdyn.open(libname))
  86. {
  87. //@write_library_functions_load();
  88. // generated-code:begin
  89. dlcurl_easy_init = (curl_easy_init_t) libdyn.dlsym("curl_easy_init");
  90. if(!dlcurl_easy_init) return false;
  91. dlcurl_easy_setopt = (curl_easy_setopt_t) libdyn.dlsym("curl_easy_setopt");
  92. if(!dlcurl_easy_setopt) return false;
  93. dlcurl_easy_getinfo = (curl_easy_getinfo_t) libdyn.dlsym("curl_easy_getinfo");
  94. if(!dlcurl_easy_getinfo) return false;
  95. dlcurl_easy_perform = (curl_easy_perform_t) libdyn.dlsym("curl_easy_perform");
  96. if(!dlcurl_easy_perform) return false;
  97. dlcurl_easy_pause = (curl_easy_pause_t) libdyn.dlsym("curl_easy_pause");
  98. if(!dlcurl_easy_pause) return false;
  99. dlcurl_easy_duphandle = (curl_easy_duphandle_t) libdyn.dlsym("curl_easy_duphandle");
  100. if(!dlcurl_easy_duphandle) return false;
  101. dlcurl_easy_reset = (curl_easy_reset_t) libdyn.dlsym("curl_easy_reset");
  102. if(!dlcurl_easy_reset) return false;
  103. dlcurl_easy_recv = (curl_easy_recv_t) libdyn.dlsym("curl_easy_recv");
  104. if(!dlcurl_easy_recv) return false;
  105. dlcurl_easy_send = (curl_easy_send_t) libdyn.dlsym("curl_easy_send");
  106. if(!dlcurl_easy_send) return false;
  107. dlcurl_easy_escape = (curl_easy_escape_t) libdyn.dlsym("curl_easy_escape");
  108. if(!dlcurl_easy_escape) return false;
  109. dlcurl_easy_unescape = (curl_easy_unescape_t) libdyn.dlsym("curl_easy_unescape");
  110. if(!dlcurl_easy_unescape) return false;
  111. dlcurl_easy_strerror = (curl_easy_strerror_t) libdyn.dlsym("curl_easy_strerror");
  112. if(!dlcurl_easy_strerror) return false;
  113. dlcurl_slist_append = (curl_slist_append_t) libdyn.dlsym("curl_slist_append");
  114. if(!dlcurl_slist_append) return false;
  115. dlcurl_slist_free_all = (curl_slist_free_all_t) libdyn.dlsym("curl_slist_free_all");
  116. if(!dlcurl_slist_free_all) return false;
  117. dlcurl_easy_cleanup = (curl_easy_cleanup_t) libdyn.dlsym("curl_easy_cleanup");
  118. if(!dlcurl_easy_cleanup) return false;
  119. // generated-code:end
  120. return true;
  121. }
  122. return false;
  123. }
  124. ////////
  125. struct sq_easycurl_st
  126. {
  127. CURL *curl;
  128. HSQUIRRELVM vm;
  129. HSQOBJECT reader_cb;
  130. HSQOBJECT reader_cb_udata;
  131. HSQOBJECT writer_cb;
  132. HSQOBJECT writer_cb_udata;
  133. HSQOBJECT slist_array;
  134. struct curl_slist *slist; //only one linked list
  135. };
  136. static const SQChar SQ_LIBNAME[] = _SC("EasyCurl");
  137. //SQ_OPT_STRING_STRLEN();
  138. static const SQChar EasyCurl_Tag[] = _SC("EasyCurl_TAG");
  139. #define GET_EasyCurl_INSTANCE() SQ_GET_INSTANCE(v, 1, sq_easycurl_st, EasyCurl_Tag) \
  140. if(self == NULL) return sq_throwerror(v, _SC("EasyCurl object already closed"));
  141. /* Auxiliary function that waits on the socket. */
  142. static int wait_on_socket(curl_socket_t sockfd, int for_recv, long timeout_ms)
  143. {
  144. struct timeval tv;
  145. fd_set infd, outfd, errfd;
  146. int res;
  147. tv.tv_sec = timeout_ms / 1000;
  148. tv.tv_usec= (timeout_ms % 1000) * 1000;
  149. FD_ZERO(&infd);
  150. FD_ZERO(&outfd);
  151. FD_ZERO(&errfd);
  152. FD_SET(sockfd, &errfd); /* always check for error */
  153. if(for_recv) {
  154. FD_SET(sockfd, &infd);
  155. }
  156. else {
  157. FD_SET(sockfd, &outfd);
  158. }
  159. /* select() returns the number of signalled sockets or -1 */
  160. res = select(sockfd + 1, &infd, &outfd, &errfd, &tv);
  161. return res;
  162. }
  163. static SQRESULT EasyCurl_release_hook(SQUserPointer p, SQInteger size, void */*ep*/)
  164. {
  165. sq_easycurl_st *self = (sq_easycurl_st*)p;
  166. if(self)
  167. {
  168. dlcurl_easy_cleanup(self->curl);
  169. sq_release(self->vm, &self->reader_cb);
  170. sq_release(self->vm, &self->reader_cb_udata);
  171. sq_release(self->vm, &self->writer_cb);
  172. sq_release(self->vm, &self->writer_cb_udata);
  173. if(self->slist)
  174. {
  175. sq_release(self->vm, &self->slist_array);
  176. dlcurl_slist_free_all(self->slist);
  177. }
  178. sq_free(self, sizeof(*self));
  179. }
  180. return 0;
  181. }
  182. static SQRESULT EasyCurl_constructor(HSQUIRRELVM v, int idx, CURL *EasyCurl, int free_on_gc)
  183. {
  184. if(!EasyCurl)
  185. return sq_throwerror(v, _SC("Could'nt create an EasyCurl object."));
  186. sq_easycurl_st *self = (sq_easycurl_st*)sq_malloc(sizeof(*self));
  187. memset(self, 0, sizeof(*self));
  188. sq_resetobject(&self->reader_cb);
  189. sq_resetobject(&self->reader_cb_udata);
  190. sq_resetobject(&self->writer_cb);
  191. sq_resetobject(&self->writer_cb_udata);
  192. sq_resetobject(&self->slist_array);
  193. self->vm = v;
  194. self->curl = EasyCurl;
  195. SQInteger rc = sq_setinstanceup(v, idx, self);
  196. if(free_on_gc) sq_setreleasehook(v,-1, EasyCurl_release_hook);
  197. return rc;
  198. }
  199. static SQRESULT EasyCurl_create_instance(HSQUIRRELVM v, CURL *EasyCurl, int free_on_gc)
  200. {
  201. sq_pushstring(v, SQ_LIBNAME, -1);
  202. if(sq_getonroottable(v) == SQ_OK){
  203. sq_pushstring(v, EasyCurl_Tag, -1);
  204. if(sq_get(v, -2) == SQ_OK){
  205. if(sq_createinstance(v, -1) == SQ_OK){
  206. return EasyCurl_constructor(v, -1, EasyCurl, free_on_gc);
  207. }
  208. }
  209. }
  210. return SQ_ERROR;
  211. }
  212. static SQRESULT sq_EasyCurl_constructor(HSQUIRRELVM v){
  213. if(!load_library(dynamicLibName)) return sq_throwerror(v, _SC("Failed to load libpcre !"));
  214. CURL *curl = dlcurl_easy_init();
  215. return EasyCurl_constructor(v, 1, curl, 1);
  216. }
  217. static SQRESULT sq_EasyCurl_duphandle(HSQUIRRELVM v){
  218. SQ_FUNC_VARS_NO_TOP(v);
  219. GET_EasyCurl_INSTANCE();
  220. CURL *curl = dlcurl_easy_duphandle(self->curl);
  221. return EasyCurl_create_instance(v, curl, 1);
  222. }
  223. static SQRESULT sq_EasyCurl_getinfo(HSQUIRRELVM v){
  224. SQ_FUNC_VARS_NO_TOP(v);
  225. GET_EasyCurl_INSTANCE();
  226. CURLcode rc = CURL_LAST;
  227. SQ_GET_INTEGER(v, 2, option);
  228. SQ_GET_INTEGER(v, 3, option_type);
  229. switch(option_type)
  230. {
  231. case 's':
  232. const SQChar *str_value;
  233. rc = dlcurl_easy_getinfo(self->curl, (CURLINFO)option, &str_value);
  234. if(rc == CURLE_OK) sq_pushstring(v, str_value, -1);
  235. break;
  236. case 'i':
  237. long int_value;
  238. rc = dlcurl_easy_getinfo(self->curl, (CURLINFO)option, &int_value);
  239. if(rc == CURLE_OK) sq_pushinteger(v, int_value);
  240. break;
  241. case 'f':
  242. double float_value;
  243. rc = dlcurl_easy_getinfo(self->curl, (CURLINFO)option, &float_value);
  244. if(rc == CURLE_OK) sq_pushfloat(v, float_value);
  245. break;
  246. default:
  247. return sq_throwerror(v, _SC("invalid option value type"));
  248. }
  249. if(rc != CURLE_OK) sq_pushinteger(v, CURL_LAST);
  250. return 1;
  251. }
  252. static SQRESULT sq_EasyCurl_setopt(HSQUIRRELVM v){
  253. SQ_FUNC_VARS_NO_TOP(v);
  254. GET_EasyCurl_INSTANCE();
  255. CURLcode rc = CURLE_OK;
  256. SQ_GET_INTEGER(v, 2, option);
  257. switch(sq_gettype(v, 3))
  258. {
  259. case OT_STRING:
  260. SQ_GET_STRING(v, 3, str_value);
  261. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, str_value);
  262. break;
  263. case OT_INTEGER:
  264. SQ_GET_INTEGER(v, 3, int_value);
  265. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, int_value);
  266. break;
  267. case OT_FLOAT:
  268. SQ_GET_FLOAT(v, 3, float_value);
  269. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, float_value);
  270. break;
  271. case OT_ARRAY:
  272. {
  273. if(self->slist)
  274. {
  275. return sq_throwerror(v, _SC("single list already used, only one right now"));
  276. }
  277. sq_getstackobj(v, 3, &self->slist_array);
  278. sq_addref(v, &self->slist_array);
  279. const SQChar *str;
  280. SQInteger asize = sq_getsize(v, 3);
  281. for(int i=0; i<asize; ++i)
  282. {
  283. sq_pushinteger(v, i);
  284. sq_get(v, 3);
  285. sq_getstring(v, -1, &str);
  286. self->slist = dlcurl_slist_append(self->slist, str);
  287. sq_poptop(v);
  288. }
  289. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, self->slist);
  290. }
  291. break;
  292. default:
  293. return sq_throwerror(v, _SC("invalid option value type"));
  294. }
  295. sq_pushinteger(v, rc);
  296. return 1;
  297. }
  298. static SQRESULT sq_EasyCurl_recv(HSQUIRRELVM v){
  299. SQ_FUNC_VARS_NO_TOP(v);
  300. GET_EasyCurl_INSTANCE();
  301. SQ_GET_INTEGER(v, 2, try_receive_n);
  302. SQChar *buffer = sq_getscratchpad(v, try_receive_n);
  303. size_t recv_size = 0;
  304. CURLcode rc = dlcurl_easy_recv(self->curl, (void *)buffer, try_receive_n, &recv_size);
  305. if(rc != CURLE_OK) sq_pushinteger(v, -rc);
  306. else sq_pushinteger(v, recv_size);
  307. return 1;
  308. }
  309. static SQRESULT sq_EasyCurl_send(HSQUIRRELVM v){
  310. SQ_FUNC_VARS_NO_TOP(v);
  311. GET_EasyCurl_INSTANCE();
  312. SQ_GET_STRING(v, 2, buffer);
  313. size_t send_size = 0;
  314. CURLcode rc = dlcurl_easy_send(self->curl, (const void *)buffer, buffer_size, &send_size);
  315. if(rc != CURLE_OK) sq_pushinteger(v, -rc);
  316. else sq_pushinteger(v, send_size);
  317. return 1;
  318. }
  319. static SQRESULT sq_EasyCurl_reset(HSQUIRRELVM v){
  320. SQ_FUNC_VARS_NO_TOP(v);
  321. GET_EasyCurl_INSTANCE();
  322. dlcurl_easy_reset(self->curl);
  323. return 0;
  324. }
  325. static SQRESULT sq_EasyCurl_perform(HSQUIRRELVM v){
  326. SQ_FUNC_VARS_NO_TOP(v);
  327. GET_EasyCurl_INSTANCE();
  328. CURLcode rc = dlcurl_easy_perform(self->curl);
  329. sq_pushinteger(v, rc);
  330. return 1;
  331. }
  332. static SQRESULT sq_EasyCurl_pause(HSQUIRRELVM v){
  333. SQ_FUNC_VARS_NO_TOP(v);
  334. GET_EasyCurl_INSTANCE();
  335. SQ_GET_INTEGER(v, 2, flag_how);
  336. CURLcode rc = dlcurl_easy_pause(self->curl, flag_how);
  337. sq_pushinteger(v, rc);
  338. return 1;
  339. }
  340. static SQRESULT sq_EasyCurl_cleanup(HSQUIRRELVM v){
  341. SQ_FUNC_VARS_NO_TOP(v);
  342. GET_EasyCurl_INSTANCE();
  343. dlcurl_easy_cleanup(self->curl);
  344. sq_setinstanceup(v, 1, 0); //invalidate this curl handle
  345. return 0;
  346. }
  347. static SQRESULT sq_EasyCurl_strerror(HSQUIRRELVM v){
  348. SQ_FUNC_VARS_NO_TOP(v);
  349. GET_EasyCurl_INSTANCE();
  350. SQ_GET_INTEGER(v, 2, error_code);
  351. sq_pushstring(v, dlcurl_easy_strerror((CURLcode)error_code), -1);
  352. return 1;
  353. }
  354. static SQRESULT sq_EasyCurl_escape(HSQUIRRELVM v){
  355. SQ_FUNC_VARS_NO_TOP(v);
  356. GET_EasyCurl_INSTANCE();
  357. SQ_GET_STRING(v, 2, str);
  358. sq_pushstring(v, dlcurl_easy_escape(self->curl, str, str_size), -1);
  359. return 1;
  360. }
  361. static SQRESULT sq_EasyCurl_unescape(HSQUIRRELVM v){
  362. SQ_FUNC_VARS_NO_TOP(v);
  363. GET_EasyCurl_INSTANCE();
  364. SQ_GET_STRING(v, 2, str);
  365. sq_pushstring(v, dlcurl_easy_unescape(self->curl, str, str_size), -1);
  366. return 1;
  367. }
  368. static size_t sq_EasyCurl_reader_writer_callback(char *bufptr, size_t size, size_t nitems, void *userp, int isReader)
  369. {
  370. SQInteger result = 0; /* abort by default */
  371. sq_easycurl_st *self = (sq_easycurl_st*)userp;
  372. HSQUIRRELVM v = self->vm;
  373. int top = sq_gettop(v);
  374. HSQOBJECT &rw_cb = (isReader ? self->reader_cb : self->writer_cb);
  375. HSQOBJECT &rw_cb_udata = (isReader ? self->reader_cb_udata : self->writer_cb_udata);
  376. size_t data_size = size * nitems;
  377. sq_pushobject(v, rw_cb);
  378. SQObjectType otype = sq_gettype(v, -1);
  379. switch(otype)
  380. {
  381. case OT_INSTANCE:
  382. {
  383. SQBlob *blob = NULL;
  384. if(SQ_FAILED(sq_getinstanceup(v,-1,(SQUserPointer*)&blob,(SQUserPointer)SQBlob::SQBlob_TAG)))
  385. break;
  386. if(!blob || !blob->IsValid())
  387. break;
  388. blob->Write(bufptr, data_size);
  389. result = data_size;
  390. }
  391. break;
  392. case OT_CLOSURE:
  393. sq_pushroottable(v);
  394. if(!isReader)
  395. {
  396. sq_pushstring(v, (const SQChar*)bufptr, data_size);
  397. }
  398. sq_pushinteger(v, size);
  399. sq_pushinteger(v, nitems);
  400. sq_pushobject(v, rw_cb_udata);
  401. /* call squilu function */
  402. if (sq_call(v, isReader ? 4 : 5, SQTrue, SQFalse) == SQ_OK)
  403. {
  404. switch(sq_gettype(v, -1))
  405. {
  406. case OT_INTEGER:
  407. sq_getinteger(v, -1, &result);
  408. break;
  409. case OT_STRING:
  410. if(isReader)
  411. {
  412. const SQChar *str;
  413. if(sq_getstr_and_size(v,-1, &str, &result) == SQ_OK)
  414. {
  415. memcpy(bufptr, str, result);
  416. }
  417. }
  418. break;
  419. }
  420. }
  421. break;
  422. }
  423. sq_settop(v, top);
  424. return result;
  425. }
  426. static size_t sq_EasyCurl_reader_callback(char *bufptr, size_t size, size_t nitems, void *userp)
  427. {
  428. return sq_EasyCurl_reader_writer_callback(bufptr, size, nitems, userp, 1);
  429. }
  430. static size_t sq_EasyCurl_writer_callback(char *bufptr, size_t size, size_t nitems, void *userp)
  431. {
  432. return sq_EasyCurl_reader_writer_callback(bufptr, size, nitems, userp, 0);
  433. }
  434. static SQRESULT sq_EasyCurl_set_reader_writer(HSQUIRRELVM v, int isReader){
  435. SQ_FUNC_VARS(v);
  436. GET_EasyCurl_INSTANCE();
  437. CURLcode rc;
  438. HSQOBJECT &rw_cb = (isReader ? self->reader_cb : self->writer_cb);
  439. HSQOBJECT &rw_cb_udata = (isReader ? self->reader_cb_udata : self->writer_cb_udata);
  440. CURLoption rw_fnc_option = (isReader ? CURLOPT_READFUNCTION : CURLOPT_WRITEFUNCTION);
  441. /* clear reader handler */
  442. rc = dlcurl_easy_setopt(self->curl, rw_fnc_option, NULL);
  443. sq_release(v, &rw_cb);
  444. sq_release(v, &rw_cb_udata);
  445. sq_resetobject(&rw_cb);
  446. sq_resetobject(&rw_cb_udata);
  447. SQObjectType otype = sq_gettype(v, 2);
  448. if(_top_ > 1 && otype != OT_NULL)
  449. {
  450. switch(otype)
  451. {
  452. case OT_INSTANCE:
  453. {
  454. SQBlob *blob = NULL;
  455. if(SQ_FAILED(sq_getinstanceup(v,2,(SQUserPointer*)&blob,(SQUserPointer)SQBlob::SQBlob_TAG)))
  456. return sq_throwerror(v,_SC("expect a blob as second parameter"));
  457. if(!blob || !blob->IsValid())
  458. return sq_throwerror(v,_SC("the blob is invalid"));
  459. }
  460. break;
  461. case OT_CLOSURE:
  462. break;
  463. default:
  464. return sq_throwerror(v, _SC("invalid fisrt parameter expected closure/blob"));
  465. }
  466. sq_getstackobj(v, 2, &rw_cb);
  467. sq_addref(v, &rw_cb);
  468. if(_top_ > 2)
  469. {
  470. sq_getstackobj(v, 3, &rw_cb_udata);
  471. sq_addref(v, &rw_cb_udata);
  472. }
  473. /* set authorizer_hook callback */
  474. rc = dlcurl_easy_setopt(self->curl, rw_fnc_option,
  475. (isReader ? sq_EasyCurl_reader_callback : sq_EasyCurl_writer_callback));
  476. if(rc == CURLE_OK) rc = dlcurl_easy_setopt(self->curl,
  477. (isReader ? CURLOPT_READDATA : CURLOPT_WRITEDATA), self);
  478. }
  479. sq_pushinteger(v, rc);
  480. return 1;
  481. }
  482. static SQRESULT sq_EasyCurl_set_reader(HSQUIRRELVM v){
  483. return sq_EasyCurl_set_reader_writer(v, 1);
  484. }
  485. static SQRESULT sq_EasyCurl_set_writer(HSQUIRRELVM v){
  486. return sq_EasyCurl_set_reader_writer(v, 0);
  487. }
  488. static SQRESULT sq_EasyCurl_wait_on_socket(HSQUIRRELVM v){
  489. SQ_FUNC_VARS_NO_TOP(v);
  490. GET_EasyCurl_INSTANCE();
  491. SQ_GET_BOOL(v, 2, for_recv);
  492. SQ_GET_INTEGER(v, 3, timeout_ms);
  493. long sockextr;
  494. CURLcode rc = dlcurl_easy_getinfo(self->curl, CURLINFO_LASTSOCKET, &sockextr);
  495. if(rc == CURLE_OK)
  496. {
  497. curl_socket_t sockfd = (curl_socket_t)sockextr;
  498. /* wait for the socket to become ready for sending */
  499. if(!wait_on_socket(sockfd, (for_recv ? 1 : 0), timeout_ms)) {
  500. return sq_throwerror(v, _SC("Error: timeout."));
  501. }
  502. }
  503. return 0;
  504. }
  505. static SQRESULT sq_EasyCurl_loadlib(HSQUIRRELVM v)
  506. {
  507. SQ_FUNC_VARS_NO_TOP(v);
  508. SQ_GET_STRING(v, 2, libname);
  509. sq_pushbool(v, load_library(libname));
  510. return 1;
  511. }
  512. #define _DECL_EASYCURL_FUNC(name,nparams,pmask) {_SC(#name),sq_EasyCurl_##name,nparams,pmask}
  513. static SQRegFunction EasyCurl_obj_funcs[]={
  514. _DECL_EASYCURL_FUNC(constructor, 1, _SC(".")),
  515. _DECL_EASYCURL_FUNC(duphandle, 1, _SC("x")),
  516. _DECL_EASYCURL_FUNC(getinfo, 3, _SC("xii")),
  517. _DECL_EASYCURL_FUNC(recv, 2, _SC("xi")),
  518. _DECL_EASYCURL_FUNC(reset, 1, _SC("x")),
  519. _DECL_EASYCURL_FUNC(send, 2, _SC("xs")),
  520. _DECL_EASYCURL_FUNC(setopt, 3, _SC("xi s|n|a")),
  521. _DECL_EASYCURL_FUNC(perform, 1, _SC("x")),
  522. _DECL_EASYCURL_FUNC(pause, 2, _SC("xi")),
  523. _DECL_EASYCURL_FUNC(cleanup, 1, _SC("x")),
  524. _DECL_EASYCURL_FUNC(strerror, 2, _SC(".i")),
  525. _DECL_EASYCURL_FUNC(set_reader, -1, _SC("x c|x .")),
  526. _DECL_EASYCURL_FUNC(set_writer, -1, _SC("x c|x .")),
  527. _DECL_EASYCURL_FUNC(wait_on_socket, 3, _SC("x bi")),
  528. _DECL_EASYCURL_FUNC(escape, 2, _SC("xs")),
  529. _DECL_EASYCURL_FUNC(unescape, 2, _SC("xs")),
  530. _DECL_EASYCURL_FUNC(loadlib,2,_SC(".s")),
  531. {0,0}
  532. };
  533. #undef _DECL_EASYCURL_FUNC
  534. typedef struct {
  535. const SQChar *Str;
  536. SQInteger Val;
  537. } KeyIntType, * KeyIntPtrType;
  538. static KeyIntType EasyCurl_constants[] = {
  539. #define MK_CONST(c) {_SC(#c), c}
  540. #define MK_CONST2(c) {_SC(#c), (SQInteger)c}
  541. //MK_CONST(SSL_SESSION_ID_SIZE),
  542. MK_CONST(CURLPROTO_DICT),
  543. MK_CONST(CURLSSH_AUTH_ANY),
  544. MK_CONST(CURLAUTH_NTLM_WB),
  545. MK_CONST(CURLSSH_AUTH_KEYBOARD),
  546. MK_CONST(CURLPROTO_SMTP),
  547. MK_CONST(CURLPROTO_RTMP),
  548. MK_CONST(CURLSSH_AUTH_PUBLICKEY),
  549. MK_CONST(CURLPROTO_POP3),
  550. //MK_CONST(CURLAUTH_NEGOTIATE),
  551. MK_CONST(CURLPROTO_POP3S),
  552. //MK_CONST(CURLHEADER_SEPARATE),
  553. MK_CONST(CURLPROTO_RTMPTE),
  554. MK_CONST(CURLAUTH_DIGEST_IE),
  555. MK_CONST(CURLPROTO_ALL),
  556. MK_CONST(CURLSSH_AUTH_HOST),
  557. MK_CONST(CURLPROTO_HTTP),
  558. MK_CONST(CURLPROTO_HTTPS),
  559. MK_CONST(CURLOPT_SSL_VERIFYPEER),
  560. MK_CONST(CURLOPT_SSL_VERIFYHOST),
  561. MK_CONST(CURLOPT_ACCEPT_ENCODING),
  562. MK_CONST(CURLAUTH_BASIC),
  563. MK_CONST(CURLPROTO_IMAPS),
  564. MK_CONST(CURLPROTO_RTMPTS),
  565. MK_CONST(CURLSSH_AUTH_NONE),
  566. MK_CONST(CURLGSSAPI_DELEGATION_NONE),
  567. MK_CONST(CURLAUTH_DIGEST),
  568. MK_CONST(CURLGSSAPI_DELEGATION_FLAG),
  569. MK_CONST(CURLPROTO_LDAP),
  570. MK_CONST(CURLAUTH_NONE),
  571. MK_CONST(CURLPROTO_SMTPS),
  572. MK_CONST(CURLPROTO_GOPHER),
  573. MK_CONST(CURLSSH_AUTH_DEFAULT),
  574. //MK_CONST(CURLPROTO_SMB),
  575. MK_CONST(CURLPROTO_RTSP),
  576. MK_CONST(CURLSSH_AUTH_PASSWORD),
  577. //MK_CONST(CURLPROTO_SMBS),
  578. MK_CONST2(CURLAUTH_ANYSAFE),
  579. MK_CONST2(CURLPROTO_LDAPS),
  580. MK_CONST2(CURLGSSAPI_DELEGATION_POLICY_FLAG),
  581. MK_CONST2(CURLPROTO_TFTP),
  582. MK_CONST2(CURLPROTO_SFTP),
  583. //MK_CONST2(CURLHEADER_UNIFIED),
  584. MK_CONST2(CURLAUTH_GSSNEGOTIATE),
  585. MK_CONST2(CURLPROTO_FILE),
  586. MK_CONST2(CURLAUTH_NTLM),
  587. MK_CONST2(CURLSSH_AUTH_AGENT),
  588. MK_CONST2(CURLPROTO_HTTPS),
  589. MK_CONST2(CURLPROTO_RTMPS),
  590. MK_CONST2(CURLAUTH_ONLY),
  591. MK_CONST2(CURLPROTO_TELNET),
  592. MK_CONST2(CURLAUTH_ANY),
  593. MK_CONST2(CURLOPT_VERBOSE),
  594. MK_CONST(CURLPROTO_FTPS),
  595. MK_CONST(CURLPROTO_IMAP),
  596. MK_CONST(CURLPROTO_RTMPE),
  597. MK_CONST(CURLPROTO_RTMPT),
  598. MK_CONST(CURLPROTO_FTP),
  599. MK_CONST(CURLPROTO_SCP),
  600. MK_CONST(CURLOPT_URL),
  601. MK_CONST(CURLOPT_FOLLOWLOCATION),
  602. MK_CONST(CURLOPT_POST),
  603. MK_CONST(CURLOPT_POSTFIELDS),
  604. MK_CONST(CURLOPT_HTTPHEADER),
  605. MK_CONST(CURLOPT_POSTFIELDS),
  606. MK_CONST(CURLINFO_EFFECTIVE_URL),
  607. MK_CONST(CURLOPT_MAIL_FROM),
  608. MK_CONST(CURLOPT_MAIL_RCPT),
  609. MK_CONST(CURLOPT_UPLOAD),
  610. MK_CONST(CURLE_OK),
  611. MK_CONST(CURLPAUSE_RECV),
  612. MK_CONST(CURLPAUSE_RECV_CONT),
  613. MK_CONST(CURLPAUSE_SEND),
  614. MK_CONST(CURLPAUSE_SEND_CONT),
  615. MK_CONST(CURLPAUSE_ALL),
  616. MK_CONST(CURLPAUSE_CONT),
  617. MK_CONST(CURLOPT_APPEND),
  618. MK_CONST(CURLOPT_COOKIE),
  619. MK_CONST(CURLOPT_COOKIEFILE),
  620. MK_CONST(CURLOPT_COOKIESESSION),
  621. MK_CONST(CURLOPT_FILE),
  622. MK_CONST(CURLOPT_REFERER),
  623. MK_CONST(CURLOPT_TIMEOUT),
  624. MK_CONST(CURLOPT_USERAGENT),
  625. MK_CONST(CURLOPT_HEADER),
  626. MK_CONST(CURLINFO_CONTENT_TYPE),
  627. MK_CONST(CURLINFO_COOKIELIST),
  628. MK_CONST(CURLINFO_EFFECTIVE_URL),
  629. MK_CONST(CURLINFO_LOCAL_IP),
  630. MK_CONST(CURLINFO_LOCAL_PORT),
  631. MK_CONST(CURLINFO_PRIMARY_IP),
  632. MK_CONST(CURLINFO_PRIMARY_PORT),
  633. MK_CONST(CURLINFO_REDIRECT_COUNT),
  634. MK_CONST(CURLINFO_REDIRECT_URL),
  635. MK_CONST(CURLINFO_RESPONSE_CODE),
  636. MK_CONST(CURL_LAST),
  637. {0,0}
  638. };
  639. /* This defines a function that opens up your library. */
  640. SQRESULT sqext_register_EasyCurl (HSQUIRRELVM v) {
  641. //add a namespace EasyCurl
  642. sq_pushstring(v, SQ_LIBNAME, -1);
  643. sq_newclass(v,SQFalse);
  644. sq_settypetag(v,-1,(SQUserPointer)EasyCurl_Tag);
  645. sq_insert_reg_funcs(v, EasyCurl_obj_funcs);
  646. //add constants
  647. KeyIntPtrType KeyIntPtr;
  648. for (KeyIntPtr = EasyCurl_constants; KeyIntPtr->Str; KeyIntPtr++) {
  649. sq_pushstring(v, KeyIntPtr->Str, -1); //first the key
  650. sq_pushinteger(v, KeyIntPtr->Val); //then the value
  651. sq_newslot(v, -3, SQFalse); //store then
  652. }
  653. sq_newslot(v,-3,SQFalse); //add EasyCurl table to the root table
  654. return SQ_OK;
  655. }
  656. #ifdef __cplusplus
  657. }
  658. #endif //USE_EASYCURL
  659. #endif