From 57abebaaf902436f8de8e50f7054c25b391067ac Mon Sep 17 00:00:00 2001 From: Joshua Haberman Date: Sun, 31 Jul 2011 15:39:19 -0700 Subject: Fixed "make descriptorgen". --- tools/upbc.c | 119 +++++++++++++++++++++++------------------------------------ 1 file changed, 47 insertions(+), 72 deletions(-) (limited to 'tools/upbc.c') diff --git a/tools/upbc.c b/tools/upbc.c index 956a1b2..ad19985 100644 --- a/tools/upbc.c +++ b/tools/upbc.c @@ -13,59 +13,39 @@ #include #include #include -#include "upb_def.h" -#include "upb_msg.h" -#include "upb_glue.h" -#include "upb_strstream.h" -#include "upb_decoder.h" +#include "upb/bytestream.h" +#include "upb/def.h" +#include "upb/msg.h" +#include "upb/pb/glue.h" /* These are in-place string transformations that do not change the length of * the string (and thus never need to re-allocate). */ // Convert to C identifier: foo.bar.Baz -> foo_bar_Baz. -static void to_cident(upb_string *str) -{ - upb_strlen_t len = upb_string_len(str); - char *buf = upb_string_getrwbuf(str, len); - for(int32_t i = 0; i < len; i++) - if(buf[i] == '.' || buf[i] == '/') - buf[i] = '_'; +static void to_cident(char *str) { + for (; *str; ++str) { + if(*str == '.' || *str == '/') *str = '_'; + } } // Convert to C proprocessor identifier: foo.bar.Baz -> FOO_BAR_BAZ. -static void to_preproc(upb_string *str) -{ +static void to_preproc(char *str) { to_cident(str); - upb_strlen_t len = upb_string_len(str); - char *buf = upb_string_getrwbuf(str, len); - for(int32_t i = 0; i < len; i++) - buf[i] = toupper(buf[i]); -} - -static int my_memrchr(const char *data, char c, size_t len) -{ - int off = len-1; - while(off > 0 && data[off] != c) --off; - return off; + for (; *str; ++str) *str = toupper(*str); } /* The _const.h file defines the constants (enums) defined in the .proto * file. */ static void write_const_h(upb_def *defs[], int num_entries, char *outfile_name, - FILE *stream) -{ + FILE *stream) { /* Header file prologue. */ - upb_string *include_guard_name = upb_strdupc(outfile_name); + char *include_guard_name = strdup(outfile_name); to_preproc(include_guard_name); - /* A bit cheesy, but will do the job. */ - upb_strlen_t len = upb_string_len(include_guard_name); - char *buf = upb_string_getrwbuf(include_guard_name, len); - buf[len-1] = 'C'; fputs("/* This file was generated by upbc (the upb compiler). " "Do not edit. */\n\n", stream), - fprintf(stream, "#ifndef " UPB_STRFMT "\n", UPB_STRARG(include_guard_name)); - fprintf(stream, "#define " UPB_STRFMT "\n\n", UPB_STRARG(include_guard_name)); + fprintf(stream, "#ifndef %s\n", include_guard_name); + fprintf(stream, "#define %s\n\n", include_guard_name); fputs("#ifdef __cplusplus\n", stream); fputs("extern \"C\" {\n", stream); fputs("#endif\n\n", stream); @@ -75,34 +55,29 @@ static void write_const_h(upb_def *defs[], int num_entries, char *outfile_name, for(int i = 0; i < num_entries; i++) { /* Foreach enum */ if(defs[i]->type != UPB_DEF_ENUM) continue; upb_enumdef *enumdef = upb_downcast_enumdef(defs[i]); - upb_string *enum_name = upb_strdup(UPB_UPCAST(enumdef)->fqname); - upb_string *enum_val_prefix = upb_strdup(enum_name); + char *enum_name = strdup(upb_def_fqname(UPB_UPCAST(enumdef))); + char *enum_val_prefix = strdup(enum_name); to_cident(enum_name); - - const char *data = upb_string_getrobuf(enum_val_prefix); - upb_strlen_t len = upb_string_len(enum_val_prefix); - upb_strlen_t lastsep = my_memrchr(data, UPB_SYMBOL_SEPARATOR, len); - upb_string_getrwbuf(enum_val_prefix, lastsep + 1); to_preproc(enum_val_prefix); - fprintf(stream, "typedef enum " UPB_STRFMT " {\n", UPB_STRARG(enum_name)); + fprintf(stream, "typedef enum %s {\n", enum_name); bool first = true; /* Foreach enum value. */ for (upb_enum_iter iter = upb_enum_begin(enumdef); !upb_enum_done(iter); iter = upb_enum_next(enumdef, iter)) { - upb_string *value_name = upb_strdup(upb_enum_iter_name(iter)); + char *value_name = strdup(upb_enum_iter_name(iter)); + uint32_t value = upb_enum_iter_number(iter); to_preproc(value_name); - /* " GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_TYPE_UINT32 = 13," */ + /* " GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_TYPE_TYPE_UINT32 = 13," */ if (!first) fputs(",\n", stream); first = false; - fprintf(stream, " " UPB_STRFMT UPB_STRFMT " = %" PRIu32, - UPB_STRARG(enum_val_prefix), UPB_STRARG(value_name), upb_enum_iter_number(iter)); - upb_string_unref(value_name); + fprintf(stream, " %s_%s = %" PRIu32, enum_val_prefix, value_name, value); + free(value_name); } - fprintf(stream, "\n} " UPB_STRFMT ";\n\n", UPB_STRARG(enum_name)); - upb_string_unref(enum_name); - upb_string_unref(enum_val_prefix); + fprintf(stream, "\n} %s;\n\n", enum_name); + free(enum_name); + free(enum_val_prefix); } /* Constants for field names and numbers. */ @@ -110,30 +85,32 @@ static void write_const_h(upb_def *defs[], int num_entries, char *outfile_name, for(int i = 0; i < num_entries; i++) { /* Foreach enum */ upb_msgdef *m = upb_dyncast_msgdef(defs[i]); if(!m) continue; - upb_string *msg_name = upb_strdup(UPB_UPCAST(m)->fqname); - upb_string *msg_val_prefix = upb_strdup(msg_name); + char *msg_name = strdup(upb_def_fqname(UPB_UPCAST(m))); + char *msg_val_prefix = strdup(msg_name); to_preproc(msg_val_prefix); upb_msg_iter i; for(i = upb_msg_begin(m); !upb_msg_done(i); i = upb_msg_next(m, i)) { upb_fielddef *f = upb_msg_iter_field(i); - upb_string *preproc_field_name = upb_strdup(f->name); + char *preproc_field_name = strdup(f->name); to_preproc(preproc_field_name); - fprintf(stream, "#define " UPB_STRFMT "_" UPB_STRFMT "_FIELDNUM %d\n", - UPB_STRARG(msg_val_prefix), UPB_STRARG(preproc_field_name), f->number); - fprintf(stream, "#define " UPB_STRFMT "_" UPB_STRFMT "_FIELDNAME \"" - UPB_STRFMT "\"\n", UPB_STRARG(msg_val_prefix), UPB_STRARG(preproc_field_name), UPB_STRARG(f->name)); - upb_string_unref(preproc_field_name); + fprintf(stream, "#define %s_%s__FIELDNUM %d\n", + msg_val_prefix, preproc_field_name, upb_fielddef_number(f)); + fprintf(stream, "#define %s_%s__FIELDNAME \"%s\"\n", + msg_val_prefix, preproc_field_name, f->name); + fprintf(stream, "#define %s_%s__FIELDTYPE %d\n\n", + msg_val_prefix, preproc_field_name, f->type); + free(preproc_field_name); } - upb_string_unref(msg_val_prefix); - upb_string_unref(msg_name); + free(msg_val_prefix); + free(msg_name); } /* Epilogue. */ fputs("#ifdef __cplusplus\n", stream); fputs("} /* extern \"C\" */\n", stream); fputs("#endif\n\n", stream); - fprintf(stream, "#endif /* " UPB_STRFMT " */\n", UPB_STRARG(include_guard_name)); - upb_string_unref(include_guard_name); + fprintf(stream, "#endif /* %s */\n", include_guard_name); + free(include_guard_name); } const char usage[] = @@ -146,15 +123,13 @@ const char usage[] = " of using the input file as a basename.\n" ; -void usage_err(char *err) -{ +void usage_err(char *err) { fprintf(stderr, "upbc: %s\n\n", err); fputs(usage, stderr); exit(1); } -void error(char *err, ...) -{ +void error(char *err, ...) { va_list args; va_start(args, err); fprintf(stderr, "upbc: "); @@ -163,8 +138,7 @@ void error(char *err, ...) exit(1); } -int main(int argc, char *argv[]) -{ +int main(int argc, char *argv[]) { /* Parse arguments. */ char *outfile_base = NULL, *input_file = NULL; for(int i = 1; i < argc; i++) { @@ -184,7 +158,8 @@ int main(int argc, char *argv[]) if(!outfile_base) outfile_base = input_file; // Read and parse input file. - upb_string *descriptor = upb_strreadfile(input_file); + size_t len; + char *descriptor = upb_readfile(input_file, &len); if(!descriptor) error("Couldn't read input file."); @@ -192,9 +167,9 @@ int main(int argc, char *argv[]) // importing descriptor.proto. upb_symtab *s = upb_symtab_new(); upb_status status = UPB_STATUS_INIT; - upb_parsedesc(s, descriptor, &status); + upb_read_descriptor(s, descriptor, len, &status); if(!upb_ok(&status)) { - upb_printerr(&status); + upb_status_print(&status, stderr); error("Failed to parse input file descriptor\n"); } upb_status_uninit(&status); @@ -213,7 +188,7 @@ int main(int argc, char *argv[]) write_const_h(defs, symcount, h_const_filename, h_const_file); for (int i = 0; i < symcount; i++) upb_def_unref(defs[i]); free(defs); - upb_string_unref(descriptor); + free(descriptor); upb_symtab_unref(s); fclose(h_const_file); -- cgit v1.2.3