| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743 | 
							- #!/usr/bin/env python2.7
 
- # Copyright 2015 gRPC authors.
 
- #
 
- # Licensed under the Apache License, Version 2.0 (the "License");
 
- # you may not use this file except in compliance with the License.
 
- # You may obtain a copy of the License at
 
- #
 
- #     http://www.apache.org/licenses/LICENSE-2.0
 
- #
 
- # Unless required by applicable law or agreed to in writing, software
 
- # distributed under the License is distributed on an "AS IS" BASIS,
 
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
- # See the License for the specific language governing permissions and
 
- # limitations under the License.
 
- import hashlib
 
- import itertools
 
- import collections
 
- import os
 
- import sys
 
- import subprocess
 
- import re
 
- import perfection
 
- # Configuration: a list of either strings or 2-tuples of strings.
 
- # A single string represents a static grpc_mdstr.
 
- # A 2-tuple represents a static grpc_mdelem (and appropriate grpc_mdstrs will
 
- # also be created).
 
- # The list of 2-tuples must begin with the static hpack table elements as
 
- # defined by RFC 7541 and be in the same order because of an hpack encoding
 
- # performance optimization that relies on this. If you want to change this, then
 
- # you must change the implementation of the encoding optimization as well.
 
- CONFIG = [
 
-     # metadata strings
 
-     'host',
 
-     'grpc-timeout',
 
-     'grpc-internal-encoding-request',
 
-     'grpc-internal-stream-encoding-request',
 
-     'grpc-payload-bin',
 
-     ':path',
 
-     'grpc-encoding',
 
-     'grpc-accept-encoding',
 
-     'user-agent',
 
-     ':authority',
 
-     'grpc-message',
 
-     'grpc-status',
 
-     'grpc-server-stats-bin',
 
-     'grpc-tags-bin',
 
-     'grpc-trace-bin',
 
-     'grpc-previous-rpc-attempts',
 
-     'grpc-retry-pushback-ms',
 
-     '1',
 
-     '2',
 
-     '3',
 
-     '4',
 
-     '',
 
-     'x-endpoint-load-metrics-bin',
 
-     # channel arg keys
 
-     'grpc.wait_for_ready',
 
-     'grpc.timeout',
 
-     'grpc.max_request_message_bytes',
 
-     'grpc.max_response_message_bytes',
 
-     # well known method names
 
-     '/grpc.lb.v1.LoadBalancer/BalanceLoad',
 
-     '/envoy.service.load_stats.v2.LoadReportingService/StreamLoadStats',
 
-     '/envoy.service.load_stats.v3.LoadReportingService/StreamLoadStats',
 
-     '/grpc.health.v1.Health/Watch',
 
-     '/envoy.service.discovery.v2.AggregatedDiscoveryService/StreamAggregatedResources',
 
-     '/envoy.service.discovery.v3.AggregatedDiscoveryService/StreamAggregatedResources',
 
-     # compression algorithm names
 
-     'deflate',
 
-     'gzip',
 
-     'stream/gzip',
 
-     # metadata elements
 
-     # begin hpack static elements
 
-     (':authority', ''),
 
-     (':method', 'GET'),
 
-     (':method', 'POST'),
 
-     (':path', '/'),
 
-     (':path', '/index.html'),
 
-     (':scheme', 'http'),
 
-     (':scheme', 'https'),
 
-     (':status', '200'),
 
-     (':status', '204'),
 
-     (':status', '206'),
 
-     (':status', '304'),
 
-     (':status', '400'),
 
-     (':status', '404'),
 
-     (':status', '500'),
 
-     ('accept-charset', ''),
 
-     ('accept-encoding', 'gzip, deflate'),
 
-     ('accept-language', ''),
 
-     ('accept-ranges', ''),
 
-     ('accept', ''),
 
-     ('access-control-allow-origin', ''),
 
-     ('age', ''),
 
-     ('allow', ''),
 
-     ('authorization', ''),
 
-     ('cache-control', ''),
 
-     ('content-disposition', ''),
 
-     ('content-encoding', ''),
 
-     ('content-language', ''),
 
-     ('content-length', ''),
 
-     ('content-location', ''),
 
-     ('content-range', ''),
 
-     ('content-type', ''),
 
-     ('cookie', ''),
 
-     ('date', ''),
 
-     ('etag', ''),
 
-     ('expect', ''),
 
-     ('expires', ''),
 
-     ('from', ''),
 
-     ('host', ''),
 
-     ('if-match', ''),
 
-     ('if-modified-since', ''),
 
-     ('if-none-match', ''),
 
-     ('if-range', ''),
 
-     ('if-unmodified-since', ''),
 
-     ('last-modified', ''),
 
-     ('link', ''),
 
-     ('location', ''),
 
-     ('max-forwards', ''),
 
-     ('proxy-authenticate', ''),
 
-     ('proxy-authorization', ''),
 
-     ('range', ''),
 
-     ('referer', ''),
 
-     ('refresh', ''),
 
-     ('retry-after', ''),
 
-     ('server', ''),
 
-     ('set-cookie', ''),
 
-     ('strict-transport-security', ''),
 
-     ('transfer-encoding', ''),
 
-     ('user-agent', ''),
 
-     ('vary', ''),
 
-     ('via', ''),
 
-     ('www-authenticate', ''),
 
-     # end hpack static elements
 
-     ('grpc-status', '0'),
 
-     ('grpc-status', '1'),
 
-     ('grpc-status', '2'),
 
-     ('grpc-encoding', 'identity'),
 
-     ('grpc-encoding', 'gzip'),
 
-     ('grpc-encoding', 'deflate'),
 
-     ('te', 'trailers'),
 
-     ('content-type', 'application/grpc'),
 
-     (':scheme', 'grpc'),
 
-     (':method', 'PUT'),
 
-     ('accept-encoding', ''),
 
-     ('content-encoding', 'identity'),
 
-     ('content-encoding', 'gzip'),
 
-     ('lb-cost-bin', ''),
 
- ]
 
