diff options
Diffstat (limited to 'tools/codegen/core/gen_settings_ids.py')
-rwxr-xr-x | tools/codegen/core/gen_settings_ids.py | 189 |
1 files changed, 104 insertions, 85 deletions
diff --git a/tools/codegen/core/gen_settings_ids.py b/tools/codegen/core/gen_settings_ids.py index 481c4216a3..bc4380605a 100755 --- a/tools/codegen/core/gen_settings_ids.py +++ b/tools/codegen/core/gen_settings_ids.py @@ -24,92 +24,114 @@ Setting = collections.namedtuple('Setting', 'id default min max on_error') OnError = collections.namedtuple('OnError', 'behavior code') clamp_invalid_value = OnError('CLAMP_INVALID_VALUE', 'PROTOCOL_ERROR') disconnect_on_invalid_value = lambda e: OnError('DISCONNECT_ON_INVALID_VALUE', e) -DecoratedSetting = collections.namedtuple('DecoratedSetting', 'enum name setting') +DecoratedSetting = collections.namedtuple('DecoratedSetting', + 'enum name setting') _SETTINGS = { - 'HEADER_TABLE_SIZE': Setting(1, 4096, 0, 0xffffffff, clamp_invalid_value), - 'ENABLE_PUSH': Setting(2, 1, 0, 1, disconnect_on_invalid_value('PROTOCOL_ERROR')), - 'MAX_CONCURRENT_STREAMS': Setting(3, 0xffffffff, 0, 0xffffffff, disconnect_on_invalid_value('PROTOCOL_ERROR')), - 'INITIAL_WINDOW_SIZE': Setting(4, 65535, 0, 0x7fffffff, disconnect_on_invalid_value('FLOW_CONTROL_ERROR')), - 'MAX_FRAME_SIZE': Setting(5, 16384, 16384, 16777215, disconnect_on_invalid_value('PROTOCOL_ERROR')), - 'MAX_HEADER_LIST_SIZE': Setting(6, _MAX_HEADER_LIST_SIZE, 0, _MAX_HEADER_LIST_SIZE, clamp_invalid_value), - 'GRPC_ALLOW_TRUE_BINARY_METADATA': Setting(0xfe03, 0, 0, 1, clamp_invalid_value), + 'HEADER_TABLE_SIZE': + Setting(1, 4096, 0, 0xffffffff, clamp_invalid_value), + 'ENABLE_PUSH': + Setting(2, 1, 0, 1, disconnect_on_invalid_value('PROTOCOL_ERROR')), + 'MAX_CONCURRENT_STREAMS': + Setting(3, 0xffffffff, 0, 0xffffffff, + disconnect_on_invalid_value('PROTOCOL_ERROR')), + 'INITIAL_WINDOW_SIZE': + Setting(4, 65535, 0, 0x7fffffff, + disconnect_on_invalid_value('FLOW_CONTROL_ERROR')), + 'MAX_FRAME_SIZE': + Setting(5, 16384, 16384, 16777215, + disconnect_on_invalid_value('PROTOCOL_ERROR')), + 'MAX_HEADER_LIST_SIZE': + Setting(6, _MAX_HEADER_LIST_SIZE, 0, _MAX_HEADER_LIST_SIZE, + clamp_invalid_value), + 'GRPC_ALLOW_TRUE_BINARY_METADATA': + Setting(0xfe03, 0, 0, 1, clamp_invalid_value), } H = open('src/core/ext/transport/chttp2/transport/http2_settings.h', 'w') C = open('src/core/ext/transport/chttp2/transport/http2_settings.c', 'w') + # 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 + for f in files: + print >> f, '/*' + for line in banner: + print >> f, ' * %s' % line + print >> f, ' */' + print >> f + # 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]) - -put_banner([H,C], ["Automatically generated by tools/codegen/core/gen_settings_ids.py"]) - -print >>H, "#ifndef GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_HTTP2_SETTINGS_H" -print >>H, "#define GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_HTTP2_SETTINGS_H" -print >>H -print >>H, "#include <stdint.h>" -print >>H, "#include <stdbool.h>" -print >>H - -print >>C, "#include \"src/core/ext/transport/chttp2/transport/http2_settings.h\"" -print >>C -print >>C, "#include <grpc/support/useful.h>" -print >>C, "#include \"src/core/lib/transport/http2_errors.h\"" -print >>C + 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]) + +put_banner( + [H, C], + ["Automatically generated by tools/codegen/core/gen_settings_ids.py"]) + +print >> H, "#ifndef GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_HTTP2_SETTINGS_H" +print >> H, "#define GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_HTTP2_SETTINGS_H" +print >> H +print >> H, "#include <stdint.h>" +print >> H, "#include <stdbool.h>" +print >> H + +print >> C, "#include \"src/core/ext/transport/chttp2/transport/http2_settings.h\"" +print >> C +print >> C, "#include <grpc/support/useful.h>" +print >> C, "#include \"src/core/lib/transport/http2_errors.h\"" +print >> C p = perfection.hash_parameters(sorted(x.id for x in _SETTINGS.values())) print p + def hash(i): - i += p.offset - x = i % p.t - y = i / p.t - return x + p.r[y] + i += p.offset + x = i % p.t + y = i / p.t + return x + p.r[y] + -decorated_settings = [DecoratedSetting(hash(setting.id), name, setting) - for name, setting in _SETTINGS.iteritems()] +decorated_settings = [ + DecoratedSetting(hash(setting.id), name, setting) + for name, setting in _SETTINGS.iteritems() +] -print >>H, 'typedef enum {' +print >> H, 'typedef enum {' for decorated_setting in sorted(decorated_settings): - print >>H, ' GRPC_CHTTP2_SETTINGS_%s = %d, /* wire id %d */' % ( - decorated_setting.name, decorated_setting.enum, decorated_setting.setting.id) -print >>H, '} grpc_chttp2_setting_id;' -print >>H -print >>H, '#define GRPC_CHTTP2_NUM_SETTINGS %d' % (max(x.enum for x in decorated_settings) + 1) - -print >>H, 'extern const uint16_t grpc_setting_id_to_wire_id[];' -print >>C, 'const uint16_t grpc_setting_id_to_wire_id[] = {%s};' % ','.join( + print >> H, ' GRPC_CHTTP2_SETTINGS_%s = %d, /* wire id %d */' % ( + decorated_setting.name, decorated_setting.enum, + decorated_setting.setting.id) +print >> H, '} grpc_chttp2_setting_id;' +print >> H +print >> H, '#define GRPC_CHTTP2_NUM_SETTINGS %d' % ( + max(x.enum for x in decorated_settings) + 1) + +print >> H, 'extern const uint16_t grpc_setting_id_to_wire_id[];' +print >> C, 'const uint16_t grpc_setting_id_to_wire_id[] = {%s};' % ','.join( '%d' % s for s in p.slots) -print >>H -print >>H, "bool grpc_wire_id_to_setting_id(uint32_t wire_id, grpc_chttp2_setting_id *out);" +print >> H +print >> H, "bool grpc_wire_id_to_setting_id(uint32_t wire_id, grpc_chttp2_setting_id *out);" cgargs = { - '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 '-' - } -print >>C, """ + '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 '-' +} +print >> C, """ bool grpc_wire_id_to_setting_id(uint32_t wire_id, grpc_chttp2_setting_id *out) { uint32_t i = wire_id %(offset_sign)s %(offset)d; uint32_t x = i %% %(t)d; @@ -118,17 +140,17 @@ bool grpc_wire_id_to_setting_id(uint32_t wire_id, grpc_chttp2_setting_id *out) { switch (y) { """ % cgargs for i, r in enumerate(p.r): - if not r: continue - if r < 0: print >>C, 'case %d: h -= %d; break;' % (i, -r) - else: print >>C, 'case %d: h += %d; break;' % (i, r) -print >>C, """ + if not r: continue + if r < 0: print >> C, 'case %d: h -= %d; break;' % (i, -r) + else: print >> C, 'case %d: h += %d; break;' % (i, r) +print >> C, """ } *out = (grpc_chttp2_setting_id)h; return h < GPR_ARRAY_SIZE(grpc_setting_id_to_wire_id) && grpc_setting_id_to_wire_id[h] == wire_id; } """ % cgargs -print >>H, """ +print >> H, """ typedef enum { GRPC_CHTTP2_CLAMP_INVALID_VALUE, GRPC_CHTTP2_DISCONNECT_ON_INVALID_VALUE @@ -145,25 +167,22 @@ typedef struct { extern const grpc_chttp2_setting_parameters grpc_chttp2_settings_parameters[GRPC_CHTTP2_NUM_SETTINGS]; """ -print >>C, "const grpc_chttp2_setting_parameters grpc_chttp2_settings_parameters[GRPC_CHTTP2_NUM_SETTINGS] = {" +print >> C, "const grpc_chttp2_setting_parameters grpc_chttp2_settings_parameters[GRPC_CHTTP2_NUM_SETTINGS] = {" i = 0 for decorated_setting in sorted(decorated_settings): - while i < decorated_setting.enum: - print >>C, "{NULL, 0, 0, 0, GRPC_CHTTP2_DISCONNECT_ON_INVALID_VALUE, GRPC_HTTP2_PROTOCOL_ERROR}," + while i < decorated_setting.enum: + print >> C, "{NULL, 0, 0, 0, GRPC_CHTTP2_DISCONNECT_ON_INVALID_VALUE, GRPC_HTTP2_PROTOCOL_ERROR}," + i += 1 + print >> C, "{\"%s\", %du, %du, %du, GRPC_CHTTP2_%s, GRPC_HTTP2_%s}," % ( + decorated_setting.name, decorated_setting.setting.default, + decorated_setting.setting.min, decorated_setting.setting.max, + decorated_setting.setting.on_error.behavior, + decorated_setting.setting.on_error.code,) i += 1 - print >>C, "{\"%s\", %du, %du, %du, GRPC_CHTTP2_%s, GRPC_HTTP2_%s}," % ( - decorated_setting.name, - decorated_setting.setting.default, - decorated_setting.setting.min, - decorated_setting.setting.max, - decorated_setting.setting.on_error.behavior, - decorated_setting.setting.on_error.code, - ) - i += 1 -print >>C, "};" - -print >>H -print >>H, "#endif /* GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_HTTP2_SETTINGS_H */" +print >> C, "};" + +print >> H +print >> H, "#endif /* GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_HTTP2_SETTINGS_H */" H.close() C.close() |