src/win32/w32_buffer.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 "w32_buffer.h"
   9
  10 #include "utf-conv.h"
  11
  12 GIT_INLINE(int) handle_wc_error(void)
  13 {
  14         if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
  15                 errno = ENAMETOOLONG;
  16         else
  17                 errno = EINVAL;
  18
  19         return -1;
  20 }
  21
  22 int git_str_put_w(git_str *buf, const wchar_t *string_w, size_t len_w)
  23 {
  24         int utf8_len, utf8_write_len;
  25         size_t new_size;
  26
  27         if (!len_w) {
  28                 return 0;
  29         } else if (len_w > INT_MAX) {
  30                 git_error_set_oom();
  31                 return -1;
  32         }
  33
  34         GIT_ASSERT(string_w);
  35
  36         /* Measure the string necessary for conversion */
  37         if ((utf8_len = WideCharToMultiByte(CP_UTF8, WC_ERR_INVALID_CHARS, string_w, (int)len_w, NULL, 0, NULL, NULL)) == 0)
  38                 return 0;
  39
  40         GIT_ASSERT(utf8_len > 0);
  41
  42         GIT_ERROR_CHECK_ALLOC_ADD(&new_size, buf->size, (size_t)utf8_len);
  43         GIT_ERROR_CHECK_ALLOC_ADD(&new_size, new_size, 1);
  44
  45         if (git_str_grow(buf, new_size) < 0)
  46                 return -1;
  47
  48         if ((utf8_write_len = WideCharToMultiByte(
  49                         CP_UTF8, WC_ERR_INVALID_CHARS, string_w, (int)len_w, &buf->ptr[buf->size], utf8_len, NULL, NULL)) == 0)
  50                 return handle_wc_error();
  51
  52         GIT_ASSERT(utf8_write_len == utf8_len);
  53
  54         buf->size += utf8_write_len;
  55         buf->ptr[buf->size] = '\0';
  56         return 0;
  57 }