- # All entries here are ignored when counting non-default initial metadata that
 
- # prevents the chttp2 server from sending a Trailers-Only response.
 
- METADATA_BATCH_CALLOUTS = [
 
-     ':path',
 
-     ':method',
 
-     ':status',
 
-     ':authority',
 
-     ':scheme',
 
-     'te',
 
-     'grpc-message',
 
-     'grpc-status',
 
-     'grpc-payload-bin',
 
-     'grpc-encoding',
 
-     'grpc-accept-encoding',
 
-     'grpc-server-stats-bin',
 
-     'grpc-tags-bin',
 
-     'grpc-trace-bin',
 
-     'content-type',
 
-     'content-encoding',
 
-     'accept-encoding',
 
-     'grpc-internal-encoding-request',
 
-     'grpc-internal-stream-encoding-request',
 
-     'user-agent',
 
-     'host',
 
-     'grpc-previous-rpc-attempts',
 
-     'grpc-retry-pushback-ms',
 
-     'x-endpoint-load-metrics-bin',
 
- ]
 
- COMPRESSION_ALGORITHMS = [
 
-     'identity',
 
-     'deflate',
 
-     'gzip',
 
- ]
 
- STREAM_COMPRESSION_ALGORITHMS = [
 
-     'identity',
 
-     'gzip',
 
- ]
 
- # utility: mangle the name of a config
 
- def mangle(elem, name=None):
 
-     xl = {
 
-         '-': '_',
 
-         ':': '',
 
-         '/': 'slash',
 
-         '.': 'dot',
 
-         ',': 'comma',
 
-         ' ': '_',
 
-     }
 
-     def m0(x):
 
-         if not x:
 
-             return 'empty'
 
-         r = ''
 
-         for c in x:
 
-             put = xl.get(c, c.lower())
 
-             if not put:
 
-                 continue
 
-             last_is_underscore = r[-1] == '_' if r else True
 
-             if last_is_underscore and put == '_':
 
-                 continue
 
-             elif len(put) > 1:
 
-                 if not last_is_underscore:
 
-                     r += '_'
 
-                 r += put
 
-                 r += '_'
 
-             else:
 
-                 r += put
 
-         if r[-1] == '_':
 
-             r = r[:-1]
 
-         return r
 
-     def n(default, name=name):
 
-         if name is None:
 
-             return 'grpc_%s_' % default
 
-         if name == '':
 
-             return ''
 
