strbuffer.c 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. /*
  2. * Copyright (c) 2009-2011 Petri Lehtinen <petri@digip.org>
  3. *
  4. * Jansson is free software; you can redistribute it and/or modify
  5. * it under the terms of the MIT license. See LICENSE for details.
  6. */
  7. #define _GNU_SOURCE
  8. #include <stdlib.h>
  9. #include <string.h>
  10. #include "jansson_private.h"
  11. #include "strbuffer.h"
  12. #define STRBUFFER_MIN_SIZE 16
  13. #define STRBUFFER_FACTOR 2
  14. int strbuffer_init(strbuffer_t *strbuff)
  15. {
  16. strbuff->size = STRBUFFER_MIN_SIZE;
  17. strbuff->length = 0;
  18. strbuff->value = jsonp_malloc(strbuff->size);
  19. if(!strbuff->value)
  20. return -1;
  21. /* initialize to empty */
  22. strbuff->value[0] = '\0';
  23. return 0;
  24. }
  25. void strbuffer_close(strbuffer_t *strbuff)
  26. {
  27. jsonp_free(strbuff->value);
  28. strbuff->size = 0;
  29. strbuff->length = 0;
  30. strbuff->value = NULL;
  31. }
  32. void strbuffer_clear(strbuffer_t *strbuff)
  33. {
  34. strbuff->length = 0;
  35. strbuff->value[0] = '\0';
  36. }
  37. const char *strbuffer_value(const strbuffer_t *strbuff)
  38. {
  39. return strbuff->value;
  40. }
  41. char *strbuffer_steal_value(strbuffer_t *strbuff)
  42. {
  43. char *result = strbuff->value;
  44. strbuffer_init(strbuff);
  45. return result;
  46. }
  47. int strbuffer_append(strbuffer_t *strbuff, const char *string)
  48. {
  49. return strbuffer_append_bytes(strbuff, string, strlen(string));
  50. }
  51. int strbuffer_append_byte(strbuffer_t *strbuff, char byte)
  52. {
  53. return strbuffer_append_bytes(strbuff, &byte, 1);
  54. }
  55. int strbuffer_append_bytes(strbuffer_t *strbuff, const char *data, int size)
  56. {
  57. if(strbuff->length + size >= strbuff->size)
  58. {
  59. size_t new_size;
  60. char *new_value;
  61. new_size = max(strbuff->size * STRBUFFER_FACTOR,
  62. strbuff->length + size + 1);
  63. new_value = jsonp_malloc(new_size);
  64. if(!new_value)
  65. return -1;
  66. memcpy(new_value, strbuff->value, strbuff->length);
  67. jsonp_free(strbuff->value);
  68. strbuff->value = new_value;
  69. strbuff->size = new_size;
  70. }
  71. memcpy(strbuff->value + strbuff->length, data, size);
  72. strbuff->length += size;
  73. strbuff->value[strbuff->length] = '\0';
  74. return 0;
  75. }
  76. char strbuffer_pop(strbuffer_t *strbuff)
  77. {
  78. if(strbuff->length > 0) {
  79. char c = strbuff->value[--strbuff->length];
  80. strbuff->value[strbuff->length] = '\0';
  81. return c;
  82. }
  83. else
  84. return '\0';
  85. }