util.h 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248
  1. /* SPDX-License-Identifier: GPL-2.0-or-later */
  2. #ifndef UTIL_H
  3. #define UTIL_H
  4. #include <stdlib.h>
  5. #include <stdarg.h>
  6. #include <stdbool.h>
  7. #include <getopt.h>
  8. /*
  9. * Copyright 2011 The Chromium Authors, All Rights Reserved.
  10. * Copyright 2008 Jon Loeliger, Freescale Semiconductor, Inc.
  11. */
  12. #ifdef __GNUC__
  13. #if __GNUC__ >= 5 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 4)
  14. #define PRINTF(i, j) __attribute__((format (gnu_printf, i, j)))
  15. #else
  16. #define PRINTF(i, j) __attribute__((format (printf, i, j)))
  17. #endif
  18. #define NORETURN __attribute__((noreturn))
  19. #else
  20. #define PRINTF(i, j)
  21. #define NORETURN
  22. #endif
  23. #define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
  24. #define stringify(s) stringify_(s)
  25. #define stringify_(s) #s
  26. static inline void NORETURN PRINTF(1, 2) die(const char *str, ...)
  27. {
  28. va_list ap;
  29. va_start(ap, str);
  30. fprintf(stderr, "FATAL ERROR: ");
  31. vfprintf(stderr, str, ap);
  32. va_end(ap);
  33. exit(1);
  34. }
  35. static inline void *xmalloc(size_t len)
  36. {
  37. void *new = malloc(len);
  38. if (!new)
  39. die("malloc() failed\n");
  40. return new;
  41. }
  42. static inline void *xrealloc(void *p, size_t len)
  43. {
  44. void *new = realloc(p, len);
  45. if (!new)
  46. die("realloc() failed (len=%zd)\n", len);
  47. return new;
  48. }
  49. extern char *xstrdup(const char *s);
  50. extern int PRINTF(2, 3) xasprintf(char **strp, const char *fmt, ...);
  51. extern int PRINTF(2, 3) xasprintf_append(char **strp, const char *fmt, ...);
  52. extern int xavsprintf_append(char **strp, const char *fmt, va_list ap);
  53. extern char *join_path(const char *path, const char *name);
  54. /**
  55. * Check a property of a given length to see if it is all printable and
  56. * has a valid terminator. The property can contain either a single string,
  57. * or multiple strings each of non-zero length.
  58. *
  59. * @param data The string to check
  60. * @param len The string length including terminator
  61. * @return 1 if a valid printable string, 0 if not
  62. */
  63. bool util_is_printable_string(const void *data, int len);
  64. /*
  65. * Parse an escaped character starting at index i in string s. The resulting
  66. * character will be returned and the index i will be updated to point at the
  67. * character directly after the end of the encoding, this may be the '\0'
  68. * terminator of the string.
  69. */
  70. char get_escape_char(const char *s, int *i);
  71. /**
  72. * Read a device tree file into a buffer. This will report any errors on
  73. * stderr.
  74. *
  75. * @param filename The filename to read, or - for stdin
  76. * @param len If non-NULL, the amount of data we managed to read
  77. * @return Pointer to allocated buffer containing fdt, or NULL on error
  78. */
  79. char *utilfdt_read(const char *filename, size_t *len);
  80. /**
  81. * Read a device tree file into a buffer. Does not report errors, but only
  82. * returns them. The value returned can be passed to strerror() to obtain
  83. * an error message for the user.
  84. *
  85. * @param filename The filename to read, or - for stdin
  86. * @param buffp Returns pointer to buffer containing fdt
  87. * @param len If non-NULL, the amount of data we managed to read
  88. * @return 0 if ok, else an errno value representing the error
  89. */
  90. int utilfdt_read_err(const char *filename, char **buffp, size_t *len);
  91. /**
  92. * Write a device tree buffer to a file. This will report any errors on
  93. * stderr.
  94. *
  95. * @param filename The filename to write, or - for stdout
  96. * @param blob Pointer to buffer containing fdt
  97. * @return 0 if ok, -1 on error
  98. */
  99. int utilfdt_write(const char *filename, const void *blob);
  100. /**
  101. * Write a device tree buffer to a file. Does not report errors, but only
  102. * returns them. The value returned can be passed to strerror() to obtain
  103. * an error message for the user.
  104. *
  105. * @param filename The filename to write, or - for stdout
  106. * @param blob Pointer to buffer containing fdt
  107. * @return 0 if ok, else an errno value representing the error
  108. */
  109. int utilfdt_write_err(const char *filename, const void *blob);
  110. /**
  111. * Decode a data type string. The purpose of this string
  112. *
  113. * The string consists of an optional character followed by the type:
  114. * Modifier characters:
  115. * hh or b 1 byte
  116. * h 2 byte
  117. * l 4 byte, default
  118. *
  119. * Type character:
  120. * s string
  121. * i signed integer
  122. * u unsigned integer
  123. * x hex
  124. *
  125. * TODO: Implement ll modifier (8 bytes)
  126. * TODO: Implement o type (octal)
  127. *
  128. * @param fmt Format string to process
  129. * @param type Returns type found(s/d/u/x), or 0 if none
  130. * @param size Returns size found(1,2,4,8) or 4 if none
  131. * @return 0 if ok, -1 on error (no type given, or other invalid format)
  132. */
  133. int utilfdt_decode_type(const char *fmt, int *type, int *size);
  134. /*
  135. * This is a usage message fragment for the -t option. It is the format
  136. * supported by utilfdt_decode_type.
  137. */
  138. #define USAGE_TYPE_MSG \
  139. "<type>\ts=string, i=int, u=unsigned, x=hex\n" \
  140. "\tOptional modifier prefix:\n" \
  141. "\t\thh or b=byte, h=2 byte, l=4 byte (default)";
  142. /**
  143. * Print property data in a readable format to stdout
  144. *
  145. * Properties that look like strings will be printed as strings. Otherwise
  146. * the data will be displayed either as cells (if len is a multiple of 4
  147. * bytes) or bytes.
  148. *
  149. * If len is 0 then this function does nothing.
  150. *
  151. * @param data Pointers to property data
  152. * @param len Length of property data
  153. */
  154. void utilfdt_print_data(const char *data, int len);
  155. /**
  156. * Show source version and exit
  157. */
  158. void NORETURN util_version(void);
  159. /**
  160. * Show usage and exit
  161. *
  162. * This helps standardize the output of various utils. You most likely want
  163. * to use the usage() helper below rather than call this.
  164. *
  165. * @param errmsg If non-NULL, an error message to display
  166. * @param synopsis The initial example usage text (and possible examples)
  167. * @param short_opts The string of short options
  168. * @param long_opts The structure of long options
  169. * @param opts_help An array of help strings (should align with long_opts)
  170. */
  171. void NORETURN util_usage(const char *errmsg, const char *synopsis,
  172. const char *short_opts,
  173. struct option const long_opts[],
  174. const char * const opts_help[]);
  175. /**
  176. * Show usage and exit
  177. *
  178. * If you name all your usage variables with usage_xxx, then you can call this
  179. * help macro rather than expanding all arguments yourself.
  180. *
  181. * @param errmsg If non-NULL, an error message to display
  182. */
  183. #define usage(errmsg) \
  184. util_usage(errmsg, usage_synopsis, usage_short_opts, \
  185. usage_long_opts, usage_opts_help)
  186. /**
  187. * Call getopt_long() with standard options
  188. *
  189. * Since all util code runs getopt in the same way, provide a helper.
  190. */
  191. #define util_getopt_long() getopt_long(argc, argv, usage_short_opts, \
  192. usage_long_opts, NULL)
  193. /* Helper for aligning long_opts array */
  194. #define a_argument required_argument
  195. /* Helper for usage_short_opts string constant */
  196. #define USAGE_COMMON_SHORT_OPTS "hV"
  197. /* Helper for usage_long_opts option array */
  198. #define USAGE_COMMON_LONG_OPTS \
  199. {"help", no_argument, NULL, 'h'}, \
  200. {"version", no_argument, NULL, 'V'}, \
  201. {NULL, no_argument, NULL, 0x0}
  202. /* Helper for usage_opts_help array */
  203. #define USAGE_COMMON_OPTS_HELP \
  204. "Print this help and exit", \
  205. "Print version and exit", \
  206. NULL
  207. /* Helper for getopt case statements */
  208. #define case_USAGE_COMMON_FLAGS \
  209. case 'h': usage(NULL); \
  210. case 'V': util_version(); \
  211. case '?': usage("unknown option");
  212. #endif /* UTIL_H */