-         return 'grpc_%s_' % name
 
-     if isinstance(elem, tuple):
 
-         return '%s%s_%s' % (n('mdelem'), m0(elem[0]), m0(elem[1]))
 
-     else:
 
-         return '%s%s' % (n('mdstr'), m0(elem))
 
- # utility: generate some hash value for a string
 
- def fake_hash(elem):
 
-     return hashlib.md5(elem).hexdigest()[0:8]
 
- # utility: print a big comment block into a set of files
 
- def put_banner(files, banner):
 
-     for f in files:
 
-         print >> f, '/*'
 
-         for line in banner:
 
-             print >> f, ' * %s' % line
 
-         print >> f, ' */'
 
-         print >> f
 
- # build a list of all the strings we need
 
- all_strs = list()
 
- all_elems = list()
 
- static_userdata = {}
 
- # put metadata batch callouts first, to make the check of if a static metadata
 
- # string is a callout trivial
 
- for elem in METADATA_BATCH_CALLOUTS:
 
-     if elem not in all_strs:
 
-         all_strs.append(elem)
 
- for elem in CONFIG:
 
-     if isinstance(elem, tuple):
 
-         if elem[0] not in all_strs:
 
-             all_strs.append(elem[0])
 
-         if elem[1] not in all_strs:
 
-             all_strs.append(elem[1])
 
-         if elem not in all_elems:
 
-             all_elems.append(elem)
 
-     else:
 
-         if elem not in all_strs:
 
-             all_strs.append(elem)
 
- compression_elems = []
 
- for mask in range(1, 1 << len(COMPRESSION_ALGORITHMS)):
 
-     val = ','.join(COMPRESSION_ALGORITHMS[alg]
 
-                    for alg in range(0, len(COMPRESSION_ALGORITHMS))
 
-                    if (1 << alg) & mask)
 
-     elem = ('grpc-accept-encoding', val)
 
-     if val not in all_strs:
 
-         all_strs.append(val)
 
-     if elem not in all_elems:
 
-         all_elems.append(elem)
 
-     compression_elems.append(elem)
 
-     static_userdata[elem] = 1 + (mask | 1)
 
- stream_compression_elems = []
 
- for mask in range(1, 1 << len(STREAM_COMPRESSION_ALGORITHMS)):
 
-     val = ','.join(STREAM_COMPRESSION_ALGORITHMS[alg]
 
-                    for alg in range(0, len(STREAM_COMPRESSION_ALGORITHMS))
 
-                    if (1 << alg) & mask)
 
-     elem = ('accept-encoding', val)
 
-     if val not in all_strs:
 
-         all_strs.append(val)
 
-     if elem not in all_elems:
 
-         all_elems.append(elem)
 
-     stream_compression_elems.append(elem)
 
-     static_userdata[elem] = 1 + (mask | 1)
 
- # output configuration
 
- args = sys.argv[1:]
 
- H = None
 
- C = None
 
- D = None
 
- if args:
 
-     if 'header' in args:
 
-         H = sys.stdout
 
-     else:
 
-         H = open('/dev/null', 'w')
 
-     if 'source' in args:
 
-         C = sys.stdout
 
-     else:
 
-         C = open('/dev/null', 'w')
 
-     if 'dictionary' in args:
 
-         D = sys.stdout
 
-     else:
 
-         D = open('/dev/null', 'w')
 
- else:
 
-     H = open(
 
-         os.path.join(os.path.dirname(sys.argv[0]),
 
-                      '../../../src/core/lib/transport/static_metadata.h'), 'w')
 
-     C = open(
 
-         os.path.join(os.path.dirname(sys.argv[0]),
 
-                      '../../../src/core/lib/transport/static_metadata.cc'), 'w')
 
-     D = open(
 
-         os.path.join(os.path.dirname(sys.argv[0]),
 
-                      '../../../test/core/end2end/fuzzers/hpack.dictionary'),
 
-         'w')
 
- # copy-paste copyright notice from this file
 
- with open(sys.argv[0]) as my_source:
 
-     copyright = []
 
-     for line in my_source:
 
-         if line[0] != '#':
 
-             break
 
-     for line in my_source:
 
-         if line[0] == '#':
 
-             copyright.append(line)
 
-             break
 
