From 88872f8ac33cfd6ed6e595fe92904aec33ce860f Mon Sep 17 00:00:00 2001
From: nihui <shuizhuyuanluo@126.com>
Date: Sun, 12 Apr 2020 17:58:05 +0800
Subject: [PATCH] fix int64 overflow, fix #1670

---
 tools/onnx/onnx2ncnn.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp
index e86793c2b..f0d2bd2f8 100644
--- a/tools/onnx/onnx2ncnn.cpp
+++ b/tools/onnx/onnx2ncnn.cpp
@@ -63,7 +63,7 @@ static std::vector<int> get_node_attr_ai(const onnx::NodeProto& node, const char
             v.resize(attr.ints_size());
             for (int j=0; j<attr.ints_size(); j++)
             {
-                v[j] = attr.ints(j);
+                v[j] = std::max(std::min(attr.ints(j), (::google::protobuf::int64)INT_MAX), (::google::protobuf::int64)INT_MIN);
             }
 
             break;
@@ -102,7 +102,7 @@ static int get_node_attr_i(const onnx::NodeProto& node, const char* key, int def
         const onnx::AttributeProto& attr = node.attribute(i);
         if (attr.name() == key)
         {
-            return attr.i();
+            return std::max(std::min(attr.i(), (::google::protobuf::int64)INT_MAX), (::google::protobuf::int64)INT_MIN);
         }
     }