more cases

Author: vitstn

ydb/library/yql/public/decimal/ut/yql_decimal_ut.cpp

@@ -228,6 +228,8 @@ Y_UNIT_TEST_SUITE(TYqlDecimalTest) {
         UNIT_ASSERT(IsError(FromStringEx("E2", 35, 15))); // empty
         UNIT_ASSERT(IsError(FromStringEx("E2E4", 35, 15))); // empty
         UNIT_ASSERT(IsError(FromStringEx("NANE5", 35, 15))); // nan with exp
+        UNIT_ASSERT(IsError(FromStringEx("infE5", 35, 15))); // inf with exp
+        UNIT_ASSERT(IsError(FromStringEx("-infe-5", 35, 15))); // inf with exp
         UNIT_ASSERT(IsError(FromStringEx("2.1E0X", 35, 2))); // not fully parsed exp
         UNIT_ASSERT(IsError(FromStringEx("2.1E+-1", 35, 2))); // two signs
     }

ydb/library/yql/public/decimal/yql_decimal.cpp

@@ -242,6 +242,15 @@ TInt128 FromStringEx(const TStringBuf& str, ui8 precision, ui8 scale) {
                 return Err();
             }
 
+            if (IsInf(r)) {
+                auto p = str.data();
+                if (*p == '+' || *p == '-')
+                    ++p;
+
+                if (!std::isdigit(*p))
+                    return Err();
+            }
+
             if (const auto e = exp > 0 ? std::max(0, s - p) : std::min(0, s)) {
                 if (r && IsNormal(r)) {
                     if (exp > 0)