• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  *
3  * Copyright 2015 gRPC authors.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  */
18 
19 #include "test/core/end2end/end2end_tests.h"
20 
21 #include <inttypes.h>
22 #include <stdio.h>
23 #include <string.h>
24 
25 #include <grpc/byte_buffer.h>
26 #include <grpc/grpc.h>
27 #include <grpc/support/alloc.h>
28 #include <grpc/support/log.h>
29 #include <grpc/support/time.h>
30 #include "src/core/lib/gpr/string.h"
31 #include "test/core/end2end/cq_verifier.h"
32 
tag(intptr_t t)33 static void* tag(intptr_t t) { return (void*)t; }
34 
begin_test(grpc_end2end_test_config config,const char * test_name,grpc_channel_args * client_args,grpc_channel_args * server_args)35 static grpc_end2end_test_fixture begin_test(grpc_end2end_test_config config,
36                                             const char* test_name,
37                                             grpc_channel_args* client_args,
38                                             grpc_channel_args* server_args) {
39   grpc_end2end_test_fixture f;
40   gpr_log(GPR_INFO, "Running test: %s/%s", test_name, config.name);
41   f = config.create_fixture(client_args, server_args);
42   config.init_server(&f, server_args);
43   config.init_client(&f, client_args);
44   return f;
45 }
46 
n_seconds_from_now(int n)47 static gpr_timespec n_seconds_from_now(int n) {
48   return grpc_timeout_seconds_to_deadline(n);
49 }
50 
five_seconds_from_now(void)51 static gpr_timespec five_seconds_from_now(void) {
52   return n_seconds_from_now(5);
53 }
54 
drain_cq(grpc_completion_queue * cq)55 static void drain_cq(grpc_completion_queue* cq) {
56   grpc_event ev;
57   do {
58     ev = grpc_completion_queue_next(cq, five_seconds_from_now(), nullptr);
59   } while (ev.type != GRPC_QUEUE_SHUTDOWN);
60 }
61 
shutdown_server(grpc_end2end_test_fixture * f)62 static void shutdown_server(grpc_end2end_test_fixture* f) {
63   if (!f->server) return;
64   grpc_server_shutdown_and_notify(f->server, f->shutdown_cq, tag(1000));
65   GPR_ASSERT(grpc_completion_queue_pluck(f->shutdown_cq, tag(1000),
66                                          grpc_timeout_seconds_to_deadline(5),
67                                          nullptr)
68                  .type == GRPC_OP_COMPLETE);
69   grpc_server_destroy(f->server);
70   f->server = nullptr;
71 }
72 
shutdown_client(grpc_end2end_test_fixture * f)73 static void shutdown_client(grpc_end2end_test_fixture* f) {
74   if (!f->client) return;
75   grpc_channel_destroy(f->client);
76   f->client = nullptr;
77 }
78 
end_test(grpc_end2end_test_fixture * f)79 static void end_test(grpc_end2end_test_fixture* f) {
80   shutdown_server(f);
81   shutdown_client(f);
82 
83   grpc_completion_queue_shutdown(f->cq);
84   drain_cq(f->cq);
85   grpc_completion_queue_destroy(f->cq);
86   grpc_completion_queue_destroy(f->shutdown_cq);
87 }
88 
simple_request_body(grpc_end2end_test_config config,grpc_end2end_test_fixture f,size_t num_ops)89 static void simple_request_body(grpc_end2end_test_config config,
90                                 grpc_end2end_test_fixture f, size_t num_ops) {
91   grpc_call* c;
92   cq_verifier* cqv = cq_verifier_create(f.cq);
93   grpc_op ops[6];
94   grpc_op* op;
95   grpc_metadata_array initial_metadata_recv;
96   grpc_metadata_array trailing_metadata_recv;
97   grpc_status_code status;
98   grpc_call_error error;
99   grpc_slice details;
100 
101   gpr_log(GPR_DEBUG, "test with %" PRIuPTR " ops", num_ops);
102 
103   gpr_timespec deadline = gpr_inf_past(GPR_CLOCK_REALTIME);
104   c = grpc_channel_create_call(f.client, nullptr, GRPC_PROPAGATE_DEFAULTS, f.cq,
105                                grpc_slice_from_static_string("/foo"), nullptr,
106                                deadline, nullptr);
107   GPR_ASSERT(c);
108 
109   grpc_metadata_array_init(&initial_metadata_recv);
110   grpc_metadata_array_init(&trailing_metadata_recv);
111 
112   memset(ops, 0, sizeof(ops));
113   op = ops;
114   op->op = GRPC_OP_RECV_STATUS_ON_CLIENT;
115   op->data.recv_status_on_client.trailing_metadata = &trailing_metadata_recv;
116   op->data.recv_status_on_client.status = &status;
117   op->data.recv_status_on_client.status_details = &details;
118   op->flags = 0;
119   op->reserved = nullptr;
120   op++;
121   op->op = GRPC_OP_RECV_INITIAL_METADATA;
122   op->data.recv_initial_metadata.recv_initial_metadata = &initial_metadata_recv;
123   op->flags = 0;
124   op->reserved = nullptr;
125   op++;
126   op->op = GRPC_OP_SEND_INITIAL_METADATA;
127   op->data.send_initial_metadata.count = 0;
128   op->flags = 0;
129   op->reserved = nullptr;
130   op++;
131   op->op = GRPC_OP_SEND_CLOSE_FROM_CLIENT;
132   op->flags = 0;
133   op->reserved = nullptr;
134   op++;
135   GPR_ASSERT(num_ops <= (size_t)(op - ops));
136   error = grpc_call_start_batch(c, ops, num_ops, tag(1), nullptr);
137   GPR_ASSERT(GRPC_CALL_OK == error);
138 
139   CQ_EXPECT_COMPLETION(cqv, tag(1), 1);
140   cq_verify(cqv);
141 
142   GPR_ASSERT(status == GRPC_STATUS_DEADLINE_EXCEEDED);
143 
144   grpc_slice_unref(details);
145   grpc_metadata_array_destroy(&initial_metadata_recv);
146   grpc_metadata_array_destroy(&trailing_metadata_recv);
147 
148   grpc_call_unref(c);
149 
150   cq_verifier_destroy(cqv);
151 }
152 
test_invoke_simple_request(grpc_end2end_test_config config,size_t num_ops)153 static void test_invoke_simple_request(grpc_end2end_test_config config,
154                                        size_t num_ops) {
155   grpc_end2end_test_fixture f;
156 
157   f = begin_test(config, "test_invoke_simple_request", nullptr, nullptr);
158   simple_request_body(config, f, num_ops);
159   end_test(&f);
160   config.tear_down_data(&f);
161 }
162 
negative_deadline(grpc_end2end_test_config config)163 void negative_deadline(grpc_end2end_test_config config) {
164   size_t i;
165   for (i = 1; i <= 4; i++) {
166     test_invoke_simple_request(config, i);
167   }
168 }
169 
negative_deadline_pre_init(void)170 void negative_deadline_pre_init(void) {}
171