1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#include "tests/test_util.h"
#include "tests/upb_test.h"
#include "upb/bindings/stdc++/string.h"
#include "google/protobuf/descriptor.upb.h"
#include "upb/pb/decoder.h"
#include "upb/pb/encoder.h"
std::string read_string(const char *filename) {
size_t len;
char *str = upb_readfile(filename, &len);
ASSERT(str);
if (!str) { return std::string(); }
std::string ret = std::string(str, len);
free(str);
return ret;
}
void test_pb_roundtrip() {
std::string input = read_string("google/protobuf/descriptor.pb");
upb::SymbolTable* symtab = upb::SymbolTable::New();
upb::HandlerCache* encoder_cache = upb::pb::Encoder::NewCache();
upb::pb::CodeCache* decoder_cache = upb::pb::CodeCache::New(encoder_cache);
ASSERT(symtab);
ASSERT(encoder_cache);
ASSERT(decoder_cache);
upb::Arena arena;
google_protobuf_FileDescriptorSet *set =
google_protobuf_FileDescriptorSet_parsenew(
upb_stringview_make(input.c_str(), input.size()), &arena);
ASSERT(set);
size_t n;
const google_protobuf_FileDescriptorProto *const *files =
google_protobuf_FileDescriptorSet_file(set, &n);
ASSERT(n == 1);
upb::Status status;
bool ok = symtab->AddFile(files[0], &status);
if (!ok) {
fprintf(stderr, "Error building def: %s\n", upb_status_errmsg(&status));
ASSERT(false);
}
const upb::MessageDef *md =
symtab->LookupMessage("google.protobuf.FileDescriptorSet");
ASSERT(md);
const upb::Handlers* encoder_handlers = encoder_cache->Get(md);
ASSERT(encoder_handlers);
const upb::pb::DecoderMethod* method = decoder_cache->Get(md);
ASSERT(method);
upb::InlinedEnvironment<512> env;
std::string output;
upb::StringSink string_sink(&output);
upb::pb::Encoder* encoder =
upb::pb::Encoder::Create(&env, encoder_handlers, string_sink.input());
upb::pb::Decoder* decoder =
upb::pb::Decoder::Create(&env, method, encoder->input());
ok = upb::BufferSource::PutBuffer(input, decoder->input());
ASSERT(ok);
ASSERT(input == output);
upb::pb::CodeCache::Free(decoder_cache);
upb::HandlerCache::Free(encoder_cache);
upb::SymbolTable::Free(symtab);
}
extern "C" {
int run_tests(int argc, char *argv[]) {
UPB_UNUSED(argc);
UPB_UNUSED(argv);
test_pb_roundtrip();
return 0;
}
}
|