generate_descriptor_proto.sh 3.01 KB
Newer Older
1
#!/usr/bin/env bash
temporal's avatar
temporal committed
2 3 4 5 6 7

# Run this script to regenerate descriptor.pb.{h,cc} after the protocol
# compiler changes.  Since these files are compiled into the protocol compiler
# itself, they cannot be generated automatically by a make rule.  "make check"
# will fail if these files do not match what the protocol compiler would
# generate.
8 9 10 11
#
# HINT:  Flags passed to generate_descriptor_proto.sh will be passed directly
#   to make when building protoc.  This is particularly useful for passing
#   -j4 to run 4 jobs simultaneously.
temporal's avatar
temporal committed
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28

if test ! -e src/google/protobuf/stubs/common.h; then
  cat >&2 << __EOF__
Could not find source code.  Make sure you are running this script from the
root of the distribution tree.
__EOF__
  exit 1
fi

if test ! -e src/Makefile; then
  cat >&2 << __EOF__
Could not find src/Makefile.  You must run ./configure (and perhaps
./autogen.sh) first.
__EOF__
  exit 1
fi

29
cd src
30 31 32 33 34 35 36 37 38 39 40 41 42 43

declare -a RUNTIME_PROTO_FILES=(\
  google/protobuf/any.proto \
  google/protobuf/api.proto \
  google/protobuf/descriptor.proto \
  google/protobuf/duration.proto \
  google/protobuf/empty.proto \
  google/protobuf/field_mask.proto \
  google/protobuf/source_context.proto \
  google/protobuf/struct.proto \
  google/protobuf/timestamp.proto \
  google/protobuf/type.proto \
  google/protobuf/wrappers.proto)

44 45 46 47 48
declare -a COMPILER_PROTO_FILES=(\
  google/protobuf/compiler/plugin.proto \
  google/protobuf/compiler/profile.proto \
)

49
CORE_PROTO_IS_CORRECT=0
50
PROCESS_ROUND=1
51
TMP=$(mktemp -d)
52
echo "Updating descriptor protos..."
53 54
while [ $CORE_PROTO_IS_CORRECT -ne 1 ]
do
55
  echo "Round $PROCESS_ROUND"
56 57
  CORE_PROTO_IS_CORRECT=1

58 59 60 61 62 63 64
  make $@ protoc
  if test $? -ne 0; then
    echo "Failed to build protoc."
    exit 1
  fi

  ./protoc --cpp_out=dllexport_decl=LIBPROTOBUF_EXPORT:$TMP ${RUNTIME_PROTO_FILES[@]} && \
65
  ./protoc --cpp_out=dllexport_decl=LIBPROTOC_EXPORT:$TMP ${COMPILER_PROTO_FILES[@]}
66

67
  for PROTO_FILE in ${RUNTIME_PROTO_FILES[@]} ${COMPILER_PROTO_FILES[@]}; do
68
    BASE_NAME=${PROTO_FILE%.*}
69
    diff ${BASE_NAME}.pb.h $TMP/${BASE_NAME}.pb.h > /dev/null
70 71 72
    if test $? -ne 0; then
      CORE_PROTO_IS_CORRECT=0
    fi
73
    diff ${BASE_NAME}.pb.cc $TMP/${BASE_NAME}.pb.cc > /dev/null
74 75 76 77 78
    if test $? -ne 0; then
      CORE_PROTO_IS_CORRECT=0
    fi
  done

79 80 81
  # Only override the output if the files are different to avoid re-compilation
  # of the protoc.
  if [ $CORE_PROTO_IS_CORRECT -ne 1 ]; then
82
    for PROTO_FILE in ${RUNTIME_PROTO_FILES[@]} ${COMPILER_PROTO_FILES[@]}; do
83 84 85 86 87
      BASE_NAME=${PROTO_FILE%.*}
      mv $TMP/${BASE_NAME}.pb.h ${BASE_NAME}.pb.h
      mv $TMP/${BASE_NAME}.pb.cc ${BASE_NAME}.pb.cc
    done
  fi
88 89

  PROCESS_ROUND=$((PROCESS_ROUND + 1))
90
done
91
cd ..
92

93
if test -x objectivec/generate_well_known_types.sh; then
94
  echo "Generating messages for objc."
95
  objectivec/generate_well_known_types.sh $@
96
fi
97 98 99 100 101

if test -x csharp/generate_protos.sh; then
  echo "Generating messages for C#."
  csharp/generate_protos.sh $@
fi
102 103 104 105 106

if test -x php/generate_descriptor_protos.sh; then
  echo "Generating messages for PHP."
  php/generate_descriptor_protos.sh $@
fi