OLD | NEW |
1 # Copyright 2011, Google Inc. | 1 #!/bin/bash |
| 2 # Copyright 2015-2016, Google Inc. |
2 # All rights reserved. | 3 # All rights reserved. |
3 # | 4 # |
4 # Redistribution and use in source and binary forms, with or without | 5 # Redistribution and use in source and binary forms, with or without |
5 # modification, are permitted provided that the following conditions are | 6 # modification, are permitted provided that the following conditions are |
6 # met: | 7 # met: |
7 # | 8 # |
8 # * Redistributions of source code must retain the above copyright | 9 # * Redistributions of source code must retain the above copyright |
9 # notice, this list of conditions and the following disclaimer. | 10 # notice, this list of conditions and the following disclaimer. |
10 # * Redistributions in binary form must reproduce the above | 11 # * Redistributions in binary form must reproduce the above |
11 # copyright notice, this list of conditions and the following disclaimer | 12 # copyright notice, this list of conditions and the following disclaimer |
12 # in the documentation and/or other materials provided with the | 13 # in the documentation and/or other materials provided with the |
13 # distribution. | 14 # distribution. |
14 # * Neither the name of Google Inc. nor the names of its | 15 # * Neither the name of Google Inc. nor the names of its |
15 # contributors may be used to endorse or promote products derived from | 16 # contributors may be used to endorse or promote products derived from |
16 # this software without specific prior written permission. | 17 # this software without specific prior written permission. |
17 # | 18 # |
18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | 19 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
19 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | 20 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
20 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | 21 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
21 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | 22 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
22 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | 23 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | 24 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 25 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 26 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 27 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 28 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
28 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 29 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
29 | 30 |
| 31 set -ex |
30 | 32 |
31 import logging | 33 apt-get install -y autoconf automake libtool curl python-virtualenv |
32 _GOODBYE_MESSAGE = u'Goodbye' | 34 |
| 35 readonly NANOPB_TMP_OUTPUT="$(mktemp -d)" |
| 36 |
| 37 # install protoc version 3 |
| 38 pushd third_party/protobuf |
| 39 ./autogen.sh |
| 40 ./configure |
| 41 make |
| 42 make install |
| 43 ldconfig |
| 44 popd |
| 45 |
| 46 if [ ! -x "/usr/local/bin/protoc" ]; then |
| 47 echo "Error: protoc not found in path" |
| 48 exit 1 |
| 49 fi |
| 50 readonly PROTOC_PATH='/usr/local/bin' |
| 51 # stack up and change to nanopb's proto generator directory |
| 52 pushd third_party/nanopb/generator/proto |
| 53 PATH="$PROTOC_PATH:$PATH" make |
| 54 |
| 55 # back to the root directory |
| 56 popd |
33 | 57 |
34 | 58 |
35 def web_socket_do_extra_handshake(request): | 59 # nanopb-compile the proto to a temp location |
36 pass # Always accept. | 60 PATH="$PROTOC_PATH:$PATH" ./tools/codegen/core/gen_load_balancing_proto.sh \ |
| 61 src/proto/grpc/lb/v0/load_balancer.proto \ |
| 62 $NANOPB_TMP_OUTPUT |
37 | 63 |
38 | 64 # compare outputs to checked compiled code |
39 def web_socket_transfer_data(request): | 65 if ! diff -r $NANOPB_TMP_OUTPUT src/core/proto/grpc/lb/v0; then |
40 while True: | 66 echo "Outputs differ: $NANOPB_TMP_OUTPUT vs src/core/proto/grpc/lb/v0" |
41 line = request.ws_stream.receive_message() | 67 exit 2 |
42 if line is None: | 68 fi |
43 return | |
44 if isinstance(line, unicode): | |
45 request.ws_stream.send_message(line, binary=False) | |
46 if line == _GOODBYE_MESSAGE: | |
47 return | |
48 else: | |
49 request.ws_stream.send_message(line, binary=True) | |
50 | |
51 | |
52 def web_socket_passive_closing_handshake(request): | |
53 return request.ws_close_code, request.ws_close_reason | |
OLD | NEW |