// Copyright (c) 2015 Baidu, Inc. // Author: Jiang,Lin (jianglin05@baidu.com) // Date: 2015/05/26 16:17:28 #include <string> #include <sstream> #include "encode_decode.h" namespace json2pb { inline int match_pattern(const std::string& str, int index) { //pattern: _Zxxx_ const char digit = '0'; const int pattern_length = 6; int length = str.size(); if (length <= index || length - index < pattern_length) { return -1; } if (str[index] != '_' || str[index + 1] != 'Z' || str[index + 5] != '_' || !isdigit(str[index + 2]) || !isdigit(str[index + 3]) || !isdigit(str[index + 4])) { return -1; } int sum = (str[index + 2] - digit) * 100 + (str[index + 3] - digit) * 10 + (str[index + 4] - digit); return sum < 256 ? sum : -1; } bool encode_name(const std::string& content, std::string& encoded_content) { int index = 0; size_t begin = 0; bool convert = false; for (std::string::const_iterator it = content.begin(); it != content.end(); ++it, ++index) { if ((!isalnum(*it) && (*it != '_')) || (it == content.begin() && isdigit(*it))) { if (!convert) { encoded_content.clear(); encoded_content.reserve(2*content.size()); convert = true; } encoded_content.append(content, begin, index - begin); begin = index + 1; char pattern[6]; pattern[0] = '_'; pattern[1] = 'Z'; int first = *it / 100; int second = (*it - first * 100) / 10; int third = *it - first * 100 - second * 10; pattern[2] = first + '0'; pattern[3] = second + '0'; pattern[4] = third + '0'; pattern[5] = '_'; encoded_content.append(pattern, sizeof(pattern)); } } if (!convert) { return false; } else { encoded_content.append(content, begin, index - begin); return true; } } bool decode_name(const std::string& content, std::string& decoded_content) { const int pattern_length = 6; int begin = 0; int second = 0; bool convert = false; for (std::string::const_iterator it = content.begin(); it < content.end(); ++it, ++second) { if (*it != '_') { continue; } int val = match_pattern(content, second); if (val != -1) { if (!convert) { decoded_content.clear(); decoded_content.reserve(content.size()); convert = true; } decoded_content.append(content, begin, second - begin); decoded_content.push_back(static_cast<char>(val)); second += pattern_length - 1; begin = second + 1; it += pattern_length - 1; } } if (!convert) { return false; } else { decoded_content.append(content, begin, second - begin); return true; } } } // namespace json2pb