1#!/usr/bin/python3 2# 3# Copyright (c) 2009-2021, Google LLC 4# All rights reserved. 5# 6# Redistribution and use in source and binary forms, with or without 7# modification, are permitted provided that the following conditions are met: 8# * Redistributions of source code must retain the above copyright 9# notice, this list of conditions and the following disclaimer. 10# * Redistributions in binary form must reproduce the above copyright 11# notice, this list of conditions and the following disclaimer in the 12# documentation and/or other materials provided with the distribution. 13# * Neither the name of Google LLC nor the 14# names of its contributors may be used to endorse or promote products 15# derived from this software without specific prior written permission. 16# 17# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 18# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20# DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY 21# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 26# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 28import sys 29import re 30 31include = sys.argv[1] 32msg_basename = sys.argv[2] 33count = 1 34 35m = re.search(r'(.*\D)(\d+)$', sys.argv[2]) 36if m: 37 msg_basename = m.group(1) 38 count = int(m.group(2)) 39 40print(''' 41#include "{include}" 42 43char buf[1]; 44 45int main() {{ 46'''.format(include=include)) 47 48def RefMessage(name): 49 print(''' 50 {{ 51 {name} proto; 52 proto.ParseFromArray(buf, 0); 53 proto.SerializePartialToArray(&buf[0], 0); 54 }} 55 '''.format(name=name)) 56 57RefMessage(msg_basename) 58 59for i in range(2, count + 1): 60 RefMessage(msg_basename + str(i)) 61 62print(''' 63 return 0; 64}''') 65