-     for line in my_source:
 
-         if line[0] != '#':
 
-             break
 
-         copyright.append(line)
 
-     put_banner([H, C], [line[2:].rstrip() for line in copyright])
 
- hex_bytes = [ord(c) for c in 'abcdefABCDEF0123456789']
 
- def esc_dict(line):
 
-     out = "\""
 
-     for c in line:
 
-         if 32 <= c < 127:
 
-             if c != ord('"'):
 
-                 out += chr(c)
 
-             else:
 
-                 out += "\\\""
 
-         else:
 
-             out += '\\x%02X' % c
 
-     return out + "\""
 
- put_banner([H, C], """WARNING: Auto-generated code.
 
- To make changes to this file, change
 
- tools/codegen/core/gen_static_metadata.py, and then re-run it.
 
- See metadata.h for an explanation of the interface here, and metadata.cc for
 
- an explanation of what's going on.
 
- """.splitlines())
 
- print >> H, '#ifndef GRPC_CORE_LIB_TRANSPORT_STATIC_METADATA_H'
 
- print >> H, '#define GRPC_CORE_LIB_TRANSPORT_STATIC_METADATA_H'
 
- print >> H
 
- print >> H, '#include <grpc/support/port_platform.h>'
 
- print >> H
 
- print >> H, '#include <cstdint>'
 
- print >> H
 
- print >> H, '#include "src/core/lib/transport/metadata.h"'
 
- print >> H
 
- print >> C, '#include <grpc/support/port_platform.h>'
 
- print >> C
 
- print >> C, '#include "src/core/lib/transport/static_metadata.h"'
 
- print >> C
 
- print >> C, '#include "src/core/lib/slice/slice_internal.h"'
 
- print >> C
 
- str_ofs = 0
 
- id2strofs = {}
 
- for i, elem in enumerate(all_strs):
 
-     id2strofs[i] = str_ofs
 
-     str_ofs += len(elem)
 
- def slice_def_for_ctx(i):
 
-     return (
 
-         'grpc_core::StaticMetadataSlice(&refcounts[%d].base, %d, g_bytes+%d)'
 
-     ) % (i, len(all_strs[i]), id2strofs[i])
 
- def slice_def(i):
 
-     return (
 
-         'grpc_core::StaticMetadataSlice(&grpc_static_metadata_refcounts()[%d].base, %d, g_bytes+%d)'
 
-     ) % (i, len(all_strs[i]), id2strofs[i])
 
- def str_idx(s):
 
-     for i, s2 in enumerate(all_strs):
 
-         if s == s2:
 
-             return i
 
- # validate configuration
 
- for elem in METADATA_BATCH_CALLOUTS:
 
-     assert elem in all_strs
 
- static_slice_dest_assert = (
 
-     'static_assert(std::is_trivially_destructible' +
 
-     '<grpc_core::StaticMetadataSlice>::value, '
 
-     '"grpc_core::StaticMetadataSlice must be trivially destructible.");')
 
- print >> H, static_slice_dest_assert
 
- print >> H, '#define GRPC_STATIC_MDSTR_COUNT %d' % len(all_strs)
 
- print >> H, '''
 
- void grpc_init_static_metadata_ctx(void);
 
- void grpc_destroy_static_metadata_ctx(void);
 
- namespace grpc_core {
 
- #ifndef NDEBUG
 
- constexpr uint64_t kGrpcStaticMetadataInitCanary = 0xCAFEF00DC0FFEE11L;
 
- uint64_t StaticMetadataInitCanary();
 
- #endif
 
- extern const StaticMetadataSlice* g_static_metadata_slice_table;
 
- }
 
- inline const grpc_core::StaticMetadataSlice* grpc_static_slice_table() {
 
-   GPR_DEBUG_ASSERT(grpc_core::StaticMetadataInitCanary()
 
-     == grpc_core::kGrpcStaticMetadataInitCanary);
 
-   GPR_DEBUG_ASSERT(grpc_core::g_static_metadata_slice_table != nullptr);
 
-   return grpc_core::g_static_metadata_slice_table;
 
- }
 
- '''
 
- for i, elem in enumerate(all_strs):
 
-     print >> H, '/* "%s" */' % elem
 
