server.c 49 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467
  1. /*
  2. *
  3. * Copyright 2015-2016, Google Inc.
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions are
  8. * met:
  9. *
  10. * * Redistributions of source code must retain the above copyright
  11. * notice, this list of conditions and the following disclaimer.
  12. * * Redistributions in binary form must reproduce the above
  13. * copyright notice, this list of conditions and the following disclaimer
  14. * in the documentation and/or other materials provided with the
  15. * distribution.
  16. * * Neither the name of Google Inc. nor the names of its
  17. * contributors may be used to endorse or promote products derived from
  18. * this software without specific prior written permission.
  19. *
  20. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23. * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  24. * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26. * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27. * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28. * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  30. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31. *
  32. */
  33. #include "src/core/lib/surface/server.h"
  34. #include <limits.h>
  35. #include <stdlib.h>
  36. #include <string.h>
  37. #include <grpc/support/alloc.h>
  38. #include <grpc/support/log.h>
  39. #include <grpc/support/string_util.h>
  40. #include <grpc/support/useful.h>
  41. #include "src/core/lib/channel/channel_args.h"
  42. #include "src/core/lib/channel/connected_channel.h"
  43. #include "src/core/lib/iomgr/iomgr.h"
  44. #include "src/core/lib/support/stack_lockfree.h"
  45. #include "src/core/lib/support/string.h"
  46. #include "src/core/lib/surface/api_trace.h"
  47. #include "src/core/lib/surface/call.h"
  48. #include "src/core/lib/surface/channel.h"
  49. #include "src/core/lib/surface/completion_queue.h"
  50. #include "src/core/lib/surface/init.h"
  51. #include "src/core/lib/transport/metadata.h"
  52. #include "src/core/lib/transport/static_metadata.h"
  53. typedef struct listener {
  54. void *arg;
  55. void (*start)(grpc_exec_ctx *exec_ctx, grpc_server *server, void *arg,
  56. grpc_pollset **pollsets, size_t pollset_count);
  57. void (*destroy)(grpc_exec_ctx *exec_ctx, grpc_server *server, void *arg,
  58. grpc_closure *closure);
  59. struct listener *next;
  60. grpc_closure destroy_done;
  61. } listener;
  62. typedef struct call_data call_data;
  63. typedef struct channel_data channel_data;
  64. typedef struct registered_method registered_method;
  65. typedef enum { BATCH_CALL, REGISTERED_CALL } requested_call_type;
  66. typedef struct requested_call {
  67. requested_call_type type;
  68. void *tag;
  69. grpc_server *server;
  70. grpc_completion_queue *cq_bound_to_call;
  71. grpc_call **call;
  72. grpc_cq_completion completion;
  73. grpc_metadata_array *initial_metadata;
  74. union {
  75. struct {
  76. grpc_call_details *details;
  77. } batch;
  78. struct {
  79. registered_method *registered_method;
  80. gpr_timespec *deadline;
  81. grpc_byte_buffer **optional_payload;
  82. } registered;
  83. } data;
  84. } requested_call;
  85. typedef struct channel_registered_method {
  86. registered_method *server_registered_method;
  87. uint32_t flags;
  88. grpc_mdstr *method;
  89. grpc_mdstr *host;
  90. } channel_registered_method;
  91. struct channel_data {
  92. grpc_server *server;
  93. grpc_connectivity_state connectivity_state;
  94. grpc_channel *channel;
  95. size_t cq_idx;
  96. /* linked list of all channels on a server */
  97. channel_data *next;
  98. channel_data *prev;
  99. channel_registered_method *registered_methods;
  100. uint32_t registered_method_slots;
  101. uint32_t registered_method_max_probes;
  102. grpc_closure finish_destroy_channel_closure;
  103. grpc_closure channel_connectivity_changed;
  104. };
  105. typedef struct shutdown_tag {
  106. void *tag;
  107. grpc_completion_queue *cq;
  108. grpc_cq_completion completion;
  109. } shutdown_tag;
  110. typedef enum {
  111. /* waiting for metadata */
  112. NOT_STARTED,
  113. /* inital metadata read, not flow controlled in yet */
  114. PENDING,
  115. /* flow controlled in, on completion queue */
  116. ACTIVATED,
  117. /* cancelled before being queued */
  118. ZOMBIED
  119. } call_state;
  120. typedef struct request_matcher request_matcher;
  121. struct call_data {
  122. grpc_call *call;
  123. /** protects state */
  124. gpr_mu mu_state;
  125. /** the current state of a call - see call_state */
  126. call_state state;
  127. grpc_mdstr *path;
  128. grpc_mdstr *host;
  129. gpr_timespec deadline;
  130. grpc_completion_queue *cq_new;
  131. grpc_metadata_batch *recv_initial_metadata;
  132. bool recv_idempotent_request;
  133. grpc_metadata_array initial_metadata;
  134. request_matcher *request_matcher;
  135. grpc_byte_buffer *payload;
  136. grpc_closure got_initial_metadata;
  137. grpc_closure server_on_recv_initial_metadata;
  138. grpc_closure kill_zombie_closure;
  139. grpc_closure *on_done_recv_initial_metadata;
  140. grpc_closure publish;
  141. call_data *pending_next;
  142. };
  143. struct request_matcher {
  144. grpc_server *server;
  145. call_data *pending_head;
  146. call_data *pending_tail;
  147. gpr_stack_lockfree **requests_per_cq;
  148. };
  149. struct registered_method {
  150. char *method;
  151. char *host;
  152. grpc_server_register_method_payload_handling payload_handling;
  153. uint32_t flags;
  154. /* one request matcher per method */
  155. request_matcher request_matcher;
  156. registered_method *next;
  157. };
  158. typedef struct {
  159. grpc_channel **channels;
  160. size_t num_channels;
  161. } channel_broadcaster;
  162. struct grpc_server {
  163. grpc_channel_args *channel_args;
  164. grpc_completion_queue **cqs;
  165. grpc_pollset **pollsets;
  166. size_t cq_count;
  167. bool started;
  168. /* The two following mutexes control access to server-state
  169. mu_global controls access to non-call-related state (e.g., channel state)
  170. mu_call controls access to call-related state (e.g., the call lists)
  171. If they are ever required to be nested, you must lock mu_global
  172. before mu_call. This is currently used in shutdown processing
  173. (grpc_server_shutdown_and_notify and maybe_finish_shutdown) */
  174. gpr_mu mu_global; /* mutex for server and channel state */
  175. gpr_mu mu_call; /* mutex for call-specific state */
  176. registered_method *registered_methods;
  177. /** one request matcher for unregistered methods */
  178. request_matcher unregistered_request_matcher;
  179. /** free list of available requested_calls indices */
  180. gpr_stack_lockfree *request_freelist;
  181. /** requested call backing data */
  182. requested_call *requested_calls;
  183. size_t max_requested_calls;
  184. gpr_atm shutdown_flag;
  185. uint8_t shutdown_published;
  186. size_t num_shutdown_tags;
  187. shutdown_tag *shutdown_tags;
  188. channel_data root_channel_data;
  189. listener *listeners;
  190. int listeners_destroyed;
  191. gpr_refcount internal_refcount;
  192. /** when did we print the last shutdown progress message */
  193. gpr_timespec last_shutdown_message_time;
  194. };
  195. #define SERVER_FROM_CALL_ELEM(elem) \
  196. (((channel_data *)(elem)->channel_data)->server)
  197. static void publish_new_rpc(grpc_exec_ctx *exec_ctx, void *calld,
  198. grpc_error *error);
  199. static void fail_call(grpc_exec_ctx *exec_ctx, grpc_server *server,
  200. size_t cq_idx, requested_call *rc, grpc_error *error);
  201. /* Before calling maybe_finish_shutdown, we must hold mu_global and not
  202. hold mu_call */
  203. static void maybe_finish_shutdown(grpc_exec_ctx *exec_ctx, grpc_server *server);
  204. /*
  205. * channel broadcaster
  206. */
  207. /* assumes server locked */
  208. static void channel_broadcaster_init(grpc_server *s, channel_broadcaster *cb) {
  209. channel_data *c;
  210. size_t count = 0;
  211. for (c = s->root_channel_data.next; c != &s->root_channel_data; c = c->next) {
  212. count++;
  213. }
  214. cb->num_channels = count;
  215. cb->channels = gpr_malloc(sizeof(*cb->channels) * cb->num_channels);
  216. count = 0;
  217. for (c = s->root_channel_data.next; c != &s->root_channel_data; c = c->next) {
  218. cb->channels[count++] = c->channel;
  219. GRPC_CHANNEL_INTERNAL_REF(c->channel, "broadcast");
  220. }
  221. }
  222. struct shutdown_cleanup_args {
  223. grpc_closure closure;
  224. gpr_slice slice;
  225. };
  226. static void shutdown_cleanup(grpc_exec_ctx *exec_ctx, void *arg,
  227. grpc_error *error) {
  228. struct shutdown_cleanup_args *a = arg;
  229. gpr_slice_unref(a->slice);
  230. gpr_free(a);
  231. }
  232. static void send_shutdown(grpc_exec_ctx *exec_ctx, grpc_channel *channel,
  233. int send_goaway, grpc_error *send_disconnect) {
  234. grpc_transport_op op;
  235. struct shutdown_cleanup_args *sc;
  236. grpc_channel_element *elem;
  237. memset(&op, 0, sizeof(op));
  238. op.send_goaway = send_goaway;
  239. sc = gpr_malloc(sizeof(*sc));
  240. sc->slice = gpr_slice_from_copied_string("Server shutdown");
  241. op.goaway_message = &sc->slice;
  242. op.goaway_status = GRPC_STATUS_OK;
  243. op.disconnect_with_error = send_disconnect;
  244. grpc_closure_init(&sc->closure, shutdown_cleanup, sc);
  245. op.on_consumed = &sc->closure;
  246. elem = grpc_channel_stack_element(grpc_channel_get_channel_stack(channel), 0);
  247. elem->filter->start_transport_op(exec_ctx, elem, &op);
  248. }
  249. static void channel_broadcaster_shutdown(grpc_exec_ctx *exec_ctx,
  250. channel_broadcaster *cb,
  251. int send_goaway,
  252. grpc_error *force_disconnect) {
  253. size_t i;
  254. for (i = 0; i < cb->num_channels; i++) {
  255. send_shutdown(exec_ctx, cb->channels[i], send_goaway,
  256. GRPC_ERROR_REF(force_disconnect));
  257. GRPC_CHANNEL_INTERNAL_UNREF(exec_ctx, cb->channels[i], "broadcast");
  258. }
  259. gpr_free(cb->channels);
  260. GRPC_ERROR_UNREF(force_disconnect);
  261. }
  262. /*
  263. * request_matcher
  264. */
  265. static void request_matcher_init(request_matcher *rm, size_t entries,
  266. grpc_server *server) {
  267. memset(rm, 0, sizeof(*rm));
  268. rm->server = server;
  269. rm->requests_per_cq =
  270. gpr_malloc(sizeof(*rm->requests_per_cq) * server->cq_count);
  271. for (size_t i = 0; i < server->cq_count; i++) {
  272. rm->requests_per_cq[i] = gpr_stack_lockfree_create(entries);
  273. }
  274. }
  275. static void request_matcher_destroy(request_matcher *rm) {
  276. for (size_t i = 0; i < rm->server->cq_count; i++) {
  277. GPR_ASSERT(gpr_stack_lockfree_pop(rm->requests_per_cq[i]) == -1);
  278. gpr_stack_lockfree_destroy(rm->requests_per_cq[i]);
  279. }
  280. gpr_free(rm->requests_per_cq);
  281. }
  282. static void kill_zombie(grpc_exec_ctx *exec_ctx, void *elem,
  283. grpc_error *error) {
  284. grpc_call_destroy(grpc_call_from_top_element(elem));
  285. }
  286. static void request_matcher_zombify_all_pending_calls(grpc_exec_ctx *exec_ctx,
  287. request_matcher *rm) {
  288. while (rm->pending_head) {
  289. call_data *calld = rm->pending_head;
  290. rm->pending_head = calld->pending_next;
  291. gpr_mu_lock(&calld->mu_state);
  292. calld->state = ZOMBIED;
  293. gpr_mu_unlock(&calld->mu_state);
  294. grpc_closure_init(
  295. &calld->kill_zombie_closure, kill_zombie,
  296. grpc_call_stack_element(grpc_call_get_call_stack(calld->call), 0));
  297. grpc_exec_ctx_push(exec_ctx, &calld->kill_zombie_closure, GRPC_ERROR_NONE,
  298. NULL);
  299. }
  300. }
  301. static void request_matcher_kill_requests(grpc_exec_ctx *exec_ctx,
  302. grpc_server *server,
  303. request_matcher *rm,
  304. grpc_error *error) {
  305. int request_id;
  306. for (size_t i = 0; i < server->cq_count; i++) {
  307. while ((request_id = gpr_stack_lockfree_pop(rm->requests_per_cq[i])) !=
  308. -1) {
  309. fail_call(exec_ctx, server, i, &server->requested_calls[request_id], GRPC_ERROR_REF(error));
  310. }
  311. }
  312. GRPC_ERROR_UNREF(error);
  313. }
  314. /*
  315. * server proper
  316. */
  317. static void server_ref(grpc_server *server) {
  318. gpr_ref(&server->internal_refcount);
  319. }
  320. static void server_delete(grpc_exec_ctx *exec_ctx, grpc_server *server) {
  321. registered_method *rm;
  322. size_t i;
  323. grpc_channel_args_destroy(server->channel_args);
  324. gpr_mu_destroy(&server->mu_global);
  325. gpr_mu_destroy(&server->mu_call);
  326. while ((rm = server->registered_methods) != NULL) {
  327. server->registered_methods = rm->next;
  328. if (server->started) {
  329. request_matcher_destroy(&rm->request_matcher);
  330. }
  331. gpr_free(rm->method);
  332. gpr_free(rm->host);
  333. gpr_free(rm);
  334. }
  335. if (server->started) {
  336. request_matcher_destroy(&server->unregistered_request_matcher);
  337. }
  338. for (i = 0; i < server->cq_count; i++) {
  339. GRPC_CQ_INTERNAL_UNREF(server->cqs[i], "server");
  340. }
  341. gpr_stack_lockfree_destroy(server->request_freelist);
  342. gpr_free(server->cqs);
  343. gpr_free(server->pollsets);
  344. gpr_free(server->shutdown_tags);
  345. gpr_free(server->requested_calls);
  346. gpr_free(server);
  347. }
  348. static void server_unref(grpc_exec_ctx *exec_ctx, grpc_server *server) {
  349. if (gpr_unref(&server->internal_refcount)) {
  350. server_delete(exec_ctx, server);
  351. }
  352. }
  353. static int is_channel_orphaned(channel_data *chand) {
  354. return chand->next == chand;
  355. }
  356. static void orphan_channel(channel_data *chand) {
  357. chand->next->prev = chand->prev;
  358. chand->prev->next = chand->next;
  359. chand->next = chand->prev = chand;
  360. }
  361. static void finish_destroy_channel(grpc_exec_ctx *exec_ctx, void *cd,
  362. grpc_error *error) {
  363. channel_data *chand = cd;
  364. grpc_server *server = chand->server;
  365. GRPC_CHANNEL_INTERNAL_UNREF(exec_ctx, chand->channel, "server");
  366. server_unref(exec_ctx, server);
  367. }
  368. static void destroy_channel(grpc_exec_ctx *exec_ctx, channel_data *chand) {
  369. if (is_channel_orphaned(chand)) return;
  370. GPR_ASSERT(chand->server != NULL);
  371. orphan_channel(chand);
  372. server_ref(chand->server);
  373. maybe_finish_shutdown(exec_ctx, chand->server);
  374. chand->finish_destroy_channel_closure.cb = finish_destroy_channel;
  375. chand->finish_destroy_channel_closure.cb_arg = chand;
  376. grpc_transport_op op;
  377. memset(&op, 0, sizeof(op));
  378. op.set_accept_stream = true;
  379. op.on_consumed = &chand->finish_destroy_channel_closure;
  380. grpc_channel_next_op(exec_ctx,
  381. grpc_channel_stack_element(
  382. grpc_channel_get_channel_stack(chand->channel), 0),
  383. &op);
  384. }
  385. static void cpstr(char **dest, size_t *capacity, grpc_mdstr *value) {
  386. gpr_slice slice = value->slice;
  387. size_t len = GPR_SLICE_LENGTH(slice);
  388. if (len + 1 > *capacity) {
  389. *capacity = GPR_MAX(len + 1, *capacity * 2);
  390. *dest = gpr_realloc(*dest, *capacity);
  391. }
  392. memcpy(*dest, grpc_mdstr_as_c_string(value), len + 1);
  393. }
  394. static void done_request_event(grpc_exec_ctx *exec_ctx, void *req,
  395. grpc_cq_completion *c) {
  396. requested_call *rc = req;
  397. grpc_server *server = rc->server;
  398. if (rc >= server->requested_calls &&
  399. rc < server->requested_calls + server->max_requested_calls) {
  400. GPR_ASSERT(rc - server->requested_calls <= INT_MAX);
  401. gpr_stack_lockfree_push(server->request_freelist,
  402. (int)(rc - server->requested_calls));
  403. } else {
  404. gpr_free(req);
  405. }
  406. server_unref(exec_ctx, server);
  407. }
  408. static void publish_call(grpc_exec_ctx *exec_ctx, grpc_server *server,
  409. call_data *calld, size_t cq_idx, requested_call *rc) {
  410. grpc_call_set_completion_queue(exec_ctx, calld->call, rc->cq_bound_to_call);
  411. grpc_call *call = calld->call;
  412. *rc->call = call;
  413. calld->cq_new = server->cqs[cq_idx];
  414. GPR_SWAP(grpc_metadata_array, *rc->initial_metadata, calld->initial_metadata);
  415. switch (rc->type) {
  416. case BATCH_CALL:
  417. GPR_ASSERT(calld->host != NULL);
  418. GPR_ASSERT(calld->path != NULL);
  419. cpstr(&rc->data.batch.details->host,
  420. &rc->data.batch.details->host_capacity, calld->host);
  421. cpstr(&rc->data.batch.details->method,
  422. &rc->data.batch.details->method_capacity, calld->path);
  423. rc->data.batch.details->deadline = calld->deadline;
  424. rc->data.batch.details->flags =
  425. 0 | (calld->recv_idempotent_request
  426. ? GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST
  427. : 0);
  428. break;
  429. case REGISTERED_CALL:
  430. *rc->data.registered.deadline = calld->deadline;
  431. if (rc->data.registered.optional_payload) {
  432. *rc->data.registered.optional_payload = calld->payload;
  433. }
  434. break;
  435. default:
  436. GPR_UNREACHABLE_CODE(return );
  437. }
  438. grpc_call_element *elem =
  439. grpc_call_stack_element(grpc_call_get_call_stack(call), 0);
  440. channel_data *chand = elem->channel_data;
  441. server_ref(chand->server);
  442. grpc_cq_end_op(exec_ctx, calld->cq_new, rc->tag, GRPC_ERROR_NONE,
  443. done_request_event, rc, &rc->completion);
  444. }
  445. static void publish_new_rpc(grpc_exec_ctx *exec_ctx, void *arg,
  446. grpc_error *error) {
  447. grpc_call_element *call_elem = arg;
  448. call_data *calld = call_elem->call_data;
  449. channel_data *chand = call_elem->channel_data;
  450. request_matcher *rm = calld->request_matcher;
  451. grpc_server *server = rm->server;
  452. if (error != GRPC_ERROR_NONE || gpr_atm_acq_load(&server->shutdown_flag)) {
  453. gpr_mu_lock(&calld->mu_state);
  454. calld->state = ZOMBIED;
  455. gpr_mu_unlock(&calld->mu_state);
  456. grpc_closure_init(
  457. &calld->kill_zombie_closure, kill_zombie,
  458. grpc_call_stack_element(grpc_call_get_call_stack(calld->call), 0));
  459. grpc_exec_ctx_push(exec_ctx, &calld->kill_zombie_closure, error, NULL);
  460. return;
  461. }
  462. for (size_t i = 0; i < server->cq_count; i++) {
  463. size_t cq_idx = (chand->cq_idx + i) % server->cq_count;
  464. int request_id = gpr_stack_lockfree_pop(rm->requests_per_cq[cq_idx]);
  465. if (request_id == -1) {
  466. continue;
  467. } else {
  468. gpr_mu_lock(&calld->mu_state);
  469. calld->state = ACTIVATED;
  470. gpr_mu_unlock(&calld->mu_state);
  471. publish_call(exec_ctx, server, calld, cq_idx,
  472. &server->requested_calls[request_id]);
  473. return; /* early out */
  474. }
  475. }
  476. /* no cq to take the request found: queue it on the slow list */
  477. gpr_mu_lock(&server->mu_call);
  478. gpr_mu_lock(&calld->mu_state);
  479. calld->state = PENDING;
  480. gpr_mu_unlock(&calld->mu_state);
  481. if (rm->pending_head == NULL) {
  482. rm->pending_tail = rm->pending_head = calld;
  483. } else {
  484. rm->pending_tail->pending_next = calld;
  485. rm->pending_tail = calld;
  486. }
  487. calld->pending_next = NULL;
  488. gpr_mu_unlock(&server->mu_call);
  489. }
  490. static void finish_start_new_rpc(
  491. grpc_exec_ctx *exec_ctx, grpc_server *server, grpc_call_element *elem,
  492. request_matcher *rm,
  493. grpc_server_register_method_payload_handling payload_handling) {
  494. call_data *calld = elem->call_data;
  495. if (gpr_atm_acq_load(&server->shutdown_flag)) {
  496. gpr_mu_lock(&calld->mu_state);
  497. calld->state = ZOMBIED;
  498. gpr_mu_unlock(&calld->mu_state);
  499. grpc_closure_init(&calld->kill_zombie_closure, kill_zombie, elem);
  500. grpc_exec_ctx_push(exec_ctx, &calld->kill_zombie_closure, GRPC_ERROR_NONE,
  501. NULL);
  502. return;
  503. }
  504. calld->request_matcher = rm;
  505. switch (payload_handling) {
  506. case GRPC_SRM_PAYLOAD_NONE:
  507. publish_new_rpc(exec_ctx, elem, GRPC_ERROR_NONE);
  508. break;
  509. case GRPC_SRM_PAYLOAD_READ_INITIAL_BYTE_BUFFER: {
  510. grpc_op op;
  511. memset(&op, 0, sizeof(op));
  512. op.op = GRPC_OP_RECV_MESSAGE;
  513. op.data.recv_message = &calld->payload;
  514. grpc_closure_init(&calld->publish, publish_new_rpc, elem);
  515. grpc_call_start_batch_and_execute(exec_ctx, calld->call, &op, 1,
  516. &calld->publish);
  517. break;
  518. }
  519. }
  520. }
  521. static void start_new_rpc(grpc_exec_ctx *exec_ctx, grpc_call_element *elem) {
  522. channel_data *chand = elem->channel_data;
  523. call_data *calld = elem->call_data;
  524. grpc_server *server = chand->server;
  525. uint32_t i;
  526. uint32_t hash;
  527. channel_registered_method *rm;
  528. if (chand->registered_methods && calld->path && calld->host) {
  529. /* TODO(ctiller): unify these two searches */
  530. /* check for an exact match with host */
  531. hash = GRPC_MDSTR_KV_HASH(calld->host->hash, calld->path->hash);
  532. for (i = 0; i <= chand->registered_method_max_probes; i++) {
  533. rm = &chand->registered_methods[(hash + i) %
  534. chand->registered_method_slots];
  535. if (!rm) break;
  536. if (rm->host != calld->host) continue;
  537. if (rm->method != calld->path) continue;
  538. if ((rm->flags & GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST) &&
  539. !calld->recv_idempotent_request)
  540. continue;
  541. finish_start_new_rpc(exec_ctx, server, elem,
  542. &rm->server_registered_method->request_matcher,
  543. rm->server_registered_method->payload_handling);
  544. return;
  545. }
  546. /* check for a wildcard method definition (no host set) */
  547. hash = GRPC_MDSTR_KV_HASH(0, calld->path->hash);
  548. for (i = 0; i <= chand->registered_method_max_probes; i++) {
  549. rm = &chand->registered_methods[(hash + i) %
  550. chand->registered_method_slots];
  551. if (!rm) break;
  552. if (rm->host != NULL) continue;
  553. if (rm->method != calld->path) continue;
  554. if ((rm->flags & GRPC_INITIAL_METADATA_IDEMPOTENT_REQUEST) &&
  555. !calld->recv_idempotent_request)
  556. continue;
  557. finish_start_new_rpc(exec_ctx, server, elem,
  558. &rm->server_registered_method->request_matcher,
  559. rm->server_registered_method->payload_handling);
  560. return;
  561. }
  562. }
  563. finish_start_new_rpc(exec_ctx, server, elem,
  564. &server->unregistered_request_matcher,
  565. GRPC_SRM_PAYLOAD_NONE);
  566. }
  567. static int num_listeners(grpc_server *server) {
  568. listener *l;
  569. int n = 0;
  570. for (l = server->listeners; l; l = l->next) {
  571. n++;
  572. }
  573. return n;
  574. }
  575. static void done_shutdown_event(grpc_exec_ctx *exec_ctx, void *server,
  576. grpc_cq_completion *completion) {
  577. server_unref(exec_ctx, server);
  578. }
  579. static int num_channels(grpc_server *server) {
  580. channel_data *chand;
  581. int n = 0;
  582. for (chand = server->root_channel_data.next;
  583. chand != &server->root_channel_data; chand = chand->next) {
  584. n++;
  585. }
  586. return n;
  587. }
  588. static void kill_pending_work_locked(grpc_exec_ctx *exec_ctx,
  589. grpc_server *server, grpc_error *error) {
  590. if (server->started) {
  591. request_matcher_kill_requests(exec_ctx, server,
  592. &server->unregistered_request_matcher, GRPC_ERROR_REF(error));
  593. request_matcher_zombify_all_pending_calls(
  594. exec_ctx, &server->unregistered_request_matcher);
  595. for (registered_method *rm = server->registered_methods; rm;
  596. rm = rm->next) {
  597. request_matcher_kill_requests(exec_ctx, server, &rm->request_matcher, GRPC_ERROR_REF(error));
  598. request_matcher_zombify_all_pending_calls(exec_ctx, &rm->request_matcher);
  599. }
  600. }
  601. GRPC_ERROR_UNREF(error);
  602. }
  603. static void maybe_finish_shutdown(grpc_exec_ctx *exec_ctx,
  604. grpc_server *server) {
  605. size_t i;
  606. if (!gpr_atm_acq_load(&server->shutdown_flag) || server->shutdown_published) {
  607. return;
  608. }
  609. kill_pending_work_locked(exec_ctx, server,
  610. GRPC_ERROR_CREATE("Server Shutdown"));
  611. if (server->root_channel_data.next != &server->root_channel_data ||
  612. server->listeners_destroyed < num_listeners(server)) {
  613. if (gpr_time_cmp(gpr_time_sub(gpr_now(GPR_CLOCK_REALTIME),
  614. server->last_shutdown_message_time),
  615. gpr_time_from_seconds(1, GPR_TIMESPAN)) >= 0) {
  616. server->last_shutdown_message_time = gpr_now(GPR_CLOCK_REALTIME);
  617. gpr_log(GPR_DEBUG,
  618. "Waiting for %d channels and %d/%d listeners to be destroyed"
  619. " before shutting down server",
  620. num_channels(server),
  621. num_listeners(server) - server->listeners_destroyed,
  622. num_listeners(server));
  623. }
  624. return;
  625. }
  626. server->shutdown_published = 1;
  627. for (i = 0; i < server->num_shutdown_tags; i++) {
  628. server_ref(server);
  629. grpc_cq_end_op(exec_ctx, server->shutdown_tags[i].cq,
  630. server->shutdown_tags[i].tag, GRPC_ERROR_NONE,
  631. done_shutdown_event, server,
  632. &server->shutdown_tags[i].completion);
  633. }
  634. }
  635. static grpc_mdelem *server_filter(void *user_data, grpc_mdelem *md) {
  636. grpc_call_element *elem = user_data;
  637. call_data *calld = elem->call_data;
  638. if (md->key == GRPC_MDSTR_PATH) {
  639. if (calld->path == NULL) {
  640. calld->path = GRPC_MDSTR_REF(md->value);
  641. }
  642. return NULL;
  643. } else if (md->key == GRPC_MDSTR_AUTHORITY) {
  644. if (calld->host == NULL) {
  645. calld->host = GRPC_MDSTR_REF(md->value);
  646. }
  647. return NULL;
  648. }
  649. return md;
  650. }
  651. static void server_on_recv_initial_metadata(grpc_exec_ctx *exec_ctx, void *ptr,
  652. grpc_error *error) {
  653. grpc_call_element *elem = ptr;
  654. call_data *calld = elem->call_data;
  655. gpr_timespec op_deadline;
  656. GRPC_ERROR_REF(error);
  657. grpc_metadata_batch_filter(calld->recv_initial_metadata, server_filter, elem);
  658. op_deadline = calld->recv_initial_metadata->deadline;
  659. if (0 != gpr_time_cmp(op_deadline, gpr_inf_future(op_deadline.clock_type))) {
  660. calld->deadline = op_deadline;
  661. }
  662. if (calld->host && calld->path) {
  663. /* do nothing */
  664. } else {
  665. GRPC_ERROR_UNREF(error);
  666. error =
  667. GRPC_ERROR_CREATE_REFERENCING("Missing :authority or :path", &error, 1);
  668. }
  669. grpc_exec_ctx_push(exec_ctx, calld->on_done_recv_initial_metadata, error,
  670. NULL);
  671. }
  672. static void server_mutate_op(grpc_call_element *elem,
  673. grpc_transport_stream_op *op) {
  674. call_data *calld = elem->call_data;
  675. if (op->recv_initial_metadata != NULL) {
  676. GPR_ASSERT(op->recv_idempotent_request == NULL);
  677. calld->recv_initial_metadata = op->recv_initial_metadata;
  678. calld->on_done_recv_initial_metadata = op->recv_initial_metadata_ready;
  679. op->recv_initial_metadata_ready = &calld->server_on_recv_initial_metadata;
  680. op->recv_idempotent_request = &calld->recv_idempotent_request;
  681. }
  682. }
  683. static void server_start_transport_stream_op(grpc_exec_ctx *exec_ctx,
  684. grpc_call_element *elem,
  685. grpc_transport_stream_op *op) {
  686. GRPC_CALL_LOG_OP(GPR_INFO, elem, op);
  687. server_mutate_op(elem, op);
  688. grpc_call_next_op(exec_ctx, elem, op);
  689. }
  690. static void got_initial_metadata(grpc_exec_ctx *exec_ctx, void *ptr,
  691. grpc_error *error) {
  692. grpc_call_element *elem = ptr;
  693. call_data *calld = elem->call_data;
  694. if (error == GRPC_ERROR_NONE) {
  695. start_new_rpc(exec_ctx, elem);
  696. } else {
  697. gpr_mu_lock(&calld->mu_state);
  698. if (calld->state == NOT_STARTED) {
  699. calld->state = ZOMBIED;
  700. gpr_mu_unlock(&calld->mu_state);
  701. grpc_closure_init(&calld->kill_zombie_closure, kill_zombie, elem);
  702. grpc_exec_ctx_push(exec_ctx, &calld->kill_zombie_closure, GRPC_ERROR_NONE,
  703. NULL);
  704. } else if (calld->state == PENDING) {
  705. calld->state = ZOMBIED;
  706. gpr_mu_unlock(&calld->mu_state);
  707. /* zombied call will be destroyed when it's removed from the pending
  708. queue... later */
  709. } else {
  710. gpr_mu_unlock(&calld->mu_state);
  711. }
  712. }
  713. }
  714. static void accept_stream(grpc_exec_ctx *exec_ctx, void *cd,
  715. grpc_transport *transport,
  716. const void *transport_server_data) {
  717. channel_data *chand = cd;
  718. /* create a call */
  719. grpc_call *call =
  720. grpc_call_create(chand->channel, NULL, 0, NULL, transport_server_data,
  721. NULL, 0, gpr_inf_future(GPR_CLOCK_MONOTONIC));
  722. grpc_call_element *elem =
  723. grpc_call_stack_element(grpc_call_get_call_stack(call), 0);
  724. call_data *calld = elem->call_data;
  725. grpc_op op;
  726. memset(&op, 0, sizeof(op));
  727. op.op = GRPC_OP_RECV_INITIAL_METADATA;
  728. op.data.recv_initial_metadata = &calld->initial_metadata;
  729. grpc_closure_init(&calld->got_initial_metadata, got_initial_metadata, elem);
  730. grpc_call_start_batch_and_execute(exec_ctx, call, &op, 1,
  731. &calld->got_initial_metadata);
  732. }
  733. static void channel_connectivity_changed(grpc_exec_ctx *exec_ctx, void *cd,
  734. grpc_error *error) {
  735. channel_data *chand = cd;
  736. grpc_server *server = chand->server;
  737. if (chand->connectivity_state != GRPC_CHANNEL_FATAL_FAILURE) {
  738. grpc_transport_op op;
  739. memset(&op, 0, sizeof(op));
  740. op.on_connectivity_state_change = &chand->channel_connectivity_changed,
  741. op.connectivity_state = &chand->connectivity_state;
  742. grpc_channel_next_op(exec_ctx,
  743. grpc_channel_stack_element(
  744. grpc_channel_get_channel_stack(chand->channel), 0),
  745. &op);
  746. } else {
  747. gpr_mu_lock(&server->mu_global);
  748. destroy_channel(exec_ctx, chand);
  749. gpr_mu_unlock(&server->mu_global);
  750. GRPC_CHANNEL_INTERNAL_UNREF(exec_ctx, chand->channel, "connectivity");
  751. }
  752. }
  753. static void init_call_elem(grpc_exec_ctx *exec_ctx, grpc_call_element *elem,
  754. grpc_call_element_args *args) {
  755. call_data *calld = elem->call_data;
  756. channel_data *chand = elem->channel_data;
  757. memset(calld, 0, sizeof(call_data));
  758. calld->deadline = gpr_inf_future(GPR_CLOCK_REALTIME);
  759. calld->call = grpc_call_from_top_element(elem);
  760. gpr_mu_init(&calld->mu_state);
  761. grpc_closure_init(&calld->server_on_recv_initial_metadata,
  762. server_on_recv_initial_metadata, elem);
  763. server_ref(chand->server);
  764. }
  765. static void destroy_call_elem(grpc_exec_ctx *exec_ctx, grpc_call_element *elem,
  766. void *ignored) {
  767. channel_data *chand = elem->channel_data;
  768. call_data *calld = elem->call_data;
  769. GPR_ASSERT(calld->state != PENDING);
  770. if (calld->host) {
  771. GRPC_MDSTR_UNREF(calld->host);
  772. }
  773. if (calld->path) {
  774. GRPC_MDSTR_UNREF(calld->path);
  775. }
  776. grpc_metadata_array_destroy(&calld->initial_metadata);
  777. gpr_mu_destroy(&calld->mu_state);
  778. server_unref(exec_ctx, chand->server);
  779. }
  780. static void init_channel_elem(grpc_exec_ctx *exec_ctx,
  781. grpc_channel_element *elem,
  782. grpc_channel_element_args *args) {
  783. channel_data *chand = elem->channel_data;
  784. GPR_ASSERT(args->is_first);
  785. GPR_ASSERT(!args->is_last);
  786. chand->server = NULL;
  787. chand->channel = NULL;
  788. chand->next = chand->prev = chand;
  789. chand->registered_methods = NULL;
  790. chand->connectivity_state = GRPC_CHANNEL_IDLE;
  791. grpc_closure_init(&chand->channel_connectivity_changed,
  792. channel_connectivity_changed, chand);
  793. }
  794. static void destroy_channel_elem(grpc_exec_ctx *exec_ctx,
  795. grpc_channel_element *elem) {
  796. size_t i;
  797. channel_data *chand = elem->channel_data;
  798. if (chand->registered_methods) {
  799. for (i = 0; i < chand->registered_method_slots; i++) {
  800. if (chand->registered_methods[i].method) {
  801. GRPC_MDSTR_UNREF(chand->registered_methods[i].method);
  802. }
  803. if (chand->registered_methods[i].host) {
  804. GRPC_MDSTR_UNREF(chand->registered_methods[i].host);
  805. }
  806. }
  807. gpr_free(chand->registered_methods);
  808. }
  809. if (chand->server) {
  810. gpr_mu_lock(&chand->server->mu_global);
  811. chand->next->prev = chand->prev;
  812. chand->prev->next = chand->next;
  813. chand->next = chand->prev = chand;
  814. maybe_finish_shutdown(exec_ctx, chand->server);
  815. gpr_mu_unlock(&chand->server->mu_global);
  816. server_unref(exec_ctx, chand->server);
  817. }
  818. }
  819. const grpc_channel_filter grpc_server_top_filter = {
  820. server_start_transport_stream_op,
  821. grpc_channel_next_op,
  822. sizeof(call_data),
  823. init_call_elem,
  824. grpc_call_stack_ignore_set_pollset,
  825. destroy_call_elem,
  826. sizeof(channel_data),
  827. init_channel_elem,
  828. destroy_channel_elem,
  829. grpc_call_next_get_peer,
  830. "server",
  831. };
  832. static void register_completion_queue(grpc_server *server,
  833. grpc_completion_queue *cq,
  834. bool is_non_listening, void *reserved) {
  835. size_t i, n;
  836. GPR_ASSERT(!reserved);
  837. for (i = 0; i < server->cq_count; i++) {
  838. if (server->cqs[i] == cq) return;
  839. }
  840. grpc_cq_mark_server_cq(cq);
  841. if (is_non_listening) {
  842. grpc_cq_mark_non_listening_server_cq(cq);
  843. }
  844. GRPC_CQ_INTERNAL_REF(cq, "server");
  845. n = server->cq_count++;
  846. server->cqs = gpr_realloc(server->cqs,
  847. server->cq_count * sizeof(grpc_completion_queue *));
  848. server->cqs[n] = cq;
  849. }
  850. void grpc_server_register_completion_queue(grpc_server *server,
  851. grpc_completion_queue *cq,
  852. void *reserved) {
  853. GRPC_API_TRACE(
  854. "grpc_server_register_completion_queue(server=%p, cq=%p, reserved=%p)", 3,
  855. (server, cq, reserved));
  856. register_completion_queue(server, cq, false, reserved);
  857. }
  858. void grpc_server_register_non_listening_completion_queue(
  859. grpc_server *server, grpc_completion_queue *cq, void *reserved) {
  860. GRPC_API_TRACE(
  861. "grpc_server_register_non_listening_completion_queue(server=%p, cq=%p, "
  862. "reserved=%p)",
  863. 3, (server, cq, reserved));
  864. register_completion_queue(server, cq, true, reserved);
  865. }
  866. grpc_server *grpc_server_create(const grpc_channel_args *args, void *reserved) {
  867. size_t i;
  868. GRPC_API_TRACE("grpc_server_create(%p, %p)", 2, (args, reserved));
  869. grpc_server *server = gpr_malloc(sizeof(grpc_server));
  870. GPR_ASSERT(grpc_is_initialized() && "call grpc_init()");
  871. memset(server, 0, sizeof(grpc_server));
  872. gpr_mu_init(&server->mu_global);
  873. gpr_mu_init(&server->mu_call);
  874. /* decremented by grpc_server_destroy */
  875. gpr_ref_init(&server->internal_refcount, 1);
  876. server->root_channel_data.next = server->root_channel_data.prev =
  877. &server->root_channel_data;
  878. /* TODO(ctiller): expose a channel_arg for this */
  879. server->max_requested_calls = 32768;
  880. server->request_freelist =
  881. gpr_stack_lockfree_create(server->max_requested_calls);
  882. for (i = 0; i < (size_t)server->max_requested_calls; i++) {
  883. gpr_stack_lockfree_push(server->request_freelist, (int)i);
  884. }
  885. server->requested_calls = gpr_malloc(server->max_requested_calls *
  886. sizeof(*server->requested_calls));
  887. server->channel_args = grpc_channel_args_copy(args);
  888. return server;
  889. }
  890. static int streq(const char *a, const char *b) {
  891. if (a == NULL && b == NULL) return 1;
  892. if (a == NULL) return 0;
  893. if (b == NULL) return 0;
  894. return 0 == strcmp(a, b);
  895. }
  896. void *grpc_server_register_method(
  897. grpc_server *server, const char *method, const char *host,
  898. grpc_server_register_method_payload_handling payload_handling,
  899. uint32_t flags) {
  900. registered_method *m;
  901. GRPC_API_TRACE(
  902. "grpc_server_register_method(server=%p, method=%s, host=%s, "
  903. "flags=0x%08x)",
  904. 4, (server, method, host, flags));
  905. if (!method) {
  906. gpr_log(GPR_ERROR,
  907. "grpc_server_register_method method string cannot be NULL");
  908. return NULL;
  909. }
  910. for (m = server->registered_methods; m; m = m->next) {
  911. if (streq(m->method, method) && streq(m->host, host)) {
  912. gpr_log(GPR_ERROR, "duplicate registration for %s@%s", method,
  913. host ? host : "*");
  914. return NULL;
  915. }
  916. }
  917. if ((flags & ~GRPC_INITIAL_METADATA_USED_MASK) != 0) {
  918. gpr_log(GPR_ERROR, "grpc_server_register_method invalid flags 0x%08x",
  919. flags);
  920. return NULL;
  921. }
  922. m = gpr_malloc(sizeof(registered_method));
  923. memset(m, 0, sizeof(*m));
  924. m->method = gpr_strdup(method);
  925. m->host = gpr_strdup(host);
  926. m->next = server->registered_methods;
  927. m->payload_handling = payload_handling;
  928. m->flags = flags;
  929. server->registered_methods = m;
  930. return m;
  931. }
  932. void grpc_server_start(grpc_server *server) {
  933. listener *l;
  934. size_t i;
  935. grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
  936. GRPC_API_TRACE("grpc_server_start(server=%p)", 1, (server));
  937. server->started = true;
  938. size_t pollset_count = 0;
  939. server->pollsets = gpr_malloc(sizeof(grpc_pollset *) * server->cq_count);
  940. for (i = 0; i < server->cq_count; i++) {
  941. if (!grpc_cq_is_non_listening_server_cq(server->cqs[i])) {
  942. server->pollsets[pollset_count++] = grpc_cq_pollset(server->cqs[i]);
  943. }
  944. }
  945. request_matcher_init(&server->unregistered_request_matcher,
  946. server->max_requested_calls, server);
  947. for (registered_method *rm = server->registered_methods; rm; rm = rm->next) {
  948. request_matcher_init(&rm->request_matcher, server->max_requested_calls,
  949. server);
  950. }
  951. for (l = server->listeners; l; l = l->next) {
  952. l->start(&exec_ctx, server, l->arg, server->pollsets, pollset_count);
  953. }
  954. grpc_exec_ctx_finish(&exec_ctx);
  955. }
  956. void grpc_server_setup_transport(grpc_exec_ctx *exec_ctx, grpc_server *s,
  957. grpc_transport *transport,
  958. grpc_pollset *accepting_pollset,
  959. const grpc_channel_args *args) {
  960. size_t num_registered_methods;
  961. size_t alloc;
  962. registered_method *rm;
  963. channel_registered_method *crm;
  964. grpc_channel *channel;
  965. channel_data *chand;
  966. grpc_mdstr *host;
  967. grpc_mdstr *method;
  968. uint32_t hash;
  969. size_t slots;
  970. uint32_t probes;
  971. uint32_t max_probes = 0;
  972. grpc_transport_op op;
  973. channel =
  974. grpc_channel_create(exec_ctx, NULL, args, GRPC_SERVER_CHANNEL, transport);
  975. chand = (channel_data *)grpc_channel_stack_element(
  976. grpc_channel_get_channel_stack(channel), 0)
  977. ->channel_data;
  978. chand->server = s;
  979. server_ref(s);
  980. chand->channel = channel;
  981. size_t cq_idx;
  982. grpc_completion_queue *accepting_cq = grpc_cq_from_pollset(accepting_pollset);
  983. for (cq_idx = 0; cq_idx < s->cq_count; cq_idx++) {
  984. if (s->cqs[cq_idx] == accepting_cq) break;
  985. }
  986. if (cq_idx == s->cq_count) {
  987. /* completion queue not found: pick a random one to publish new calls to */
  988. cq_idx = (size_t)rand() % s->cq_count;
  989. }
  990. chand->cq_idx = cq_idx;
  991. num_registered_methods = 0;
  992. for (rm = s->registered_methods; rm; rm = rm->next) {
  993. num_registered_methods++;
  994. }
  995. /* build a lookup table phrased in terms of mdstr's in this channels context
  996. to quickly find registered methods */
  997. if (num_registered_methods > 0) {
  998. slots = 2 * num_registered_methods;
  999. alloc = sizeof(channel_registered_method) * slots;
  1000. chand->registered_methods = gpr_malloc(alloc);
  1001. memset(chand->registered_methods, 0, alloc);
  1002. for (rm = s->registered_methods; rm; rm = rm->next) {
  1003. host = rm->host ? grpc_mdstr_from_string(rm->host) : NULL;
  1004. method = grpc_mdstr_from_string(rm->method);
  1005. hash = GRPC_MDSTR_KV_HASH(host ? host->hash : 0, method->hash);
  1006. for (probes = 0; chand->registered_methods[(hash + probes) % slots]
  1007. .server_registered_method != NULL;
  1008. probes++)
  1009. ;
  1010. if (probes > max_probes) max_probes = probes;
  1011. crm = &chand->registered_methods[(hash + probes) % slots];
  1012. crm->server_registered_method = rm;
  1013. crm->flags = rm->flags;
  1014. crm->host = host;
  1015. crm->method = method;
  1016. }
  1017. GPR_ASSERT(slots <= UINT32_MAX);
  1018. chand->registered_method_slots = (uint32_t)slots;
  1019. chand->registered_method_max_probes = max_probes;
  1020. }
  1021. gpr_mu_lock(&s->mu_global);
  1022. chand->next = &s->root_channel_data;
  1023. chand->prev = chand->next->prev;
  1024. chand->next->prev = chand->prev->next = chand;
  1025. gpr_mu_unlock(&s->mu_global);
  1026. GRPC_CHANNEL_INTERNAL_REF(channel, "connectivity");
  1027. memset(&op, 0, sizeof(op));
  1028. op.set_accept_stream = true;
  1029. op.set_accept_stream_fn = accept_stream;
  1030. op.set_accept_stream_user_data = chand;
  1031. op.on_connectivity_state_change = &chand->channel_connectivity_changed;
  1032. op.connectivity_state = &chand->connectivity_state;
  1033. if (gpr_atm_acq_load(&s->shutdown_flag) != 0) {
  1034. op.disconnect_with_error = GRPC_ERROR_CREATE("Server shutdown");
  1035. }
  1036. grpc_transport_perform_op(exec_ctx, transport, &op);
  1037. }
  1038. void done_published_shutdown(grpc_exec_ctx *exec_ctx, void *done_arg,
  1039. grpc_cq_completion *storage) {
  1040. (void)done_arg;
  1041. gpr_free(storage);
  1042. }
  1043. static void listener_destroy_done(grpc_exec_ctx *exec_ctx, void *s,
  1044. grpc_error *error) {
  1045. grpc_server *server = s;
  1046. gpr_mu_lock(&server->mu_global);
  1047. server->listeners_destroyed++;
  1048. maybe_finish_shutdown(exec_ctx, server);
  1049. gpr_mu_unlock(&server->mu_global);
  1050. }
  1051. void grpc_server_shutdown_and_notify(grpc_server *server,
  1052. grpc_completion_queue *cq, void *tag) {
  1053. listener *l;
  1054. shutdown_tag *sdt;
  1055. channel_broadcaster broadcaster;
  1056. grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
  1057. GRPC_API_TRACE("grpc_server_shutdown_and_notify(server=%p, cq=%p, tag=%p)", 3,
  1058. (server, cq, tag));
  1059. /* lock, and gather up some stuff to do */
  1060. gpr_mu_lock(&server->mu_global);
  1061. grpc_cq_begin_op(cq, tag);
  1062. if (server->shutdown_published) {
  1063. grpc_cq_end_op(&exec_ctx, cq, tag, GRPC_ERROR_NONE, done_published_shutdown,
  1064. NULL, gpr_malloc(sizeof(grpc_cq_completion)));
  1065. gpr_mu_unlock(&server->mu_global);
  1066. goto done;
  1067. }
  1068. server->shutdown_tags =
  1069. gpr_realloc(server->shutdown_tags,
  1070. sizeof(shutdown_tag) * (server->num_shutdown_tags + 1));
  1071. sdt = &server->shutdown_tags[server->num_shutdown_tags++];
  1072. sdt->tag = tag;
  1073. sdt->cq = cq;
  1074. if (gpr_atm_acq_load(&server->shutdown_flag)) {
  1075. gpr_mu_unlock(&server->mu_global);
  1076. goto done;
  1077. }
  1078. server->last_shutdown_message_time = gpr_now(GPR_CLOCK_REALTIME);
  1079. channel_broadcaster_init(server, &broadcaster);
  1080. gpr_atm_rel_store(&server->shutdown_flag, 1);
  1081. /* collect all unregistered then registered calls */
  1082. gpr_mu_lock(&server->mu_call);
  1083. kill_pending_work_locked(&exec_ctx, server,
  1084. GRPC_ERROR_CREATE("Server Shutdown"));
  1085. gpr_mu_unlock(&server->mu_call);
  1086. maybe_finish_shutdown(&exec_ctx, server);
  1087. gpr_mu_unlock(&server->mu_global);
  1088. /* Shutdown listeners */
  1089. for (l = server->listeners; l; l = l->next) {
  1090. grpc_closure_init(&l->destroy_done, listener_destroy_done, server);
  1091. l->destroy(&exec_ctx, server, l->arg, &l->destroy_done);
  1092. }
  1093. channel_broadcaster_shutdown(&exec_ctx, &broadcaster, 1, 0);
  1094. done:
  1095. grpc_exec_ctx_finish(&exec_ctx);
  1096. }
  1097. void grpc_server_cancel_all_calls(grpc_server *server) {
  1098. channel_broadcaster broadcaster;
  1099. grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
  1100. GRPC_API_TRACE("grpc_server_cancel_all_calls(server=%p)", 1, (server));
  1101. gpr_mu_lock(&server->mu_global);
  1102. channel_broadcaster_init(server, &broadcaster);
  1103. gpr_mu_unlock(&server->mu_global);
  1104. channel_broadcaster_shutdown(&exec_ctx, &broadcaster, 0,
  1105. GRPC_ERROR_CREATE("Cancelling all calls"));
  1106. grpc_exec_ctx_finish(&exec_ctx);
  1107. }
  1108. void grpc_server_destroy(grpc_server *server) {
  1109. listener *l;
  1110. grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
  1111. GRPC_API_TRACE("grpc_server_destroy(server=%p)", 1, (server));
  1112. gpr_mu_lock(&server->mu_global);
  1113. GPR_ASSERT(gpr_atm_acq_load(&server->shutdown_flag) || !server->listeners);
  1114. GPR_ASSERT(server->listeners_destroyed == num_listeners(server));
  1115. while (server->listeners) {
  1116. l = server->listeners;
  1117. server->listeners = l->next;
  1118. gpr_free(l);
  1119. }
  1120. gpr_mu_unlock(&server->mu_global);
  1121. server_unref(&exec_ctx, server);
  1122. grpc_exec_ctx_finish(&exec_ctx);
  1123. }
  1124. void grpc_server_add_listener(
  1125. grpc_exec_ctx *exec_ctx, grpc_server *server, void *arg,
  1126. void (*start)(grpc_exec_ctx *exec_ctx, grpc_server *server, void *arg,
  1127. grpc_pollset **pollsets, size_t pollset_count),
  1128. void (*destroy)(grpc_exec_ctx *exec_ctx, grpc_server *server, void *arg,
  1129. grpc_closure *on_done)) {
  1130. listener *l = gpr_malloc(sizeof(listener));
  1131. l->arg = arg;
  1132. l->start = start;
  1133. l->destroy = destroy;
  1134. l->next = server->listeners;
  1135. server->listeners = l;
  1136. }
  1137. static grpc_call_error queue_call_request(grpc_exec_ctx *exec_ctx,
  1138. grpc_server *server, size_t cq_idx,
  1139. requested_call *rc) {
  1140. call_data *calld = NULL;
  1141. request_matcher *rm = NULL;
  1142. int request_id;
  1143. if (gpr_atm_acq_load(&server->shutdown_flag)) {
  1144. fail_call(exec_ctx, server, cq_idx, rc,
  1145. GRPC_ERROR_CREATE("Server Shutdown"));
  1146. return GRPC_CALL_OK;
  1147. }
  1148. request_id = gpr_stack_lockfree_pop(server->request_freelist);
  1149. if (request_id == -1) {
  1150. /* out of request ids: just fail this one */
  1151. fail_call(exec_ctx, server, cq_idx, rc,
  1152. GRPC_ERROR_CREATE("Server Shutdown"));
  1153. return GRPC_CALL_OK;
  1154. }
  1155. switch (rc->type) {
  1156. case BATCH_CALL:
  1157. rm = &server->unregistered_request_matcher;
  1158. break;
  1159. case REGISTERED_CALL:
  1160. rm = &rc->data.registered.registered_method->request_matcher;
  1161. break;
  1162. }
  1163. server->requested_calls[request_id] = *rc;
  1164. gpr_free(rc);
  1165. if (gpr_stack_lockfree_push(rm->requests_per_cq[cq_idx], request_id)) {
  1166. /* this was the first queued request: we need to lock and start
  1167. matching calls */
  1168. gpr_mu_lock(&server->mu_call);
  1169. while ((calld = rm->pending_head) != NULL) {
  1170. request_id = gpr_stack_lockfree_pop(rm->requests_per_cq[cq_idx]);
  1171. if (request_id == -1) break;
  1172. rm->pending_head = calld->pending_next;
  1173. gpr_mu_unlock(&server->mu_call);
  1174. gpr_mu_lock(&calld->mu_state);
  1175. if (calld->state == ZOMBIED) {
  1176. gpr_mu_unlock(&calld->mu_state);
  1177. grpc_closure_init(
  1178. &calld->kill_zombie_closure, kill_zombie,
  1179. grpc_call_stack_element(grpc_call_get_call_stack(calld->call), 0));
  1180. grpc_exec_ctx_push(exec_ctx, &calld->kill_zombie_closure,
  1181. GRPC_ERROR_NONE, NULL);
  1182. } else {
  1183. GPR_ASSERT(calld->state == PENDING);
  1184. calld->state = ACTIVATED;
  1185. gpr_mu_unlock(&calld->mu_state);
  1186. publish_call(exec_ctx, server, calld, cq_idx,
  1187. &server->requested_calls[request_id]);
  1188. }
  1189. gpr_mu_lock(&server->mu_call);
  1190. }
  1191. gpr_mu_unlock(&server->mu_call);
  1192. }
  1193. return GRPC_CALL_OK;
  1194. }
  1195. grpc_call_error grpc_server_request_call(
  1196. grpc_server *server, grpc_call **call, grpc_call_details *details,
  1197. grpc_metadata_array *initial_metadata,
  1198. grpc_completion_queue *cq_bound_to_call,
  1199. grpc_completion_queue *cq_for_notification, void *tag) {
  1200. grpc_call_error error;
  1201. grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
  1202. requested_call *rc = gpr_malloc(sizeof(*rc));
  1203. GRPC_API_TRACE(
  1204. "grpc_server_request_call("
  1205. "server=%p, call=%p, details=%p, initial_metadata=%p, "
  1206. "cq_bound_to_call=%p, cq_for_notification=%p, tag=%p)",
  1207. 7, (server, call, details, initial_metadata, cq_bound_to_call,
  1208. cq_for_notification, tag));
  1209. size_t cq_idx;
  1210. for (cq_idx = 0; cq_idx < server->cq_count; cq_idx++) {
  1211. if (server->cqs[cq_idx] == cq_for_notification) {
  1212. break;
  1213. }
  1214. }
  1215. if (cq_idx == server->cq_count) {
  1216. gpr_free(rc);
  1217. error = GRPC_CALL_ERROR_NOT_SERVER_COMPLETION_QUEUE;
  1218. goto done;
  1219. }
  1220. grpc_cq_begin_op(cq_for_notification, tag);
  1221. details->reserved = NULL;
  1222. rc->type = BATCH_CALL;
  1223. rc->server = server;
  1224. rc->tag = tag;
  1225. rc->cq_bound_to_call = cq_bound_to_call;
  1226. rc->call = call;
  1227. rc->data.batch.details = details;
  1228. rc->initial_metadata = initial_metadata;
  1229. error = queue_call_request(&exec_ctx, server, cq_idx, rc);
  1230. done:
  1231. grpc_exec_ctx_finish(&exec_ctx);
  1232. return error;
  1233. }
  1234. grpc_call_error grpc_server_request_registered_call(
  1235. grpc_server *server, void *rmp, grpc_call **call, gpr_timespec *deadline,
  1236. grpc_metadata_array *initial_metadata, grpc_byte_buffer **optional_payload,
  1237. grpc_completion_queue *cq_bound_to_call,
  1238. grpc_completion_queue *cq_for_notification, void *tag) {
  1239. grpc_call_error error;
  1240. grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
  1241. requested_call *rc = gpr_malloc(sizeof(*rc));
  1242. registered_method *rm = rmp;
  1243. GRPC_API_TRACE(
  1244. "grpc_server_request_registered_call("
  1245. "server=%p, rmp=%p, call=%p, deadline=%p, initial_metadata=%p, "
  1246. "optional_payload=%p, cq_bound_to_call=%p, cq_for_notification=%p, "
  1247. "tag=%p)",
  1248. 9, (server, rmp, call, deadline, initial_metadata, optional_payload,
  1249. cq_bound_to_call, cq_for_notification, tag));
  1250. size_t cq_idx;
  1251. for (cq_idx = 0; cq_idx < server->cq_count; cq_idx++) {
  1252. if (server->cqs[cq_idx] == cq_for_notification) {
  1253. break;
  1254. }
  1255. }
  1256. if (cq_idx == server->cq_count) {
  1257. gpr_free(rc);
  1258. error = GRPC_CALL_ERROR_NOT_SERVER_COMPLETION_QUEUE;
  1259. goto done;
  1260. }
  1261. if ((optional_payload == NULL) !=
  1262. (rm->payload_handling == GRPC_SRM_PAYLOAD_NONE)) {
  1263. gpr_free(rc);
  1264. error = GRPC_CALL_ERROR_PAYLOAD_TYPE_MISMATCH;
  1265. goto done;
  1266. }
  1267. grpc_cq_begin_op(cq_for_notification, tag);
  1268. rc->type = REGISTERED_CALL;
  1269. rc->server = server;
  1270. rc->tag = tag;
  1271. rc->cq_bound_to_call = cq_bound_to_call;
  1272. rc->call = call;
  1273. rc->data.registered.registered_method = rm;
  1274. rc->data.registered.deadline = deadline;
  1275. rc->initial_metadata = initial_metadata;
  1276. rc->data.registered.optional_payload = optional_payload;
  1277. error = queue_call_request(&exec_ctx, server, cq_idx, rc);
  1278. done:
  1279. grpc_exec_ctx_finish(&exec_ctx);
  1280. return error;
  1281. }
  1282. static void fail_call(grpc_exec_ctx *exec_ctx, grpc_server *server,
  1283. size_t cq_idx, requested_call *rc, grpc_error *error) {
  1284. *rc->call = NULL;
  1285. rc->initial_metadata->count = 0;
  1286. GPR_ASSERT(error != GRPC_ERROR_NONE);
  1287. server_ref(server);
  1288. grpc_cq_end_op(exec_ctx, server->cqs[cq_idx], rc->tag, error,
  1289. done_request_event, rc, &rc->completion);
  1290. }
  1291. const grpc_channel_args *grpc_server_get_channel_args(grpc_server *server) {
  1292. return server->channel_args;
  1293. }
  1294. int grpc_server_has_open_connections(grpc_server *server) {
  1295. int r;
  1296. gpr_mu_lock(&server->mu_global);
  1297. r = server->root_channel_data.next != &server->root_channel_data;
  1298. gpr_mu_unlock(&server->mu_global);
  1299. return r;
  1300. }