aboutsummaryrefslogtreecommitdiff
path: root/src/chunk.h
blob: 22594b154e19d852e5d488c6f41f27054abc24db (plain)
  1. #ifndef CMARK_CHUNK_H
  2. #define CMARK_CHUNK_H
  3. #include <string.h>
  4. #include <ctype.h>
  5. #include <stdlib.h>
  6. #include <assert.h>
  7. #include "buffer.h"
  8. typedef struct {
  9. unsigned char *data;
  10. int len;
  11. int alloc; // also implies a NULL-terminated string
  12. } cmark_chunk;
  13. static inline void cmark_chunk_free(cmark_chunk *c)
  14. {
  15. if (c->alloc)
  16. free(c->data);
  17. c->data = NULL;
  18. c->alloc = 0;
  19. c->len = 0;
  20. }
  21. static inline void cmark_chunk_ltrim(cmark_chunk *c)
  22. {
  23. assert(!c->alloc);
  24. while (c->len && isspace(c->data[0])) {
  25. c->data++;
  26. c->len--;
  27. }
  28. }
  29. static inline void cmark_chunk_rtrim(cmark_chunk *c)
  30. {
  31. while (c->len > 0) {
  32. if (!isspace(c->data[c->len - 1]))
  33. break;
  34. c->len--;
  35. }
  36. }
  37. static inline void cmark_chunk_trim(cmark_chunk *c)
  38. {
  39. cmark_chunk_ltrim(c);
  40. cmark_chunk_rtrim(c);
  41. }
  42. static inline int cmark_chunk_strchr(cmark_chunk *ch, int c, int offset)
  43. {
  44. const unsigned char *p = (unsigned char *)memchr(ch->data + offset, c, ch->len - offset);
  45. return p ? (int)(p - ch->data) : ch->len;
  46. }
  47. static inline const char *cmark_chunk_to_cstr(cmark_chunk *c)
  48. {
  49. unsigned char *str;
  50. if (c->alloc) {
  51. return (char *)c->data;
  52. }
  53. str = (unsigned char *)malloc(c->len + 1);
  54. if(str != NULL) {
  55. memcpy(str, c->data, c->len);
  56. str[c->len] = 0;
  57. }
  58. c->data = str;
  59. c->alloc = 1;
  60. return (char *)str;
  61. }
  62. static inline void cmark_chunk_set_cstr(cmark_chunk *c, const char *str)
  63. {
  64. if (c->alloc) {
  65. free(c->data);
  66. }
  67. c->len = strlen(str);
  68. c->data = (unsigned char *)malloc(c->len + 1);
  69. c->alloc = 1;
  70. memcpy(c->data, str, c->len + 1);
  71. }
  72. static inline cmark_chunk cmark_chunk_literal(const char *data)
  73. {
  74. cmark_chunk c = {(unsigned char *)data, data ? strlen(data) : 0, 0};
  75. return c;
  76. }
  77. static inline cmark_chunk cmark_chunk_dup(const cmark_chunk *ch, int pos, int len)
  78. {
  79. cmark_chunk c = {ch->data + pos, len, 0};
  80. return c;
  81. }
  82. static inline cmark_chunk cmark_chunk_buf_detach(cmark_strbuf *buf)
  83. {
  84. cmark_chunk c;
  85. c.len = buf->size;
  86. c.data = cmark_strbuf_detach(buf);
  87. c.alloc = 1;
  88. return c;
  89. }
  90. // Convenience macros
  91. #define chunk cmark_chunk
  92. #define chunk_free cmark_chunk_free
  93. #define chunk_ltrim cmark_chunk_ltrim
  94. #define chunk_rtrim cmark_chunk_rtrim
  95. #define chunk_trim cmark_chunk_trim
  96. #define chunk_strchr cmark_chunk_strchr
  97. #define chunk_to_cstr cmark_chunk_to_cstr
  98. #define chunk_literal cmark_chunk_literal
  99. #define chunk_dup cmark_chunk_dup
  100. #define chunk_buf_detach cmark_chunk_buf_detach
  101. #endif