sq_libcurl.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726
  1. #ifdef __cplusplus
  2. extern "C" {
  3. #endif
  4. #ifdef SQ_USE_EASYCURL
  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()
  83. {
  84. if(dlcurl_easy_cleanup) return true;
  85. if(libdyn.open(dynamicLibName))
  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. };
  134. static const SQChar SQ_LIBNAME[] = _SC("EasyCurl");
  135. SQ_OPT_STRING_STRLEN();
  136. static const SQChar EasyCurl_Tag[] = _SC("EasyCurl_TAG");
  137. #define GET_EasyCurl_INSTANCE() SQ_GET_INSTANCE(v, 1, sq_easycurl_st, EasyCurl_Tag) \
  138. if(self == NULL) return sq_throwerror(v, _SC("EasyCurl object already closed"));
  139. /* Auxiliary function that waits on the socket. */
  140. static int wait_on_socket(curl_socket_t sockfd, int for_recv, long timeout_ms)
  141. {
  142. struct timeval tv;
  143. fd_set infd, outfd, errfd;
  144. int res;
  145. tv.tv_sec = timeout_ms / 1000;
  146. tv.tv_usec= (timeout_ms % 1000) * 1000;
  147. FD_ZERO(&infd);
  148. FD_ZERO(&outfd);
  149. FD_ZERO(&errfd);
  150. FD_SET(sockfd, &errfd); /* always check for error */
  151. if(for_recv) {
  152. FD_SET(sockfd, &infd);
  153. }
  154. else {
  155. FD_SET(sockfd, &outfd);
  156. }
  157. /* select() returns the number of signalled sockets or -1 */
  158. res = select(sockfd + 1, &infd, &outfd, &errfd, &tv);
  159. return res;
  160. }
  161. static SQRESULT EasyCurl_release_hook(SQUserPointer p, SQInteger size, void */*ep*/)
  162. {
  163. sq_easycurl_st *self = (sq_easycurl_st*)p;
  164. if(self)
  165. {
  166. dlcurl_easy_cleanup(self->curl);
  167. sq_release(self->vm, &self->reader_cb);
  168. sq_release(self->vm, &self->reader_cb_udata);
  169. sq_release(self->vm, &self->writer_cb);
  170. sq_release(self->vm, &self->writer_cb_udata);
  171. sq_free(self, sizeof(*self));
  172. }
  173. return 0;
  174. }
  175. static SQRESULT EasyCurl_constructor(HSQUIRRELVM v, int idx, CURL *EasyCurl, int free_on_gc)
  176. {
  177. if(!EasyCurl)
  178. return sq_throwerror(v, _SC("Could'nt create an EasyCurl object."));
  179. sq_easycurl_st *self = (sq_easycurl_st*)sq_malloc(sizeof(*self));
  180. memset(self, 0, sizeof(*self));
  181. sq_resetobject(&self->reader_cb);
  182. sq_resetobject(&self->reader_cb_udata);
  183. sq_resetobject(&self->writer_cb);
  184. sq_resetobject(&self->writer_cb_udata);
  185. self->vm = v;
  186. self->curl = EasyCurl;
  187. SQInteger rc = sq_setinstanceup(v, idx, self);
  188. if(free_on_gc) sq_setreleasehook(v,-1, EasyCurl_release_hook);
  189. return rc;
  190. }
  191. static SQRESULT EasyCurl_create_instance(HSQUIRRELVM v, CURL *EasyCurl, int free_on_gc)
  192. {
  193. sq_pushstring(v, SQ_LIBNAME, -1);
  194. if(sq_getonroottable(v) == SQ_OK){
  195. sq_pushstring(v, EasyCurl_Tag, -1);
  196. if(sq_get(v, -2) == SQ_OK){
  197. if(sq_createinstance(v, -1) == SQ_OK){
  198. return EasyCurl_constructor(v, -1, EasyCurl, free_on_gc);
  199. }
  200. }
  201. }
  202. return SQ_ERROR;
  203. }
  204. static SQRESULT sq_EasyCurl_constructor(HSQUIRRELVM v){
  205. if(!load_library()) return sq_throwerror(v, _SC("Failed to load libcurl !"));
  206. CURL *curl = dlcurl_easy_init();
  207. return EasyCurl_constructor(v, 1, curl, 1);
  208. }
  209. static SQRESULT sq_EasyCurl_duphandle(HSQUIRRELVM v){
  210. SQ_FUNC_VARS_NO_TOP(v);
  211. GET_EasyCurl_INSTANCE();
  212. CURL *curl = dlcurl_easy_duphandle(self->curl);
  213. return EasyCurl_create_instance(v, curl, 1);
  214. }
  215. static SQRESULT sq_EasyCurl_getinfo(HSQUIRRELVM v){
  216. SQ_FUNC_VARS_NO_TOP(v);
  217. GET_EasyCurl_INSTANCE();
  218. CURLcode rc = CURL_LAST;
  219. SQ_GET_INTEGER(v, 2, option);
  220. SQ_GET_INTEGER(v, 3, option_type);
  221. switch(option_type)
  222. {
  223. case 's':
  224. const SQChar *str_value;
  225. rc = dlcurl_easy_getinfo(self->curl, (CURLINFO)option, &str_value);
  226. if(rc == CURLE_OK) sq_pushstring(v, str_value, -1);
  227. break;
  228. case 'i':
  229. long int_value;
  230. rc = dlcurl_easy_getinfo(self->curl, (CURLINFO)option, &int_value);
  231. if(rc == CURLE_OK) sq_pushinteger(v, int_value);
  232. break;
  233. case 'f':
  234. double float_value;
  235. rc = dlcurl_easy_getinfo(self->curl, (CURLINFO)option, &float_value);
  236. if(rc == CURLE_OK) sq_pushfloat(v, float_value);
  237. break;
  238. default:
  239. return sq_throwerror(v, _SC("invalid option value type"));
  240. }
  241. if(rc != CURLE_OK) sq_pushinteger(v, CURL_LAST);
  242. return 1;
  243. }
  244. static SQRESULT sq_EasyCurl_setopt(HSQUIRRELVM v){
  245. SQ_FUNC_VARS_NO_TOP(v);
  246. GET_EasyCurl_INSTANCE();
  247. CURLcode rc = CURLE_OK;
  248. SQ_GET_INTEGER(v, 2, option);
  249. switch(sq_gettype(v, 3))
  250. {
  251. case OT_STRING:
  252. SQ_GET_STRING(v, 3, str_value);
  253. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, str_value);
  254. break;
  255. case OT_INTEGER:
  256. SQ_GET_INTEGER(v, 3, int_value);
  257. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, int_value);
  258. break;
  259. case OT_FLOAT:
  260. SQ_GET_FLOAT(v, 3, float_value);
  261. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, float_value);
  262. break;
  263. /*
  264. case OT_ARRAY:
  265. {
  266. struct curl_slist *optlist = NULL;
  267. const SQChar *str;
  268. SQInteger asize = sq_getsize(v, 3);
  269. for(int i=0; i<asize; ++i)
  270. {
  271. sq_pushinteger(v, i);
  272. sq_get(v, 3);
  273. sq_getstring(v, -1, &str);
  274. optlist = dlcurl_slist_append(optlist, str);
  275. sq_poptop(v);
  276. }
  277. rc = dlcurl_easy_setopt(self->curl, (CURLoption)option, optlist);
  278. dlcurl_slist_free_all(optlist);
  279. }
  280. break;
  281. */
  282. default:
  283. return sq_throwerror(v, _SC("invalid option value type"));
  284. }
  285. sq_pushinteger(v, rc);
  286. return 1;
  287. }
  288. static SQRESULT sq_EasyCurl_recv(HSQUIRRELVM v){
  289. SQ_FUNC_VARS_NO_TOP(v);
  290. GET_EasyCurl_INSTANCE();
  291. SQ_GET_INTEGER(v, 2, try_receive_n);
  292. SQChar *buffer = sq_getscratchpad(v, try_receive_n);
  293. size_t recv_size = 0;
  294. CURLcode rc = dlcurl_easy_recv(self->curl, (void *)buffer, try_receive_n, &recv_size);
  295. if(rc != CURLE_OK) sq_pushinteger(v, -rc);
  296. else sq_pushinteger(v, recv_size);
  297. return 1;
  298. }
  299. static SQRESULT sq_EasyCurl_send(HSQUIRRELVM v){
  300. SQ_FUNC_VARS_NO_TOP(v);
  301. GET_EasyCurl_INSTANCE();
  302. SQ_GET_STRING(v, 2, buffer);
  303. size_t send_size = 0;
  304. CURLcode rc = dlcurl_easy_send(self->curl, (const void *)buffer, buffer_size, &send_size);
  305. if(rc != CURLE_OK) sq_pushinteger(v, -rc);
  306. else sq_pushinteger(v, send_size);
  307. return 1;
  308. }
  309. static SQRESULT sq_EasyCurl_reset(HSQUIRRELVM v){
  310. SQ_FUNC_VARS_NO_TOP(v);
  311. GET_EasyCurl_INSTANCE();
  312. dlcurl_easy_reset(self->curl);
  313. return 0;
  314. }
  315. static SQRESULT sq_EasyCurl_perform(HSQUIRRELVM v){
  316. SQ_FUNC_VARS_NO_TOP(v);
  317. GET_EasyCurl_INSTANCE();
  318. CURLcode rc = dlcurl_easy_perform(self->curl);
  319. sq_pushinteger(v, rc);
  320. return 1;
  321. }
  322. static SQRESULT sq_EasyCurl_pause(HSQUIRRELVM v){
  323. SQ_FUNC_VARS_NO_TOP(v);
  324. GET_EasyCurl_INSTANCE();
  325. SQ_GET_INTEGER(v, 2, flag_how);
  326. CURLcode rc = dlcurl_easy_pause(self->curl, flag_how);
  327. sq_pushinteger(v, rc);
  328. return 1;
  329. }
  330. static SQRESULT sq_EasyCurl_cleanup(HSQUIRRELVM v){
  331. SQ_FUNC_VARS_NO_TOP(v);
  332. GET_EasyCurl_INSTANCE();
  333. dlcurl_easy_cleanup(self->curl);
  334. sq_setinstanceup(v, 1, 0); //invalidate this curl handle
  335. return 0;
  336. }
  337. static SQRESULT sq_EasyCurl_strerror(HSQUIRRELVM v){
  338. SQ_FUNC_VARS_NO_TOP(v);
  339. GET_EasyCurl_INSTANCE();
  340. SQ_GET_INTEGER(v, 2, error_code);
  341. sq_pushstring(v, dlcurl_easy_strerror((CURLcode)error_code), -1);
  342. return 1;
  343. }
  344. static SQRESULT sq_EasyCurl_escape(HSQUIRRELVM v){
  345. SQ_FUNC_VARS_NO_TOP(v);
  346. GET_EasyCurl_INSTANCE();
  347. SQ_GET_STRING(v, 2, str);
  348. sq_pushstring(v, dlcurl_easy_escape(self->curl, str, str_size), -1);
  349. return 1;
  350. }
  351. static SQRESULT sq_EasyCurl_unescape(HSQUIRRELVM v){
  352. SQ_FUNC_VARS_NO_TOP(v);
  353. GET_EasyCurl_INSTANCE();
  354. SQ_GET_STRING(v, 2, str);
  355. sq_pushstring(v, dlcurl_easy_unescape(self->curl, str, str_size), -1);
  356. return 1;
  357. }
  358. static size_t sq_EasyCurl_reader_writer_callback(char *bufptr, size_t size, size_t nitems, void *userp, int isReader)
  359. {
  360. SQInteger result = 0; /* abort by default */
  361. sq_easycurl_st *self = (sq_easycurl_st*)userp;
  362. HSQUIRRELVM v = self->vm;
  363. int top = sq_gettop(v);
  364. HSQOBJECT &rw_cb = (isReader ? self->reader_cb : self->writer_cb);
  365. HSQOBJECT &rw_cb_udata = (isReader ? self->reader_cb_udata : self->writer_cb_udata);
  366. size_t data_size = size * nitems;
  367. sq_pushobject(v, rw_cb);
  368. SQObjectType otype = sq_gettype(v, -1);
  369. switch(otype)
  370. {
  371. case OT_INSTANCE:
  372. {
  373. SQBlob *blob = NULL;
  374. if(SQ_FAILED(sq_getinstanceup(v,-1,(SQUserPointer*)&blob,(SQUserPointer)SQBlob::SQBlob_TAG)))
  375. break;
  376. if(!blob || !blob->IsValid())
  377. break;
  378. blob->Write(bufptr, data_size);
  379. result = data_size;
  380. }
  381. break;
  382. case OT_CLOSURE:
  383. sq_pushroottable(v);
  384. sq_pushinteger(v, size);
  385. sq_pushinteger(v, nitems);
  386. sq_pushobject(v, rw_cb_udata);
  387. if(!isReader)
  388. {
  389. sq_pushstring(v, (const SQChar*)bufptr, data_size);
  390. }
  391. /* call squilu function */
  392. if (sq_call(v, isReader ? 4 : 5, SQTrue, SQFalse) == SQ_OK)
  393. {
  394. switch(sq_gettype(v, -1))
  395. {
  396. case OT_INTEGER:
  397. sq_getinteger(v, -1, &result);
  398. break;
  399. case OT_STRING:
  400. if(isReader)
  401. {
  402. const SQChar *str;
  403. if(sq_getstr_and_size(v,-1, &str, &result) == SQ_OK)
  404. {
  405. memcpy(bufptr, str, result);
  406. }
  407. }
  408. break;
  409. }
  410. }
  411. break;
  412. }
  413. sq_settop(v, top);
  414. return result;
  415. }
  416. static size_t sq_EasyCurl_reader_callback(char *bufptr, size_t size, size_t nitems, void *userp)
  417. {
  418. return sq_EasyCurl_reader_writer_callback(bufptr, size, nitems, userp, 1);
  419. }
  420. static size_t sq_EasyCurl_writer_callback(char *bufptr, size_t size, size_t nitems, void *userp)
  421. {
  422. return sq_EasyCurl_reader_writer_callback(bufptr, size, nitems, userp, 0);
  423. }
  424. static SQRESULT sq_EasyCurl_set_reader_writer(HSQUIRRELVM v, int isReader){
  425. SQ_FUNC_VARS(v);
  426. GET_EasyCurl_INSTANCE();
  427. CURLcode rc;
  428. HSQOBJECT &rw_cb = (isReader ? self->reader_cb : self->writer_cb);
  429. HSQOBJECT &rw_cb_udata = (isReader ? self->reader_cb_udata : self->writer_cb_udata);
  430. CURLoption rw_fnc_option = (isReader ? CURLOPT_READFUNCTION : CURLOPT_WRITEFUNCTION);
  431. /* clear reader handler */
  432. rc = dlcurl_easy_setopt(self->curl, rw_fnc_option, NULL);
  433. sq_release(v, &rw_cb);
  434. sq_release(v, &rw_cb_udata);
  435. sq_resetobject(&rw_cb);
  436. sq_resetobject(&rw_cb_udata);
  437. SQObjectType otype = sq_gettype(v, 2);
  438. if(_top_ > 1 && otype != OT_NULL)
  439. {
  440. switch(otype)
  441. {
  442. case OT_INSTANCE:
  443. {
  444. SQBlob *blob = NULL;
  445. if(SQ_FAILED(sq_getinstanceup(v,2,(SQUserPointer*)&blob,(SQUserPointer)SQBlob::SQBlob_TAG)))
  446. return sq_throwerror(v,_SC("expect a blob as second parameter"));
  447. if(!blob || !blob->IsValid())
  448. return sq_throwerror(v,_SC("the blob is invalid"));
  449. }
  450. break;
  451. case OT_CLOSURE:
  452. break;
  453. default:
  454. return sq_throwerror(v, _SC("invalid fisrt parameter expected closure/blob"));
  455. }
  456. sq_getstackobj(v, 2, &rw_cb);
  457. sq_addref(v, &rw_cb);
  458. if(_top_ > 2)
  459. {
  460. sq_getstackobj(v, 3, &rw_cb_udata);
  461. sq_addref(v, &rw_cb_udata);
  462. }
  463. /* set authorizer_hook callback */
  464. rc = dlcurl_easy_setopt(self->curl, rw_fnc_option,
  465. (isReader ? sq_EasyCurl_reader_callback : sq_EasyCurl_writer_callback));
  466. if(rc == CURLE_OK) rc = dlcurl_easy_setopt(self->curl,
  467. (isReader ? CURLOPT_READDATA : CURLOPT_WRITEDATA), self);
  468. }
  469. sq_pushinteger(v, rc);
  470. return 1;
  471. }
  472. static SQRESULT sq_EasyCurl_set_reader(HSQUIRRELVM v){
  473. return sq_EasyCurl_set_reader_writer(v, 1);
  474. }
  475. static SQRESULT sq_EasyCurl_set_writer(HSQUIRRELVM v){
  476. return sq_EasyCurl_set_reader_writer(v, 0);
  477. }
  478. static SQRESULT sq_EasyCurl_wait_on_socket(HSQUIRRELVM v){
  479. SQ_FUNC_VARS_NO_TOP(v);
  480. GET_EasyCurl_INSTANCE();
  481. SQ_GET_BOOL(v, 2, for_recv);
  482. SQ_GET_INTEGER(v, 3, timeout_ms);
  483. long sockextr;
  484. CURLcode rc = dlcurl_easy_getinfo(self->curl, CURLINFO_LASTSOCKET, &sockextr);
  485. if(rc == CURLE_OK)
  486. {
  487. curl_socket_t sockfd = (curl_socket_t)sockextr;
  488. /* wait for the socket to become ready for sending */
  489. if(!wait_on_socket(sockfd, (for_recv ? 1 : 0), timeout_ms)) {
  490. return sq_throwerror(v, _SC("Error: timeout."));
  491. }
  492. }
  493. return 0;
  494. }
  495. #define _DECL_EASYCURL_FUNC(name,nparams,pmask) {_SC(#name),sq_EasyCurl_##name,nparams,pmask}
  496. static SQRegFunction EasyCurl_obj_funcs[]={
  497. _DECL_EASYCURL_FUNC(constructor, 1, _SC(".")),
  498. _DECL_EASYCURL_FUNC(duphandle, 1, _SC("x")),
  499. _DECL_EASYCURL_FUNC(getinfo, 3, _SC("xii")),
  500. _DECL_EASYCURL_FUNC(recv, 2, _SC("xi")),
  501. _DECL_EASYCURL_FUNC(reset, 1, _SC("x")),
  502. _DECL_EASYCURL_FUNC(send, 2, _SC("xs")),
  503. _DECL_EASYCURL_FUNC(setopt, 3, _SC("xi s|n|a")),
  504. _DECL_EASYCURL_FUNC(perform, 1, _SC("x")),
  505. _DECL_EASYCURL_FUNC(pause, 2, _SC("xi")),
  506. _DECL_EASYCURL_FUNC(cleanup, 1, _SC("x")),
  507. _DECL_EASYCURL_FUNC(strerror, 2, _SC(".i")),
  508. _DECL_EASYCURL_FUNC(set_reader, -1, _SC("x c|x .")),
  509. _DECL_EASYCURL_FUNC(set_writer, -1, _SC("x c|x .")),
  510. _DECL_EASYCURL_FUNC(wait_on_socket, 3, _SC("x bi")),
  511. _DECL_EASYCURL_FUNC(escape, 2, _SC("xs")),
  512. _DECL_EASYCURL_FUNC(unescape, 2, _SC("xs")),
  513. {0,0}
  514. };
  515. #undef _DECL_EASYCURL_FUNC
  516. typedef struct {
  517. const SQChar *Str;
  518. SQInteger Val;
  519. } KeyIntType, * KeyIntPtrType;
  520. static KeyIntType EasyCurl_constants[] = {
  521. #define MK_CONST(c) {_SC(#c), c}
  522. #define MK_CONST2(c) {_SC(#c), (SQInteger)c}
  523. //MK_CONST(SSL_SESSION_ID_SIZE),
  524. MK_CONST(CURLPROTO_DICT),
  525. MK_CONST(CURLSSH_AUTH_ANY),
  526. MK_CONST(CURLAUTH_NTLM_WB),
  527. MK_CONST(CURLSSH_AUTH_KEYBOARD),
  528. MK_CONST(CURLPROTO_SMTP),
  529. MK_CONST(CURLPROTO_RTMP),
  530. MK_CONST(CURLSSH_AUTH_PUBLICKEY),
  531. MK_CONST(CURLPROTO_POP3),
  532. MK_CONST(CURLAUTH_NEGOTIATE),
  533. MK_CONST(CURLPROTO_POP3S),
  534. MK_CONST(CURLHEADER_SEPARATE),
  535. MK_CONST(CURLPROTO_RTMPTE),
  536. MK_CONST(CURLAUTH_DIGEST_IE),
  537. MK_CONST(CURLPROTO_ALL),
  538. MK_CONST(CURLSSH_AUTH_HOST),
  539. MK_CONST(CURLPROTO_HTTP),
  540. MK_CONST(CURLAUTH_BASIC),
  541. MK_CONST(CURLPROTO_IMAPS),
  542. MK_CONST(CURLPROTO_RTMPTS),
  543. MK_CONST(CURLSSH_AUTH_NONE),
  544. MK_CONST(CURLGSSAPI_DELEGATION_NONE),
  545. MK_CONST(CURLAUTH_DIGEST),
  546. MK_CONST(CURLGSSAPI_DELEGATION_FLAG),
  547. MK_CONST(CURLPROTO_LDAP),
  548. MK_CONST(CURLAUTH_NONE),
  549. MK_CONST(CURLPROTO_SMTPS),
  550. MK_CONST(CURLPROTO_GOPHER),
  551. MK_CONST(CURLSSH_AUTH_DEFAULT),
  552. MK_CONST(CURLPROTO_SMB),
  553. MK_CONST(CURLPROTO_RTSP),
  554. MK_CONST(CURLSSH_AUTH_PASSWORD),
  555. MK_CONST(CURLPROTO_SMBS),
  556. MK_CONST2(CURLAUTH_ANYSAFE),
  557. MK_CONST2(CURLPROTO_LDAPS),
  558. MK_CONST2(CURLGSSAPI_DELEGATION_POLICY_FLAG),
  559. MK_CONST2(CURLPROTO_TFTP),
  560. MK_CONST2(CURLPROTO_SFTP),
  561. MK_CONST2(CURLHEADER_UNIFIED),
  562. MK_CONST2(CURLAUTH_GSSNEGOTIATE),
  563. MK_CONST2(CURLPROTO_FILE),
  564. MK_CONST2(CURLAUTH_NTLM),
  565. MK_CONST2(CURLSSH_AUTH_AGENT),
  566. MK_CONST2(CURLPROTO_HTTPS),
  567. MK_CONST2(CURLPROTO_RTMPS),
  568. MK_CONST2(CURLAUTH_ONLY),
  569. MK_CONST2(CURLPROTO_TELNET),
  570. MK_CONST2(CURLAUTH_ANY),
  571. MK_CONST(CURLPROTO_FTPS),
  572. MK_CONST(CURLPROTO_IMAP),
  573. MK_CONST(CURLPROTO_RTMPE),
  574. MK_CONST(CURLPROTO_RTMPT),
  575. MK_CONST(CURLPROTO_FTP),
  576. MK_CONST(CURLPROTO_SCP),
  577. MK_CONST(CURLOPT_URL),
  578. MK_CONST(CURLINFO_EFFECTIVE_URL),
  579. MK_CONST(CURLOPT_MAIL_FROM),
  580. MK_CONST(CURLOPT_MAIL_RCPT),
  581. MK_CONST(CURLOPT_UPLOAD),
  582. MK_CONST(CURLE_OK),
  583. MK_CONST(CURLPAUSE_RECV),
  584. MK_CONST(CURLPAUSE_RECV_CONT),
  585. MK_CONST(CURLPAUSE_SEND),
  586. MK_CONST(CURLPAUSE_SEND_CONT),
  587. MK_CONST(CURLPAUSE_ALL),
  588. MK_CONST(CURLPAUSE_CONT),
  589. MK_CONST(CURL_LAST),
  590. {0,0}
  591. };
  592. /* This defines a function that opens up your library. */
  593. SQRESULT sqext_register_EasyCurl (HSQUIRRELVM v) {
  594. //add a namespace EasyCurl
  595. sq_pushstring(v, SQ_LIBNAME, -1);
  596. sq_newclass(v,SQFalse);
  597. sq_settypetag(v,-1,(SQUserPointer)EasyCurl_Tag);
  598. sq_insert_reg_funcs(v, EasyCurl_obj_funcs);
  599. //add constants
  600. KeyIntPtrType KeyIntPtr;
  601. for (KeyIntPtr = EasyCurl_constants; KeyIntPtr->Str; KeyIntPtr++) {
  602. sq_pushstring(v, KeyIntPtr->Str, -1); //first the key
  603. sq_pushinteger(v, KeyIntPtr->Val); //then the value
  604. sq_newslot(v, -3, SQFalse); //store then
  605. }
  606. sq_newslot(v,-3,SQFalse); //add EasyCurl table to the root table
  607. return SQ_OK;
  608. }
  609. #ifdef __cplusplus
  610. }
  611. #endif //USE_EASYCURL
  612. #endif