strbuffer.c 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. /*
  2. * Copyright (c) 2009, 2010 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 "strbuffer.h"
  11. #include "util.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 = 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. 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. strbuff->size = max(strbuff->size * STRBUFFER_FACTOR,
  60. strbuff->length + size + 1);
  61. strbuff->value = realloc(strbuff->value, strbuff->size);
  62. if(!strbuff->value)
  63. return -1;
  64. }
  65. memcpy(strbuff->value + strbuff->length, data, size);
  66. strbuff->length += size;
  67. strbuff->value[strbuff->length] = '\0';
  68. return 0;
  69. }
  70. char strbuffer_pop(strbuffer_t *strbuff)
  71. {
  72. if(strbuff->length > 0) {
  73. char c = strbuff->value[--strbuff->length];
  74. strbuff->value[strbuff->length] = '\0';
  75. return c;
  76. }
  77. else
  78. return '\0';
  79. }