toolkit.c 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213
  1. #include "precompile.h"
  2. #include "toolkit.h"
  3. #include "strutil.h"
  4. #ifndef _WIN32
  5. #include <locale.h>
  6. #endif //NOT _WIN32
  7. #define TAG TOOLKIT_TAG("main")
  8. #if 0
  9. static int inet_ntop4(const unsigned char* src, char* dst, size_t size);
  10. static int inet_ntop6(const unsigned char* src, char* dst, size_t size);
  11. int toolkit_inet_ntop(int af, const void* src, char* dst, size_t size)
  12. {
  13. switch (af) {
  14. case AF_INET:
  15. return (inet_ntop4(src, dst, size));
  16. case AF_INET6:
  17. return (inet_ntop6(src, dst, size));
  18. default:
  19. return TOOLKIT_EAFNOSUPPORT;
  20. }
  21. /* NOTREACHED */
  22. }
  23. static int inet_ntop4(const unsigned char* src, char* dst, size_t size)
  24. {
  25. static const char fmt[] = "%u.%u.%u.%u";
  26. char tmp[16];
  27. int l;
  28. l = snprintf(tmp, sizeof(tmp), fmt, src[0], src[1], src[2], src[3]);
  29. if (l <= 0 || (size_t)l >= size) {
  30. return TOOLKIT_ENOSPC;
  31. }
  32. strscpy(dst, tmp, size);
  33. return 0;
  34. }
  35. static int inet_ntop6(const unsigned char* src, char* dst, size_t size)
  36. {
  37. /*
  38. * Note that int32_t and int16_t need only be "at least" large enough
  39. * to contain a value of the specified size. On some systems, like
  40. * Crays, there is no such thing as an integer variable with 16 bits.
  41. * Keep this in mind if you think this function should have been coded
  42. * to use pointer overlays. All the world's not a VAX.
  43. */
  44. char tmp[46], * tp;
  45. struct { int base, len; } best, cur;
  46. unsigned int words[sizeof(struct in6_addr) / sizeof(uint16_t)];
  47. int i;
  48. /*
  49. * Preprocess:
  50. * Copy the input (bytewise) array into a wordwise array.
  51. * Find the longest run of 0x00's in src[] for :: shorthanding.
  52. */
  53. memset(words, '\0', sizeof words);
  54. for (i = 0; i < (int)sizeof(struct in6_addr); i++)
  55. words[i / 2] |= (src[i] << ((1 - (i % 2)) << 3));
  56. best.base = -1;
  57. best.len = 0;
  58. cur.base = -1;
  59. cur.len = 0;
  60. for (i = 0; i < (int)array_size(words); i++) {
  61. if (words[i] == 0) {
  62. if (cur.base == -1)
  63. cur.base = i, cur.len = 1;
  64. else
  65. cur.len++;
  66. } else {
  67. if (cur.base != -1) {
  68. if (best.base == -1 || cur.len > best.len)
  69. best = cur;
  70. cur.base = -1;
  71. }
  72. }
  73. }
  74. if (cur.base != -1) {
  75. if (best.base == -1 || cur.len > best.len)
  76. best = cur;
  77. }
  78. if (best.base != -1 && best.len < 2)
  79. best.base = -1;
  80. /*
  81. * Format the result.
  82. */
  83. tp = tmp;
  84. for (i = 0; i < (int)array_size(words); i++) {
  85. /* Are we inside the best run of 0x00's? */
  86. if (best.base != -1 && i >= best.base &&
  87. i < (best.base + best.len)) {
  88. if (i == best.base)
  89. *tp++ = ':';
  90. continue;
  91. }
  92. /* Are we following an initial run of 0x00s or any real hex? */
  93. if (i != 0)
  94. *tp++ = ':';
  95. /* Is this address an encapsulated IPv4? */
  96. if (i == 6 && best.base == 0 && (best.len == 6 ||
  97. (best.len == 7 && words[7] != 0x0001) ||
  98. (best.len == 5 && words[5] == 0xffff))) {
  99. int err = inet_ntop4(src + 12, tp, sizeof tmp - (tp - tmp));
  100. if (err)
  101. return err;
  102. tp += strlen(tp);
  103. break;
  104. }
  105. tp += sprintf(tp, "%x", words[i]);
  106. }
  107. /* Was it a trailing run of 0x00's? */
  108. if (best.base != -1 && (best.base + best.len) == array_size(words))
  109. *tp++ = ':';
  110. *tp++ = '\0';
  111. if (TOOLKIT_E2BIG == strscpy(dst, tmp, size))
  112. return TOOLKIT_ENOSPC;
  113. return 0;
  114. }
  115. #endif
  116. int toolkit_ip4_name(const struct sockaddr_in* src, char* dst, size_t size)
  117. {
  118. return toolkit_inet_ntop(AF_INET, &src->sin_addr, dst, size);
  119. }
  120. int toolkit_ip6_name(const struct sockaddr_in6* src, char* dst, size_t size)
  121. {
  122. return toolkit_inet_ntop(AF_INET6, &src->sin6_addr, dst, size);
  123. }
  124. void toolkit_free_environ(toolkit_env_item_t* envitems, int count)
  125. {
  126. int i;
  127. for (i = 0; i < count; i++) {
  128. FREE(envitems[i].name);
  129. }
  130. FREE(envitems);
  131. }
  132. void toolkit_free_cpu_info(toolkit_cpu_info_t* cpu_infos, int count)
  133. {
  134. int i;
  135. for (i = 0; i < count; i++) {
  136. FREE(cpu_infos[i].model);
  137. }
  138. FREE(cpu_infos);
  139. }
  140. /** WARNING: need to FREE the copy msg!!!!*/
  141. static const char* toolkit__unknown_err_code(int err)
  142. {
  143. char buf[32];
  144. char* copy;
  145. snprintf(buf, sizeof(buf), "Unknown system error %d", err);
  146. copy = STRDUP(buf);
  147. return copy != NULL ? copy : "Unknown system error";
  148. }
  149. #define TOOLKIT_STRERROR_GEN(name, msg) case TOOLKIT_ ## name: return msg;
  150. const char* toolkit_strerror(int err)
  151. {
  152. switch (err) {
  153. TOOLKIT_ERRNO_MAP(TOOLKIT_STRERROR_GEN)
  154. }
  155. return toolkit__unknown_err_code(err);
  156. }
  157. #undef TOOLKIT_STRERROR_GEN
  158. #define TOOLKIT_STRERROR_GEN_R(name, msg) \
  159. case TOOLKIT_ ## name: \
  160. snprintf(buf, buflen, "%s", msg); break;
  161. char* toolkit_strerror_r(int err, char* buf, size_t buflen)
  162. {
  163. switch (err) {
  164. TOOLKIT_ERRNO_MAP(TOOLKIT_STRERROR_GEN_R)
  165. default: snprintf(buf, buflen, "Unknown system error %d", err);
  166. }
  167. return buf;
  168. }
  169. #undef TOOLKIT_STRERROR_GEN_R
  170. #define TOOLKIT_ERR_NAME_GEN(name, _) case TOOLKIT_ ## name: return #name;
  171. const char* toolkit_err_name(int err)
  172. {
  173. switch (err) {
  174. TOOLKIT_ERRNO_MAP(TOOLKIT_ERR_NAME_GEN)
  175. }
  176. return toolkit__unknown_err_code(err);
  177. }
  178. #undef TOOLKIT_ERR_NAME_GEN
  179. #define TOOLKIT_ERR_NAME_GEN_R(name, _) \
  180. case TOOLKIT_## name: \
  181. snprintf(buf, buflen, "%s", #name); break;
  182. char* toolkit_err_name_r(int err, char* buf, size_t buflen)
  183. {
  184. switch (err) {
  185. TOOLKIT_ERRNO_MAP(TOOLKIT_ERR_NAME_GEN_R)
  186. default: snprintf(buf, buflen, "Unknown system error %d", err);
  187. }
  188. return buf;
  189. }
  190. #undef TOOLKIT_ERR_NAME_GEN_R