util.h 7.4 KB

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