-     print >> H, '#define %s (grpc_static_slice_table()[%d])' % (
 
-         mangle(elem).upper(), i)
 
- print >> H
 
- print >> C, 'static constexpr uint8_t g_bytes[] = {%s};' % (','.join(
 
-     '%d' % ord(c) for c in ''.join(all_strs)))
 
- print >> C
 
- print >> H, '''
 
- namespace grpc_core {
 
- struct StaticSliceRefcount;
 
- extern StaticSliceRefcount* g_static_metadata_slice_refcounts;
 
- }
 
- inline grpc_core::StaticSliceRefcount* grpc_static_metadata_refcounts() {
 
-   GPR_DEBUG_ASSERT(grpc_core::StaticMetadataInitCanary()
 
-     == grpc_core::kGrpcStaticMetadataInitCanary);
 
-   GPR_DEBUG_ASSERT(grpc_core::g_static_metadata_slice_refcounts != nullptr);
 
-   return grpc_core::g_static_metadata_slice_refcounts;
 
- }
 
- '''
 
- print >> C, 'grpc_slice_refcount grpc_core::StaticSliceRefcount::kStaticSubRefcount;'
 
- print >> C, '''
 
- namespace grpc_core {
 
- struct StaticMetadataCtx {
 
- #ifndef NDEBUG
 
-   const uint64_t init_canary = kGrpcStaticMetadataInitCanary;
 
- #endif
 
-   StaticSliceRefcount
 
-     refcounts[GRPC_STATIC_MDSTR_COUNT] = {
 
- '''
 
- for i, elem in enumerate(all_strs):
 
-     print >> C, '  StaticSliceRefcount(%d), ' % i
 
- print >> C, '};'  # static slice refcounts
 
- print >> C
 
- print >> C, '''
 
-   const StaticMetadataSlice
 
-     slices[GRPC_STATIC_MDSTR_COUNT] = {
 
- '''
 
- for i, elem in enumerate(all_strs):
 
-     print >> C, slice_def_for_ctx(i) + ','
 
- print >> C, '};'  # static slices
 
- print >> C, 'StaticMetadata static_mdelem_table[GRPC_STATIC_MDELEM_COUNT] = {'
 
- for idx, (a, b) in enumerate(all_elems):
 
-     print >> C, 'StaticMetadata(%s,%s, %d),' % (slice_def_for_ctx(
 
-         str_idx(a)), slice_def_for_ctx(str_idx(b)), idx)
 
- print >> C, '};'  # static_mdelem_table
 
- print >> C, ('''
 
- /* Warning: the core static metadata currently operates under the soft constraint
 
- that the first GRPC_CHTTP2_LAST_STATIC_ENTRY (61) entries must contain
 
- metadata specified by the http2 hpack standard. The CHTTP2 transport reads the
 
- core metadata with this assumption in mind. If the order of the core static
 
- metadata is to be changed, then the CHTTP2 transport must be changed as well to
 
- stop relying on the core metadata. */
 
- ''')
 
- print >> C, ('grpc_mdelem '
 
-              'static_mdelem_manifested[GRPC_STATIC_MDELEM_COUNT] = {')
 
- print >> C, '// clang-format off'
 
- static_mds = []
 
- for i, elem in enumerate(all_elems):
 
-     md_name = mangle(elem).upper()
 
-     md_human_readable = '"%s": "%s"' % elem
 
-     md_spec = '    /* %s: \n     %s */\n' % (md_name, md_human_readable)
 
-     md_spec += '    GRPC_MAKE_MDELEM(\n'
 
-     md_spec += (('        &static_mdelem_table[%d].data(),\n' % i) +
 
-                 '        GRPC_MDELEM_STORAGE_STATIC)')
 
-     static_mds.append(md_spec)
 
- print >> C, ',\n'.join(static_mds)
 
- print >> C, '// clang-format on'
 
- print >> C, ('};')  # static_mdelem_manifested
 
- print >> C, '};'  # struct StaticMetadataCtx
 
- print >> C, '}'  # namespace grpc_core
 
