sq_libcurl.cpp 25 KB

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