sys_request_client.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747
  1. /**
  2. * @file sys_request_client.c
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * Redistribution and use in source and binary forms, with or without
  8. * modification, are permitted provided that the following conditions are met:
  9. * 1. Redistributions of source code must retain the above copyright
  10. * notice, this list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * 3. Neither the name of the author nor the
  15. * names of its contributors may be used to endorse or promote products
  16. * derived from this software without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  19. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21. * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  22. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28. *
  29. * @section DESCRIPTION
  30. *
  31. * Synopsis:
  32. * sys.request_client(string connect_addr)
  33. *
  34. * Description:
  35. * Connects to a request server (sys.request_server()).
  36. * Goes up when the connection, and dies with error when it is broken.
  37. * When requested to die, dies immediately, breaking the connection.
  38. *
  39. * The connect_addr argument must be one of:
  40. * - {"unix", socket_path}
  41. * Connects to a Unix socket.
  42. * - {"tcp", ip_address, port_number}
  43. * Connects to a TCP server. The address must be numeric and not a name.
  44. * For IPv6, the address must be enclosed in [].
  45. *
  46. * Synopsis:
  47. * sys.request_client::request(request_data, string reply_handler, string finished_handler, list args)
  48. *
  49. * Description:
  50. * Sends a request to the server and dispatches replies to the provided handlers.
  51. *
  52. * The 'request_data' argument is sent as part of the request and is used by the server
  53. * to determine what to do with the request.
  54. *
  55. * When a reply is received, a new template process is created from 'reply_handler' to process the
  56. * reply. This process can access the reply data sent by the server using '_reply.data'.
  57. * Similarly, if the server finishes the request, a process is created from 'finished_handler'.
  58. * In both cases, the process can access objects as seen from the request statement via "_caller".
  59. * Termination of these processes is initiated immediately after they completes. They are created
  60. * synchronously - if a reply or a finished message arrives before a previous process is has
  61. * finished, it is queued. Once the finished message has been processed by 'finished_handler', no
  62. * more processes will be created.
  63. *
  64. * When the request statement is requested to terminate, it initiates termination of the current
  65. * handler process and waits for it to terminate (if any is running), and then dies.
  66. * If the corresponding client statement dies after being requested to die, or as a result of
  67. * an error, the request statement will not react to this. It will dispatch any pending messages
  68. * and then proceed to do nothing. In this case, if a finished message was not received, it will
  69. * not be dispatched.
  70. *
  71. * The request statement may however die at any time due to errors. In this case, it will
  72. * initiate termination of the current process and wait for it to terminate (if any) before dying.
  73. *
  74. * The request protocol and the server allow the client the abort requests at any time, and to
  75. * have the client notified only after the request has been completely aborted (i.e. the handler
  76. * process of sys.request_server() has deinitialized completely). This client implementation will
  77. * automatically request abortion of active requests when the request statement is requested
  78. * to die. However, the request statement will not wait for the abortion to finish before dying.
  79. * This means, for instance, that if you initialize a request statement right after having
  80. * deinitiazed it, the requests may overlap on the server side.
  81. */
  82. #include <stdlib.h>
  83. #include <string.h>
  84. #include <inttypes.h>
  85. #include <limits.h>
  86. #include <misc/offset.h>
  87. #include <misc/parse_number.h>
  88. #include <structure/LinkedList0.h>
  89. #include <structure/LinkedList1.h>
  90. #include <system/BAddr.h>
  91. #include <ncd/NCDModule.h>
  92. #include <ncd/NCDRequestClient.h>
  93. #include <ncd/NCDValCompat.h>
  94. #include <generated/blog_channel_ncd_sys_request_client.h>
  95. #define ModuleLog(i, ...) NCDModuleInst_Backend_Log((i), BLOG_CURRENT_CHANNEL, __VA_ARGS__)
  96. #define CSTATE_CONNECTING 1
  97. #define CSTATE_CONNECTED 2
  98. #define RRSTATE_SENDING_REQUEST 1
  99. #define RRSTATE_READY 2
  100. #define RRSTATE_GONE_BAD 3
  101. #define RRSTATE_GONE_GOOD 4
  102. #define RPSTATE_NONE 1
  103. #define RPSTATE_WORKING 2
  104. #define RPSTATE_TERMINATING 3
  105. #define RDSTATE_NONE 1
  106. #define RDSTATE_DYING 2
  107. #define RDSTATE_DYING_ERROR 3
  108. struct instance {
  109. NCDModuleInst *i;
  110. NCDRequestClient client;
  111. LinkedList0 requests_list;
  112. int state;
  113. };
  114. struct request_instance {
  115. NCDModuleInst *i;
  116. const char *reply_handler;
  117. const char *finished_handler;
  118. NCDValRef args;
  119. struct instance *client;
  120. NCDRequestClientRequest request;
  121. LinkedList0Node requests_list_node;
  122. LinkedList1 replies_list;
  123. NCDModuleProcess process;
  124. int process_is_finished;
  125. NCDValMem process_reply_mem;
  126. NCDValRef process_reply_data;
  127. int rstate;
  128. int pstate;
  129. int dstate;
  130. };
  131. struct reply {
  132. LinkedList1Node replies_list_node;
  133. NCDValMem mem;
  134. NCDValRef val;
  135. };
  136. static void client_handler_error (struct instance *o);
  137. static void client_handler_connected (struct instance *o);
  138. static void request_handler_sent (struct request_instance *o);
  139. static void request_handler_reply (struct request_instance *o, NCDValue reply_data);
  140. static void request_handler_finished (struct request_instance *o, int is_error);
  141. static void request_process_handler_event (struct request_instance *o, int event);
  142. static int request_process_func_getspecialobj (struct request_instance *o, const char *name, NCDObject *out_object);
  143. static int request_process_caller_obj_func_getobj (struct request_instance *o, const char *name, NCDObject *out_object);
  144. static int request_process_reply_obj_func_getvar (struct request_instance *o, const char *name, NCDValMem *mem, NCDValRef *out);
  145. static void request_gone (struct request_instance *o, int is_bad);
  146. static void request_terminate_process (struct request_instance *o);
  147. static void request_die (struct request_instance *o, int is_error);
  148. static void request_free_reply (struct request_instance *o, struct reply *r, int have_value);
  149. static int request_init_reply_process (struct request_instance *o, NCDValMem reply_mem, NCDValSafeRef reply_data);
  150. static int request_init_finished_process (struct request_instance *o);
  151. static int get_connect_addr (struct instance *o, NCDValRef connect_addr_arg, struct NCDRequestClient_addr *out_addr);
  152. static void instance_free (struct instance *o, int with_error);
  153. static void request_instance_free (struct request_instance *o, int with_error);
  154. static void client_handler_error (struct instance *o)
  155. {
  156. ModuleLog(o->i, BLOG_ERROR, "client error");
  157. // free instance
  158. instance_free(o, 1);
  159. }
  160. static void client_handler_connected (struct instance *o)
  161. {
  162. ASSERT(o->state == CSTATE_CONNECTING)
  163. // signal up
  164. NCDModuleInst_Backend_Up(o->i);
  165. // set state connected
  166. o->state = CSTATE_CONNECTED;
  167. }
  168. static void request_handler_sent (struct request_instance *o)
  169. {
  170. ASSERT(o->rstate == RRSTATE_SENDING_REQUEST)
  171. // signal up
  172. NCDModuleInst_Backend_Up(o->i);
  173. // set rstate ready
  174. o->rstate = RRSTATE_READY;
  175. }
  176. static void request_handler_reply (struct request_instance *o, NCDValue reply_data)
  177. {
  178. ASSERT(o->rstate == RRSTATE_READY)
  179. NCDValMem mem;
  180. NCDValMem_Init(&mem);
  181. NCDValRef val;
  182. int res = NCDValCompat_ValueToVal(&reply_data, &mem, &val);
  183. NCDValue_Free(&reply_data);
  184. if (!res) {
  185. ModuleLog(o->i, BLOG_ERROR, "NCDValCompat_ValueToVal failed");
  186. goto fail1;
  187. }
  188. // queue reply if process is running
  189. if (o->pstate != RPSTATE_NONE) {
  190. struct reply *r = malloc(sizeof(*r));
  191. if (!r) {
  192. ModuleLog(o->i, BLOG_ERROR, "malloc failed");
  193. goto fail1;
  194. }
  195. r->mem = mem;
  196. r->val = NCDVal_Moved(&r->mem, val);
  197. LinkedList1_Append(&o->replies_list, &r->replies_list_node);
  198. return;
  199. }
  200. // start reply process
  201. if (!request_init_reply_process(o, mem, NCDVal_ToSafe(val))) {
  202. goto fail1;
  203. }
  204. return;
  205. fail1:
  206. NCDValMem_Free(&mem);
  207. request_die(o, 1);
  208. }
  209. static void request_handler_finished (struct request_instance *o, int is_error)
  210. {
  211. ASSERT(o->rstate == RRSTATE_SENDING_REQUEST || o->rstate == RRSTATE_READY)
  212. ASSERT(is_error || o->rstate == RRSTATE_READY)
  213. if (is_error) {
  214. ModuleLog(o->i, BLOG_ERROR, "received error reply");
  215. goto fail;
  216. }
  217. // request gone good
  218. request_gone(o, 0);
  219. // start process for reporting finished, if possible
  220. if (o->pstate == RPSTATE_NONE) {
  221. if (!request_init_finished_process(o)) {
  222. goto fail;
  223. }
  224. }
  225. return;
  226. fail:
  227. request_die(o, 1);
  228. }
  229. static void request_process_handler_event (struct request_instance *o, int event)
  230. {
  231. ASSERT(o->pstate != RPSTATE_NONE)
  232. switch (event) {
  233. case NCDMODULEPROCESS_EVENT_UP: {
  234. ASSERT(o->pstate == RPSTATE_WORKING)
  235. // request process termination
  236. request_terminate_process(o);
  237. } break;
  238. case NCDMODULEPROCESS_EVENT_DOWN: {
  239. ASSERT(0)
  240. } break;
  241. case NCDMODULEPROCESS_EVENT_TERMINATED: {
  242. ASSERT(o->pstate == RPSTATE_TERMINATING)
  243. ASSERT(o->rstate != RRSTATE_SENDING_REQUEST)
  244. // free process
  245. NCDModuleProcess_Free(&o->process);
  246. // free reply data
  247. if (!o->process_is_finished) {
  248. NCDValMem_Free(&o->process_reply_mem);
  249. }
  250. // set process state none
  251. o->pstate = RPSTATE_NONE;
  252. // die finally if requested
  253. if (o->dstate == RDSTATE_DYING || o->dstate == RDSTATE_DYING_ERROR) {
  254. request_instance_free(o, o->dstate == RDSTATE_DYING_ERROR);
  255. return;
  256. }
  257. if (!LinkedList1_IsEmpty(&o->replies_list)) {
  258. // get first reply
  259. struct reply *r = UPPER_OBJECT(LinkedList1_GetFirst(&o->replies_list), struct reply, replies_list_node);
  260. // start reply process
  261. if (!request_init_reply_process(o, r->mem, NCDVal_ToSafe(r->val))) {
  262. goto fail;
  263. }
  264. // free reply
  265. request_free_reply(o, r, 0);
  266. }
  267. else if (o->rstate == RRSTATE_GONE_GOOD && !o->process_is_finished) {
  268. // start process for reporting finished
  269. if (!request_init_finished_process(o)) {
  270. goto fail;
  271. }
  272. }
  273. return;
  274. fail:
  275. request_die(o, 1);
  276. } break;
  277. }
  278. }
  279. static int request_process_func_getspecialobj (struct request_instance *o, const char *name, NCDObject *out_object)
  280. {
  281. ASSERT(o->pstate != RPSTATE_NONE)
  282. if (!strcmp(name, "_caller")) {
  283. *out_object = NCDObject_Build(NULL, o, NULL, (NCDObject_func_getobj)request_process_caller_obj_func_getobj);
  284. return 1;
  285. }
  286. if (!o->process_is_finished && !strcmp(name, "_reply")) {
  287. *out_object = NCDObject_Build(NULL, o, (NCDObject_func_getvar)request_process_reply_obj_func_getvar, NULL);
  288. return 1;
  289. }
  290. return 0;
  291. }
  292. static int request_process_caller_obj_func_getobj (struct request_instance *o, const char *name, NCDObject *out_object)
  293. {
  294. ASSERT(o->pstate != RPSTATE_NONE)
  295. return NCDModuleInst_Backend_GetObj(o->i, name, out_object);
  296. }
  297. static int request_process_reply_obj_func_getvar (struct request_instance *o, const char *name, NCDValMem *mem, NCDValRef *out)
  298. {
  299. ASSERT(o->pstate != RPSTATE_NONE)
  300. ASSERT(!o->process_is_finished)
  301. if (!strcmp(name, "data")) {
  302. *out = NCDVal_NewCopy(mem, o->process_reply_data);
  303. if (NCDVal_IsInvalid(*out)) {
  304. ModuleLog(o->i, BLOG_ERROR, "NCDVal_NewCopy failed");
  305. }
  306. return 1;
  307. }
  308. return 0;
  309. }
  310. static void request_gone (struct request_instance *o, int is_bad)
  311. {
  312. ASSERT(o->rstate != RRSTATE_GONE_BAD)
  313. ASSERT(o->rstate != RRSTATE_GONE_GOOD)
  314. // remove from requests list
  315. LinkedList0_Remove(&o->client->requests_list, &o->requests_list_node);
  316. // free request
  317. NCDRequestClientRequest_Free(&o->request);
  318. // set state over
  319. o->rstate = (is_bad ? RRSTATE_GONE_BAD : RRSTATE_GONE_GOOD);
  320. }
  321. static void request_terminate_process (struct request_instance *o)
  322. {
  323. ASSERT(o->pstate == RPSTATE_WORKING)
  324. // request process termination
  325. NCDModuleProcess_Terminate(&o->process);
  326. // set process state terminating
  327. o->pstate = RPSTATE_TERMINATING;
  328. }
  329. static void request_die (struct request_instance *o, int is_error)
  330. {
  331. // if we have no process, die right away, else we have to wait for process to terminate
  332. if (o->pstate == RPSTATE_NONE) {
  333. request_instance_free(o, is_error);
  334. return;
  335. }
  336. // release request
  337. if (o->rstate != RRSTATE_GONE_BAD && o->rstate != RRSTATE_GONE_GOOD) {
  338. request_gone(o, 1);
  339. }
  340. // initiate process termination, if needed
  341. if (o->pstate != RPSTATE_TERMINATING) {
  342. request_terminate_process(o);
  343. }
  344. // set dstate
  345. o->dstate = (is_error ? RDSTATE_DYING_ERROR : RDSTATE_DYING);
  346. }
  347. static void request_free_reply (struct request_instance *o, struct reply *r, int have_value)
  348. {
  349. // remove from replies list
  350. LinkedList1_Remove(&o->replies_list, &r->replies_list_node);
  351. // free value
  352. if (have_value) {
  353. NCDValMem_Free(&r->mem);
  354. }
  355. // free structure
  356. free(r);
  357. }
  358. static int request_init_reply_process (struct request_instance *o, NCDValMem reply_mem, NCDValSafeRef reply_data)
  359. {
  360. ASSERT(o->pstate == RPSTATE_NONE)
  361. // set parameters
  362. o->process_is_finished = 0;
  363. o->process_reply_mem = reply_mem;
  364. o->process_reply_data = NCDVal_FromSafe(&o->process_reply_mem, reply_data);
  365. // init process
  366. if (!NCDModuleProcess_Init(&o->process, o->i, o->reply_handler, o->args, o, (NCDModuleProcess_handler_event)request_process_handler_event)) {
  367. ModuleLog(o->i, BLOG_ERROR, "NCDModuleProcess_Init failed");
  368. goto fail0;
  369. }
  370. // set special objects function
  371. NCDModuleProcess_SetSpecialFuncs(&o->process, (NCDModuleProcess_func_getspecialobj)request_process_func_getspecialobj);
  372. // set process state working
  373. o->pstate = RPSTATE_WORKING;
  374. return 1;
  375. fail0:
  376. return 0;
  377. }
  378. static int request_init_finished_process (struct request_instance *o)
  379. {
  380. ASSERT(o->pstate == RPSTATE_NONE)
  381. // set parameters
  382. o->process_is_finished = 1;
  383. // init process
  384. if (!NCDModuleProcess_Init(&o->process, o->i, o->finished_handler, o->args, o, (NCDModuleProcess_handler_event)request_process_handler_event)) {
  385. ModuleLog(o->i, BLOG_ERROR, "NCDModuleProcess_Init failed");
  386. goto fail0;
  387. }
  388. // set special objects function
  389. NCDModuleProcess_SetSpecialFuncs(&o->process, (NCDModuleProcess_func_getspecialobj)request_process_func_getspecialobj);
  390. // set process state working
  391. o->pstate = RPSTATE_WORKING;
  392. return 1;
  393. fail0:
  394. return 0;
  395. }
  396. static int get_connect_addr (struct instance *o, NCDValRef connect_addr_arg, struct NCDRequestClient_addr *out_addr)
  397. {
  398. if (!NCDVal_IsList(connect_addr_arg)) {
  399. goto bad;
  400. }
  401. if (NCDVal_ListCount(connect_addr_arg) < 1) {
  402. goto bad;
  403. }
  404. NCDValRef type_arg = NCDVal_ListGet(connect_addr_arg, 0);
  405. if (!NCDVal_IsStringNoNulls(type_arg)) {
  406. goto bad;
  407. }
  408. const char *type = NCDVal_StringValue(type_arg);
  409. if (!strcmp(type, "unix")) {
  410. NCDValRef socket_path_arg;
  411. if (!NCDVal_ListRead(connect_addr_arg, 2, &type_arg, &socket_path_arg)) {
  412. goto bad;
  413. }
  414. if (!NCDVal_IsStringNoNulls(socket_path_arg)) {
  415. goto bad;
  416. }
  417. *out_addr = NCDREQUESTCLIENT_UNIX_ADDR(NCDVal_StringValue(socket_path_arg));
  418. }
  419. else if (!strcmp(type, "tcp")) {
  420. NCDValRef ip_address_arg;
  421. NCDValRef port_number_arg;
  422. if (!NCDVal_ListRead(connect_addr_arg, 3, &type_arg, &ip_address_arg, &port_number_arg)) {
  423. goto bad;
  424. }
  425. if (!NCDVal_IsStringNoNulls(ip_address_arg) || !NCDVal_IsStringNoNulls(port_number_arg)) {
  426. goto bad;
  427. }
  428. BIPAddr ipaddr;
  429. if (!BIPAddr_Resolve(&ipaddr, (char *)NCDVal_StringValue(ip_address_arg), 1)) {
  430. goto bad;
  431. }
  432. uintmax_t port;
  433. if (!parse_unsigned_integer(NCDVal_StringValue(port_number_arg), &port) || port > UINT16_MAX) {
  434. goto bad;
  435. }
  436. BAddr addr;
  437. BAddr_InitFromIpaddrAndPort(&addr, ipaddr, hton16(port));
  438. *out_addr = NCDREQUESTCLIENT_TCP_ADDR(addr);
  439. }
  440. else {
  441. goto bad;
  442. }
  443. return 1;
  444. bad:
  445. ModuleLog(o->i, BLOG_ERROR, "bad connect address argument");
  446. return 0;
  447. }
  448. static void func_new (NCDModuleInst *i)
  449. {
  450. // allocate structure
  451. struct instance *o = malloc(sizeof(*o));
  452. if (!o) {
  453. ModuleLog(i, BLOG_ERROR, "malloc failed");
  454. goto fail0;
  455. }
  456. o->i = i;
  457. NCDModuleInst_Backend_SetUser(i, o);
  458. // check arguments
  459. NCDValRef connect_addr_arg;
  460. if (!NCDVal_ListRead(i->args, 1, &connect_addr_arg)) {
  461. ModuleLog(o->i, BLOG_ERROR, "wrong arity");
  462. goto fail1;
  463. }
  464. // get address
  465. struct NCDRequestClient_addr addr;
  466. if (!get_connect_addr(o, connect_addr_arg, &addr)) {
  467. goto fail1;
  468. }
  469. // init client
  470. if (!NCDRequestClient_Init(&o->client, addr, i->iparams->reactor, o,
  471. (NCDRequestClient_handler_error)client_handler_error,
  472. (NCDRequestClient_handler_connected)client_handler_connected)) {
  473. ModuleLog(o->i, BLOG_ERROR, "NCDRequestClient_Init failed");
  474. goto fail1;
  475. }
  476. // init requests list
  477. LinkedList0_Init(&o->requests_list);
  478. // set state connecting
  479. o->state = CSTATE_CONNECTING;
  480. return;
  481. fail1:
  482. free(o);
  483. fail0:
  484. NCDModuleInst_Backend_SetError(i);
  485. NCDModuleInst_Backend_Dead(i);
  486. }
  487. static void instance_free (struct instance *o, int with_error)
  488. {
  489. NCDModuleInst *i = o->i;
  490. // deal with requests
  491. LinkedList0Node *ln;
  492. while (ln = LinkedList0_GetFirst(&o->requests_list)) {
  493. struct request_instance *req = UPPER_OBJECT(ln, struct request_instance, requests_list_node);
  494. request_gone(req, 1);
  495. }
  496. // free client
  497. NCDRequestClient_Free(&o->client);
  498. // free structure
  499. free(o);
  500. if (with_error) {
  501. NCDModuleInst_Backend_SetError(i);
  502. }
  503. NCDModuleInst_Backend_Dead(i);
  504. }
  505. static void func_die (void *vo)
  506. {
  507. struct instance *o = vo;
  508. instance_free(o, 0);
  509. }
  510. static void request_func_new (NCDModuleInst *i)
  511. {
  512. // allocate structure
  513. struct request_instance *o = malloc(sizeof(*o));
  514. if (!o) {
  515. ModuleLog(i, BLOG_ERROR, "malloc failed");
  516. goto fail0;
  517. }
  518. o->i = i;
  519. NCDModuleInst_Backend_SetUser(i, o);
  520. // check arguments
  521. NCDValRef request_data_arg;
  522. NCDValRef reply_handler_arg;
  523. NCDValRef finished_handler_arg;
  524. NCDValRef args_arg;
  525. if (!NCDVal_ListRead(i->args, 4, &request_data_arg, &reply_handler_arg, &finished_handler_arg, &args_arg)) {
  526. ModuleLog(o->i, BLOG_ERROR, "wrong arity");
  527. goto fail1;
  528. }
  529. if (!NCDVal_IsStringNoNulls(reply_handler_arg) || !NCDVal_IsStringNoNulls(finished_handler_arg) ||
  530. !NCDVal_IsList(args_arg)
  531. ) {
  532. ModuleLog(o->i, BLOG_ERROR, "wrong type");
  533. goto fail1;
  534. }
  535. o->reply_handler = NCDVal_StringValue(reply_handler_arg);
  536. o->finished_handler = NCDVal_StringValue(finished_handler_arg);
  537. o->args = args_arg;
  538. // get client
  539. struct instance *client = NCDModuleInst_Backend_GetUser((NCDModuleInst *)i->method_user);
  540. o->client = client;
  541. // check client state
  542. if (client->state != CSTATE_CONNECTED) {
  543. ModuleLog(o->i, BLOG_ERROR, "client is not connected");
  544. goto fail1;
  545. }
  546. // convert argument to old format
  547. NCDValue request_data_compat;
  548. if (!NCDValCompat_ValToValue(request_data_arg, &request_data_compat)) {
  549. ModuleLog(o->i, BLOG_ERROR, "NCDValCompat_ValToValue failed");
  550. goto fail1;
  551. }
  552. // init request
  553. if (!NCDRequestClientRequest_Init(&o->request, &client->client, &request_data_compat, o,
  554. (NCDRequestClientRequest_handler_sent)request_handler_sent,
  555. (NCDRequestClientRequest_handler_reply)request_handler_reply,
  556. (NCDRequestClientRequest_handler_finished)request_handler_finished)) {
  557. ModuleLog(o->i, BLOG_ERROR, "NCDRequestClientRequest_Init failed");
  558. NCDValue_Free(&request_data_compat);
  559. goto fail1;
  560. }
  561. NCDValue_Free(&request_data_compat);
  562. // add to requests list
  563. LinkedList0_Prepend(&client->requests_list, &o->requests_list_node);
  564. // init replies list
  565. LinkedList1_Init(&o->replies_list);
  566. // set state
  567. o->rstate = RRSTATE_SENDING_REQUEST;
  568. o->pstate = RPSTATE_NONE;
  569. o->dstate = RDSTATE_NONE;
  570. return;
  571. fail1:
  572. free(o);
  573. fail0:
  574. NCDModuleInst_Backend_SetError(i);
  575. NCDModuleInst_Backend_Dead(i);
  576. }
  577. static void request_instance_free (struct request_instance *o, int with_error)
  578. {
  579. ASSERT(o->pstate == RPSTATE_NONE)
  580. NCDModuleInst *i = o->i;
  581. // free replies
  582. LinkedList1Node *ln;
  583. while (ln = LinkedList1_GetFirst(&o->replies_list)) {
  584. struct reply *r = UPPER_OBJECT(ln, struct reply, replies_list_node);
  585. request_free_reply(o, r, 1);
  586. }
  587. // release request
  588. if (o->rstate != RRSTATE_GONE_BAD && o->rstate != RRSTATE_GONE_GOOD) {
  589. request_gone(o, 1);
  590. }
  591. // free structure
  592. free(o);
  593. if (with_error) {
  594. NCDModuleInst_Backend_SetError(i);
  595. }
  596. NCDModuleInst_Backend_Dead(i);
  597. }
  598. static void request_func_die (void *vo)
  599. {
  600. struct request_instance *o = vo;
  601. request_die(o, 0);
  602. }
  603. static const struct NCDModule modules[] = {
  604. {
  605. .type = "sys.request_client",
  606. .func_new = func_new,
  607. .func_die = func_die
  608. }, {
  609. .type = "sys.request_client::request",
  610. .func_new = request_func_new,
  611. .func_die = request_func_die
  612. }, {
  613. .type = NULL
  614. }
  615. };
  616. const struct NCDModuleGroup ncdmodule_sys_request_client = {
  617. .modules = modules
  618. };