- print >> C, '''
 
- namespace grpc_core {
 
- static StaticMetadataCtx* g_static_metadata_slice_ctx = nullptr;
 
- const StaticMetadataSlice* g_static_metadata_slice_table = nullptr;
 
- StaticSliceRefcount* g_static_metadata_slice_refcounts = nullptr;
 
- StaticMetadata* g_static_mdelem_table = nullptr;
 
- grpc_mdelem* g_static_mdelem_manifested = nullptr;
 
- #ifndef NDEBUG
 
- uint64_t StaticMetadataInitCanary() {
 
-   return g_static_metadata_slice_ctx->init_canary;
 
- }
 
- #endif
 
- }
 
- void grpc_init_static_metadata_ctx(void) {
 
-   grpc_core::g_static_metadata_slice_ctx
 
-     = new grpc_core::StaticMetadataCtx();
 
-   grpc_core::g_static_metadata_slice_table
 
-     = grpc_core::g_static_metadata_slice_ctx->slices;
 
-   grpc_core::g_static_metadata_slice_refcounts
 
-     = grpc_core::g_static_metadata_slice_ctx->refcounts;
 
-   grpc_core::g_static_mdelem_table
 
-     = grpc_core::g_static_metadata_slice_ctx->static_mdelem_table;
 
-   grpc_core::g_static_mdelem_manifested =
 
-       grpc_core::g_static_metadata_slice_ctx->static_mdelem_manifested;
 
- }
 
- void grpc_destroy_static_metadata_ctx(void) {
 
-   delete grpc_core::g_static_metadata_slice_ctx;
 
-   grpc_core::g_static_metadata_slice_ctx = nullptr;
 
-   grpc_core::g_static_metadata_slice_table = nullptr;
 
-   grpc_core::g_static_metadata_slice_refcounts = nullptr;
 
-   grpc_core::g_static_mdelem_table = nullptr;
 
-   grpc_core::g_static_mdelem_manifested = nullptr;
 
- }
 
- '''
 
- print >> C
 
- print >> H, '#define GRPC_IS_STATIC_METADATA_STRING(slice) \\'
 
- print >> H, ('  ((slice).refcount != NULL && (slice).refcount->GetType() == '
 
-              'grpc_slice_refcount::Type::STATIC)')
 
- print >> H
 
- print >> C
 
- print >> H, '#define GRPC_STATIC_METADATA_INDEX(static_slice) \\'
 
- print >> H, '(reinterpret_cast<grpc_core::StaticSliceRefcount*>((static_slice).refcount)->index)'
 
- print >> H
 
- print >> D, '# hpack fuzzing dictionary'
 
- for i, elem in enumerate(all_strs):
 
-     print >> D, '%s' % (esc_dict([len(elem)] + [ord(c) for c in elem]))
 
- for i, elem in enumerate(all_elems):
 
-     print >> D, '%s' % (esc_dict([0, len(elem[0])] + [ord(c) for c in elem[0]] +
 
-                                  [len(elem[1])] + [ord(c) for c in elem[1]]))
 
- print >> H, '#define GRPC_STATIC_MDELEM_COUNT %d' % len(all_elems)
 
- print >> H, '''
 
- namespace grpc_core {
 
- extern StaticMetadata* g_static_mdelem_table;
 
- extern grpc_mdelem* g_static_mdelem_manifested;
 
- }
 
- inline grpc_core::StaticMetadata* grpc_static_mdelem_table() {
 
-   GPR_DEBUG_ASSERT(grpc_core::StaticMetadataInitCanary()
 
-     == grpc_core::kGrpcStaticMetadataInitCanary);
 
-   GPR_DEBUG_ASSERT(grpc_core::g_static_mdelem_table != nullptr);
 
-   return grpc_core::g_static_mdelem_table;
 
- }
 
- inline grpc_mdelem* grpc_static_mdelem_manifested() {
 
-   GPR_DEBUG_ASSERT(grpc_core::StaticMetadataInitCanary()
 
-     == grpc_core::kGrpcStaticMetadataInitCanary);
 
-   GPR_DEBUG_ASSERT(grpc_core::g_static_mdelem_manifested != nullptr);
 
-   return grpc_core::g_static_mdelem_manifested;
 
- }
 
- '''
 
- print >> H, ('extern uintptr_t '
 
-              'grpc_static_mdelem_user_data[GRPC_STATIC_MDELEM_COUNT];')
 
