server.c 51 KB

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