Bug fix for NTS YANGs.

[sim/o1-interface.git] / ntsimulator / regxstring / gen_regex.cpp

diff --git a/ntsimulator/regxstring/gen_regex.cpp b/ntsimulator/regxstring/gen_regex.cpp
index e0a4953..42c0660 100644 (file)
--- a/ntsimulator/regxstring/gen_regex.cpp
+++ b/ntsimulator/regxstring/gen_regex.cpp
@@ -14,13 +14,30 @@
 ***************************************************************************/\r
 \r
 #include <string>\r
+#include <cstring>\r
 #include <cstdlib>\r
 #include <iostream>\r
+#include <cstdio>\r
+#include <assert.h>\r
 \r
 #include "regxstring.h"\r
 \r
 using namespace std;\r
 \r
+static char b64_encoding_table[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',\r
+                                    'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',\r
+                                    'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X',\r
+                                    'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f',\r
+                                    'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',\r
+                                    'o', 'p', 'q', 'r', 's', 't', 'u', 'v',\r
+                                    'w', 'x', 'y', 'z', '0', '1', '2', '3',\r
+                                    '4', '5', '6', '7', '8', '9', '+', '/'};\r
+\r
+static char b64_decoding_table[256] = {0};\r
+\r
+static int b64_mod_table[] = {0, 2, 1};\r
+\r
+\r
 static string trim(std::string str){\r
     size_t i = 0,e = str.length();\r
     for(;i < e && std::isspace(str[i]);++i);\r
@@ -54,15 +71,69 @@ static void rand_init(void) {
     srandom(seed);\r
 }\r
 \r
+static uint8_t *b64_decode(const char *data, size_t input_length, size_t *output_length) {\r
+    assert(data);\r
+    assert(input_length);\r
+    assert(output_length);\r
+\r
+    int i, j;\r
+\r
+    //one time compute decoding table\r
+    if(b64_decoding_table['A'] == 0) {\r
+        for(i = 0; i < 64; i++) {\r
+            b64_decoding_table[(unsigned char)b64_encoding_table[i]] = i;\r
+        }\r
+    }\r
+\r
+    if(input_length % 4 != 0) {\r
+        return 0;\r
+    }\r
+\r
+    *output_length = input_length / 4 * 3;\r
+    if(data[input_length - 1] == '=') {\r
+        (*output_length )--;\r
+    }\r
+    if(data[input_length - 2] == '=') {\r
+        (*output_length )--;\r
+    }\r
+\r
+    uint8_t *decoded_data = (uint8_t*)malloc(*output_length + 1);\r
+    if(decoded_data == 0) {\r
+        return 0;\r
+    }\r
+\r
+    for(i = 0, j = 0; i < input_length;) {\r
+        uint32_t sextet_a = data[i] == '=' ? 0 & i++ : b64_decoding_table[(int)data[i++]];\r
+        uint32_t sextet_b = data[i] == '=' ? 0 & i++ : b64_decoding_table[(int)data[i++]];\r
+        uint32_t sextet_c = data[i] == '=' ? 0 & i++ : b64_decoding_table[(int)data[i++]];\r
+        uint32_t sextet_d = data[i] == '=' ? 0 & i++ : b64_decoding_table[(int)data[i++]];\r
+        uint32_t triple = ( sextet_a << 3 * 6 ) + ( sextet_b << 2 * 6 ) + ( sextet_c << 1 * 6 ) + ( sextet_d << 0 * 6 );\r
+\r
+        if(j < *output_length) {\r
+            decoded_data[j++] = (triple >> 2 * 8) & 0xFF;\r
+        }\r
+\r
+        if(j < *output_length) {\r
+            decoded_data[j++] = (triple >> 1 * 8) & 0xFF;\r
+        }\r
+\r
+        if(j < *output_length) {\r
+            decoded_data[j++] = (triple >> 0 * 8) & 0xFF;\r
+        }\r
+    }\r
+\r
+    return decoded_data;\r
+}\r
+\r
 int main(int argc, const char ** argv)\r
 {\r
     CRegxString regxstr;\r
-    string regx = "";\r
+    string regx64 = "";\r
 \r
     switch(argc) {\r
         case 2:\r
             rand_init();\r
-            regx = argv[1];\r
+            regx64 = argv[1];\r
             break;\r
 \r
         case 3:\r
@@ -75,10 +146,17 @@ int main(int argc, const char ** argv)
             }\r
             srand(pseudo_seed);\r
             srandom(pseudo_seed);\r
-            regx = argv[2];\r
+            regx64 = argv[2];\r
             break;\r
     }\r
 \r
+    size_t ol;\r
+    char *x = (char *)b64_decode(regx64.c_str(), strlen(regx64.c_str()), &ol);\r
+    x[ol] = 0;\r
+\r
+    string regx = x;\r
+    free(x);\r
+\r
     regxstr.ParseRegx(pre_handle(regx).c_str());\r
     cout << regxstr.RandString() << endl;\r
     \r