- for i, elem in enumerate(all_elems):
 
-     md_name = mangle(elem).upper()
 
-     print >> H, '/* "%s": "%s" */' % elem
 
-     print >> H, ('#define %s (grpc_static_mdelem_manifested()[%d])' %
 
-                  (md_name, i))
 
- print >> H
 
- print >> C, ('uintptr_t grpc_static_mdelem_user_data[GRPC_STATIC_MDELEM_COUNT] '
 
-              '= {')
 
- print >> C, '  %s' % ','.join(
 
-     '%d' % static_userdata.get(elem, 0) for elem in all_elems)
 
- print >> C, '};'
 
- print >> C
 
- def md_idx(m):
 
-     for i, m2 in enumerate(all_elems):
 
-         if m == m2:
 
-             return i
 
- def offset_trials(mink):
 
-     yield 0
 
-     for i in range(1, 100):
 
-         for mul in [-1, 1]:
 
-             yield mul * i
 
- def perfect_hash(keys, name):
 
-     p = perfection.hash_parameters(keys)
 
-     def f(i, p=p):
 
-         i += p.offset
 
-         x = i % p.t
 
-         y = i / p.t
 
-         return x + p.r[y]
 
-     return {
 
-         'PHASHNKEYS':
 
-             len(p.slots),
 
-         'pyfunc':
 
-             f,
 
-         'code':
 
-             """
 
- static const int8_t %(name)s_r[] = {%(r)s};
 
- static uint32_t %(name)s_phash(uint32_t i) {
 
-   i %(offset_sign)s= %(offset)d;
 
-   uint32_t x = i %% %(t)d;
 
-   uint32_t y = i / %(t)d;
 
-   uint32_t h = x;
 
-   if (y < GPR_ARRAY_SIZE(%(name)s_r)) {
 
-     uint32_t delta = (uint32_t)%(name)s_r[y];
 
-     h += delta;
 
-   }
 
-   return h;
 
- }
 
-     """ % {
 
-                 'name': name,
 
-                 'r': ','.join('%d' % (r if r is not None else 0) for r in p.r),
 
-                 't': p.t,
 
-                 'offset': abs(p.offset),
 
-                 'offset_sign': '+' if p.offset > 0 else '-'
 
-             }
 
-     }
 
- elem_keys = [
 
-     str_idx(elem[0]) * len(all_strs) + str_idx(elem[1]) for elem in all_elems
 
- ]
 
- elem_hash = perfect_hash(elem_keys, 'elems')
 
- print >> C, elem_hash['code']
 
- keys = [0] * int(elem_hash['PHASHNKEYS'])
 
- idxs = [255] * int(elem_hash['PHASHNKEYS'])
 
- for i, k in enumerate(elem_keys):
 
-     h = elem_hash['pyfunc'](k)
 
-     assert keys[h] == 0
 
-     keys[h] = k
 
-     idxs[h] = i
 
- print >> C, 'static const uint16_t elem_keys[] = {%s};' % ','.join(
 
-     '%d' % k for k in keys)
 
- print >> C, 'static const uint8_t elem_idxs[] = {%s};' % ','.join(
 
-     '%d' % i for i in idxs)
 
- print >> C
 
- print >> H, 'grpc_mdelem grpc_static_mdelem_for_static_strings(intptr_t a, intptr_t b);'
 
- print >> C, 'grpc_mdelem grpc_static_mdelem_for_static_strings(intptr_t a, intptr_t b) {'
 
- print >> C, '  if (a == -1 || b == -1) return GRPC_MDNULL;'
 
- print >> C, '  uint32_t k = static_cast<uint32_t>(a * %d + b);' % len(all_strs)
 
- print >> C, '  uint32_t h = elems_phash(k);'
 
- print >> C, '  return h < GPR_ARRAY_SIZE(elem_keys) && elem_keys[h] == k && elem_idxs[h] != 255 ? GRPC_MAKE_MDELEM(&grpc_static_mdelem_table()[elem_idxs[h]].data(), GRPC_MDELEM_STORAGE_STATIC) : GRPC_MDNULL;'
 
- print >> C, '}'
 
- print >> C
 
- print >> H, 'typedef enum {'
 
- for elem in METADATA_BATCH_CALLOUTS:
 
