value_utils.h 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174
  1. /**
  2. * @file value_utils.h
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * Redistribution and use in source and binary forms, with or without
  8. * modification, are permitted provided that the following conditions are met:
  9. * 1. Redistributions of source code must retain the above copyright
  10. * notice, this list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * 3. Neither the name of the author nor the
  15. * names of its contributors may be used to endorse or promote products
  16. * derived from this software without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  19. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21. * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  22. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28. */
  29. #ifndef NCD_VALUE_UTILS_H
  30. #define NCD_VALUE_UTILS_H
  31. #include <stdint.h>
  32. #include <limits.h>
  33. #include <misc/debug.h>
  34. #include <misc/parse_number.h>
  35. #include <misc/strdup.h>
  36. #include <misc/balloc.h>
  37. #include <system/BTime.h>
  38. #include <ncd/NCDVal.h>
  39. #include <ncd/NCDStringIndex.h>
  40. #include <ncd/static_strings.h>
  41. static int ncd_is_none (NCDValRef val);
  42. static NCDValRef ncd_make_boolean (NCDValMem *mem, int value, NCDStringIndex *string_index);
  43. static int ncd_read_boolean (NCDValRef val);
  44. static int ncd_read_uintmax (NCDValRef string, uintmax_t *out) WARN_UNUSED;
  45. static int ncd_read_time (NCDValRef string, btime_t *out) WARN_UNUSED;
  46. static NCD_string_id_t ncd_get_string_id (NCDValRef string, NCDStringIndex *string_index);
  47. static NCDValRef ncd_make_uintmax (NCDValMem *mem, uintmax_t value);
  48. static char * ncd_strdup (NCDValRef stringnonulls);
  49. static int ncd_is_none (NCDValRef string)
  50. {
  51. ASSERT(NCDVal_IsString(string))
  52. if (NCDVal_IsIdString(string)) {
  53. return NCDVal_IdStringId(string) == NCD_STRING_NONE;
  54. } else {
  55. return NCDVal_StringEquals(string, "<none>");
  56. }
  57. }
  58. static NCDValRef ncd_make_boolean (NCDValMem *mem, int value, NCDStringIndex *string_index)
  59. {
  60. ASSERT(mem)
  61. ASSERT(string_index)
  62. NCD_string_id_t str_id = (value ? NCD_STRING_TRUE : NCD_STRING_FALSE);
  63. return NCDVal_NewIdString(mem, str_id, string_index);
  64. }
  65. static int ncd_read_boolean (NCDValRef string)
  66. {
  67. ASSERT(NCDVal_IsString(string))
  68. if (NCDVal_IsIdString(string)) {
  69. return NCDVal_IdStringId(string) == NCD_STRING_TRUE;
  70. } else {
  71. return NCDVal_StringEquals(string, "true");
  72. }
  73. }
  74. static int ncd_read_uintmax (NCDValRef string, uintmax_t *out)
  75. {
  76. ASSERT(NCDVal_IsString(string))
  77. ASSERT(out)
  78. size_t length = NCDVal_StringLength(string);
  79. if (NCDVal_IsContinuousString(string)) {
  80. return parse_unsigned_integer_bin(NCDVal_StringData(string), length, out);
  81. }
  82. b_cstring cstr = NCDVal_StringCstring(string);
  83. return parse_unsigned_integer_cstr(cstr, 0, cstr.length, out);
  84. }
  85. static int ncd_read_time (NCDValRef string, btime_t *out)
  86. {
  87. ASSERT(NCDVal_IsString(string))
  88. ASSERT(out)
  89. uintmax_t n;
  90. if (!ncd_read_uintmax(string, &n)) {
  91. return 0;
  92. }
  93. if (n > INT64_MAX) {
  94. return 0;
  95. }
  96. *out = n;
  97. return 1;
  98. }
  99. static NCD_string_id_t ncd_get_string_id (NCDValRef string, NCDStringIndex *string_index)
  100. {
  101. ASSERT(NCDVal_IsString(string))
  102. ASSERT(string_index)
  103. if (NCDVal_IsIdString(string)) {
  104. return NCDVal_IdStringId(string);
  105. } else if (NCDVal_IsContinuousString(string)) {
  106. return NCDStringIndex_GetBin(string_index, NCDVal_StringData(string), NCDVal_StringLength(string));
  107. }
  108. b_cstring cstr = NCDVal_StringCstring(string);
  109. char *temp = b_cstring_strdup(cstr, 0, cstr.length);
  110. if (!temp) {
  111. return -1;
  112. }
  113. NCD_string_id_t res = NCDStringIndex_GetBin(string_index, temp, cstr.length);
  114. BFree(temp);
  115. return res;
  116. }
  117. static NCDValRef ncd_make_uintmax (NCDValMem *mem, uintmax_t value)
  118. {
  119. ASSERT(mem)
  120. int size = compute_decimal_repr_size(value);
  121. NCDValRef val = NCDVal_NewStringUninitialized(mem, size);
  122. if (!NCDVal_IsInvalid(val)) {
  123. char *data = (char *)NCDVal_StringData(val);
  124. generate_decimal_repr(value, data, size);
  125. }
  126. return val;
  127. }
  128. static char * ncd_strdup (NCDValRef stringnonulls)
  129. {
  130. ASSERT(NCDVal_IsStringNoNulls(stringnonulls))
  131. size_t length = NCDVal_StringLength(stringnonulls);
  132. if (NCDVal_IsContinuousString(stringnonulls)) {
  133. return b_strdup_bin(NCDVal_StringData(stringnonulls), length);
  134. }
  135. b_cstring cstr = NCDVal_StringCstring(stringnonulls);
  136. return b_cstring_strdup(cstr, 0, cstr.length);
  137. }
  138. #endif