test/core/bad_client/tests/duplicate_header.cc

   1 /*
   2  *
   3  * Copyright 2018 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 <string.h>
  20
  21 #include <grpc/grpc.h>
  22
  23 #include "src/core/lib/surface/server.h"
  24 #include "test/core/bad_client/bad_client.h"
  25 #include "test/core/end2end/cq_verifier.h"
  26
  27 #define PFX_STR                      \
  28   "PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n" \
  29   "\x00\x00\x00\x04\x00\x00\x00\x00\x00" /* settings frame */
  30
  31 #define HEADER_STR                                                         \
  32   "\x00\x00\xc9\x01\x04\x00\x00\x00\x01" /* headers: generated from        \
  33                                             simple_request.headers in this \
  34                                             directory */                   \
  35   "\x10\x05:path\x08/foo/bar"                                              \
  36   "\x10\x07:scheme\x04http"                                                \
  37   "\x10\x07:method\x04POST"                                                \
  38   "\x10\x0a:authority\x09localhost"                                        \
  39   "\x10\x0c"                                                               \
  40   "content-type\x10"                                                       \
  41   "application/grpc"                                                       \
  42   "\x10\x14grpc-accept-encoding\x15"                                       \
  43   "deflate,identity,gzip"                                                  \
  44   "\x10\x02te\x08trailers"                                                 \
  45   "\x10\x0auser-agent\"bad-client grpc-c/0.12.0.0 (linux)"
  46
  47 #define PAYLOAD_STR                      \
  48   "\x00\x00\x20\x00\x00\x00\x00\x00\x01" \
  49   "\x00\x00\x00\x00"
  50
  51 static void* tag(intptr_t t) { return reinterpret_cast<void*>(t); }
  52
  53 static void verifier(grpc_server* server, grpc_completion_queue* cq,
  54                      void* /*registered_method*/) {
  55   grpc_call_error error;
  56   grpc_call* s;
  57   grpc_call_details call_details;
  58   grpc_byte_buffer* request_payload_recv = nullptr;
  59   grpc_op* op;
  60   grpc_op ops[6];
  61   cq_verifier* cqv = cq_verifier_create(cq);
  62   grpc_metadata_array request_metadata_recv;
  63   int was_cancelled = 2;
  64
  65   grpc_call_details_init(&call_details);
  66   grpc_metadata_array_init(&request_metadata_recv);
  67
  68   error = grpc_server_request_call(server, &s, &call_details,
  69                                    &request_metadata_recv, cq, cq, tag(101));
  70   GPR_ASSERT(GRPC_CALL_OK == error);
  71   CQ_EXPECT_COMPLETION(cqv, tag(101), 1);
  72   cq_verify(cqv);
  73
  74   GPR_ASSERT(0 == grpc_slice_str_cmp(call_details.host, "localhost"));
  75   GPR_ASSERT(0 == grpc_slice_str_cmp(call_details.method, "/foo/bar"));
  76
  77   memset(ops, 0, sizeof(ops));
  78   op = ops;
  79   op->op = GRPC_OP_SEND_INITIAL_METADATA;
  80   op->data.send_initial_metadata.count = 0;
  81   op->flags = 0;
  82   op->reserved = nullptr;
  83   op++;
  84   op->op = GRPC_OP_RECV_MESSAGE;
  85   op->data.recv_message.recv_message = &request_payload_recv;
  86   op->flags = 0;
  87   op->reserved = nullptr;
  88   op++;
  89   error = grpc_call_start_batch(s, ops, static_cast<size_t>(op - ops), tag(102),
  90                                 nullptr);
  91   GPR_ASSERT(GRPC_CALL_OK == error);
  92
  93   CQ_EXPECT_COMPLETION(cqv, tag(102), 1);
  94   cq_verify(cqv);
  95
  96   memset(ops, 0, sizeof(ops));
  97   op = ops;
  98   op->op = GRPC_OP_RECV_CLOSE_ON_SERVER;
  99   op->data.recv_close_on_server.cancelled = &was_cancelled;
 100   op->flags = 0;
 101   op->reserved = nullptr;
 102   op++;
 103   op->op = GRPC_OP_SEND_STATUS_FROM_SERVER;
 104   op->data.send_status_from_server.trailing_metadata_count = 0;
 105   op->data.send_status_from_server.status = GRPC_STATUS_UNIMPLEMENTED;
 106   grpc_slice status_details = grpc_slice_from_static_string("xyz");
 107   op->data.send_status_from_server.status_details = &status_details;
 108   op->flags = 0;
 109   op->reserved = nullptr;
 110   op++;
 111   error = grpc_call_start_batch(s, ops, static_cast<size_t>(op - ops), tag(103),
 112                                 nullptr);
 113   GPR_ASSERT(GRPC_CALL_OK == error);
 114
 115   CQ_EXPECT_COMPLETION(cqv, tag(103), 1);
 116
 117   grpc_metadata_array_destroy(&request_metadata_recv);
 118   grpc_call_details_destroy(&call_details);
 119   grpc_call_unref(s);
 120   cq_verifier_destroy(cqv);
 121 }
 122
 123 int main(int argc, char** argv) {
 124   grpc::testing::TestEnvironment env(argc, argv);
 125   grpc_init();
 126
 127   /* Verify that sending multiple headers doesn't segfault */
 128   GRPC_RUN_BAD_CLIENT_TEST(verifier, nullptr,
 129                            PFX_STR HEADER_STR HEADER_STR PAYLOAD_STR, 0);
 130   GRPC_RUN_BAD_CLIENT_TEST(verifier, nullptr,
 131                            PFX_STR HEADER_STR HEADER_STR HEADER_STR PAYLOAD_STR,
 132                            0);
 133   grpc_shutdown();
 134   return 0;
 135 }