-     print >> H, '  %s,' % mangle(elem, 'batch').upper()
 
- print >> H, '  GRPC_BATCH_CALLOUTS_COUNT'
 
- print >> H, '} grpc_metadata_batch_callouts_index;'
 
- print >> H
 
- print >> H, 'typedef union {'
 
- print >> H, '  struct grpc_linked_mdelem *array[GRPC_BATCH_CALLOUTS_COUNT];'
 
- print >> H, '  struct {'
 
- for elem in METADATA_BATCH_CALLOUTS:
 
-     print >> H, '  struct grpc_linked_mdelem *%s;' % mangle(elem, '').lower()
 
- print >> H, '  } named;'
 
- print >> H, '} grpc_metadata_batch_callouts;'
 
- print >> H
 
- batch_idx_of_hdr = '#define GRPC_BATCH_INDEX_OF(slice) \\'
 
- static_slice = 'GRPC_IS_STATIC_METADATA_STRING((slice))'
 
- slice_to_slice_ref = '(slice).refcount'
 
- static_slice_ref_type = 'grpc_core::StaticSliceRefcount*'
 
- slice_ref_as_static = ('reinterpret_cast<' + static_slice_ref_type + '>(' +
 
-                        slice_to_slice_ref + ')')
 
- slice_ref_idx = slice_ref_as_static + '->index'
 
- batch_idx_type = 'grpc_metadata_batch_callouts_index'
 
- slice_ref_idx_to_batch_idx = ('static_cast<' + batch_idx_type + '>(' +
 
-                               slice_ref_idx + ')')
 
- batch_invalid_idx = 'GRPC_BATCH_CALLOUTS_COUNT'
 
- batch_invalid_u32 = 'static_cast<uint32_t>(' + batch_invalid_idx + ')'
 
- # Assemble GRPC_BATCH_INDEX_OF(slice) macro as a join for ease of reading.
 
- batch_idx_of_pieces = [
 
-     batch_idx_of_hdr, '\n', '(', static_slice, '&&', slice_ref_idx, '<=',
 
-     batch_invalid_u32, '?', slice_ref_idx_to_batch_idx, ':', batch_invalid_idx,
 
-     ')'
 
- ]
 
- print >> H, ''.join(batch_idx_of_pieces)
 
- print >> H
 
- print >> H, 'extern const uint8_t grpc_static_accept_encoding_metadata[%d];' % (
 
-     1 << len(COMPRESSION_ALGORITHMS))
 
- print >> C, 'const uint8_t grpc_static_accept_encoding_metadata[%d] = {' % (
 
-     1 << len(COMPRESSION_ALGORITHMS))
 
- print >> C, '0,%s' % ','.join('%d' % md_idx(elem) for elem in compression_elems)
 
- print >> C, '};'
 
- print >> C
 
- print >> H, '#define GRPC_MDELEM_ACCEPT_ENCODING_FOR_ALGORITHMS(algs) (GRPC_MAKE_MDELEM(&grpc_static_mdelem_table()[grpc_static_accept_encoding_metadata[(algs)]].data(), GRPC_MDELEM_STORAGE_STATIC))'
 
- print >> H
 
- print >> H, 'extern const uint8_t grpc_static_accept_stream_encoding_metadata[%d];' % (
 
-     1 << len(STREAM_COMPRESSION_ALGORITHMS))
 
- print >> C, 'const uint8_t grpc_static_accept_stream_encoding_metadata[%d] = {' % (
 
-     1 << len(STREAM_COMPRESSION_ALGORITHMS))
 
- print >> C, '0,%s' % ','.join(
 
-     '%d' % md_idx(elem) for elem in stream_compression_elems)
 
- print >> C, '};'
 
- print >> H, '#define GRPC_MDELEM_ACCEPT_STREAM_ENCODING_FOR_ALGORITHMS(algs) (GRPC_MAKE_MDELEM(&grpc_static_mdelem_table()[grpc_static_accept_stream_encoding_metadata[(algs)]].data(), GRPC_MDELEM_STORAGE_STATIC))'
 
- print >> H, '#endif /* GRPC_CORE_LIB_TRANSPORT_STATIC_METADATA_H */'
 
- H.close()
 
- C.close()
 
 
  |