TEZ-4584: TezUtilsInternal.readUserSpecifiedTezConfiguration throws c…

…om.google.protobuf.CodedInputStream exception
apache · Oct 31, 2024 · 0712849 · 0712849
1 parent 607b2bc
commit 0712849
Showing 1 changed file with 4 additions and 1 deletion.
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
@@ -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);
     }
   }