src/buf.c

   1 /*
   2  * Copyright (C) the libgit2 contributors. All rights reserved.
   3  *
   4  * This file is part of libgit2, distributed under the GNU GPL v2 with
   5  * a Linking Exception. For full terms see the included COPYING file.
   6  */
   7
   8 #include "buf.h"
   9 #include "common.h"
  10
  11 int git_buf_sanitize(git_buf *buf)
  12 {
  13         GIT_ASSERT_ARG(buf);
  14
  15         if (buf->reserved > 0)
  16                 buf->ptr[0] = '\0';
  17         else
  18                 buf->ptr = git_str__initstr;
  19
  20         buf->size = 0;
  21         return 0;
  22 }
  23
  24 int git_buf_tostr(git_str *out, git_buf *buf)
  25 {
  26         GIT_ASSERT_ARG(out);
  27         GIT_ASSERT_ARG(buf);
  28
  29         if (git_buf_sanitize(buf) < 0)
  30                 return -1;
  31
  32         out->ptr = buf->ptr;
  33         out->asize = buf->reserved;
  34         out->size = buf->size;
  35
  36         buf->ptr = git_str__initstr;
  37         buf->reserved = 0;
  38         buf->size = 0;
  39
  40         return 0;
  41 }
  42
  43 int git_buf_fromstr(git_buf *out, git_str *str)
  44 {
  45         GIT_ASSERT_ARG(out);
  46         GIT_ASSERT_ARG(str);
  47
  48         out->ptr = str->ptr;
  49         out->reserved = str->asize;
  50         out->size = str->size;
  51
  52         str->ptr = git_str__initstr;
  53         str->asize = 0;
  54         str->size = 0;
  55
  56         return 0;
  57 }
  58
  59 void git_buf_dispose(git_buf *buf)
  60 {
  61         if (!buf)
  62                 return;
  63
  64         if (buf->ptr != git_str__initstr)
  65                 git__free(buf->ptr);
  66
  67         buf->ptr = git_str__initstr;
  68         buf->reserved = 0;
  69         buf->size = 0;
  70 }
  71
  72 #ifndef GIT_DEPRECATE_HARD
  73 int git_buf_grow(git_buf *buffer, size_t target_size)
  74 {
  75         char *newptr;
  76
  77         if (buffer->reserved >= target_size)
  78                 return 0;
  79
  80         if (buffer->ptr == git_str__initstr)
  81                 newptr = git__malloc(target_size);
  82         else
  83                 newptr = git__realloc(buffer->ptr, target_size);
  84
  85         if (!newptr)
  86                 return -1;
  87
  88         buffer->ptr = newptr;
  89         buffer->reserved = target_size;
  90         return 0;
  91 }
  92
  93 int git_buf_set(git_buf *buffer, const void *data, size_t datalen)
  94 {
  95         size_t alloclen;
  96
  97         GIT_ERROR_CHECK_ALLOC_ADD(&alloclen, datalen, 1);
  98
  99         if (git_buf_grow(buffer, alloclen) < 0)
 100                 return -1;
 101
 102         memmove(buffer->ptr, data, datalen);
 103         buffer->size = datalen;
 104         buffer->ptr[buffer->size] = '\0';
 105
 106         return 0;
 107 }
 108
 109 int git_buf_is_binary(const git_buf *buf)
 110 {
 111         git_str str = GIT_STR_INIT_CONST(buf->ptr, buf->size);
 112         return git_str_is_binary(&str);
 113 }
 114
 115 int git_buf_contains_nul(const git_buf *buf)
 116 {
 117         git_str str = GIT_STR_INIT_CONST(buf->ptr, buf->size);
 118         return git_str_contains_nul(&str);
 119 }
 120
 121 void git_buf_free(git_buf *buffer)
 122 {
 123         git_buf_dispose(buffer);
 124 }
 125
 126 #endif