diff --git a/tez-common/src/main/java/org/apache/tez/common/TezUtilsInternal.java b/tez-common/src/main/java/org/apache/tez/common/TezUtilsInternal.java index 477134e4d2..c05d9958c6 100644 --- a/tez-common/src/main/java/org/apache/tez/common/TezUtilsInternal.java +++ b/tez-common/src/main/java/org/apache/tez/common/TezUtilsInternal.java @@ -38,6 +38,7 @@ import java.util.zip.Inflater; import com.google.protobuf.ByteString; +import com.google.protobuf.CodedInputStream; import com.google.protobuf.Descriptors; import com.google.protobuf.TextFormat; import org.apache.hadoop.classification.InterfaceAudience.Private; @@ -77,7 +78,9 @@ public static ConfigurationProto readUserSpecifiedTezConfiguration(String baseDi IOException { File confPBFile = new File(baseDir, TezConstants.TEZ_PB_BINARY_CONF_NAME); try (FileInputStream fis = new FileInputStream(confPBFile)) { - return ConfigurationProto.parseFrom(fis); + CodedInputStream in = CodedInputStream.newInstance(fis); + in.setSizeLimit(Integer.MAX_VALUE); + return ConfigurationProto.parseFrom(in); } }