msvc.c

   1 /*
   2  * Copyright (C) 2012, 2013, 2014
   3  *     Dale Weiler
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a copy of
   6  * this software and associated documentation files (the "Software"), to deal in
   7  * the Software without restriction, including without limitation the rights to
   8  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
   9  * of the Software, and to permit persons to whom the Software is furnished to do
  10  * so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in all
  13  * copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  */
  23 #define GMQCC_PLATFORM_HEADER
  24 #include <string.h>
  25 #include <stdlib.h>
  26
  27 #include "platform.h"
  28
  29 #define CTIME_BUFFER    64
  30 #define GETENV_BUFFER   4096
  31 #define STRERROR_BUFFER 128
  32
  33 static void **platform_mem_pool = NULL;
  34 static void platform_mem_atexit() {
  35     size_t i;
  36     for (i = 0; i < vec_size(platform_mem_pool); i++)
  37         mem_d(platform_mem_pool[i]);
  38     vec_free(platform_mem_pool);
  39 }
  40
  41 static void *platform_mem_allocate(size_t bytes) {
  42     void *mem = NULL;
  43     if (!platform_mem_pool) {
  44         atexit(&platform_mem_atexit);
  45         vec_push(platform_mem_pool, NULL);
  46     }
  47
  48     mem = mem_a(bytes);
  49     vec_push(platform_mem_pool, mem);
  50
  51     return mem;
  52 }
  53
  54 int platform_vsnprintf(char *buffer, size_t bytes, const char *format, va_list arg) {
  55     return vsnprintf_s(buffer, bytes, bytes, format, arg);
  56 }
  57
  58 int platform_vsscanf(const char *str, const char *format, va_list va) {
  59     return vsscanf_s(str, format, va);
  60 }
  61
  62 const struct tm *platform_localtime(const time_t *timer) {
  63     struct tm *t;
  64     t = (struct tm*)platform_mem_allocate(sizeof(struct tm));
  65     localtime_s(&t, timer);
  66     return t;
  67 }
  68
  69 const char *platform_ctime(const time_t *timer) {
  70     char *buffer = (char *)platform_mem_allocate(CTIME_BUFFER);
  71     ctime_s(buffer, CTIME_BUFFER, timer);
  72     return buffer;
  73 }
  74
  75 char *platform_strncat(char *dest, const char *src, size_t num) {
  76     return strncat_s(dest, num, src, _TRUNCATE);
  77 }
  78
  79 const char *platform_getenv(const char *var) {
  80     char  *buffer = (char *)platform_mem_allocate(GETENV_BUFFER);
  81     size_t size;
  82     getenv_s(&size, buffer, GETENV_BUFFER, var);
  83     return buffer;
  84 }
  85
  86 /*
  87  * TODO: this isn't exactly 'accurate' for MSVC but it seems to work,
  88  * at least to some extent.
  89  */
  90 int platform_vasprintf(char **dat, const char *fmt, va_list args) {
  91     int   ret;
  92     int   len;
  93     char *tmp = NULL;
  94
  95     if ((len = _vscprintf(fmt, args)) < 0) {
  96         *dat = NULL;
  97         return -1;
  98     }
  99
 100     tmp = (char*)mem_a(len + 1);
 101     if ((ret = _vsnprintf_s(tmp, len+1, len+1, fmt, args)) != len) {
 102         mem_d(tmp);
 103         *dat = NULL;
 104         return -1;
 105     }
 106     *dat = tmp;
 107     return len;
 108 }
 109
 110 char *platform_strcat(char *dest, const char *src) {
 111     strcat_s(dest, strlen(src), src);
 112     return dest;
 113 }
 114
 115 char *platform_strncpy(char *dest, const char *src, size_t num) {
 116     strncpy_s(dest, num, src, num);
 117     return dest;
 118 }
 119
 120 const char *platform_strerror(int err) {
 121     char *buffer = (char*)platform_mem_allocate(STRERROR_BUFFER);
 122     strerror_s(buffer, STRERROR_BUFFER, err);
 123     return buffer;
 124 }
 125
 126 FILE *platform_fopen(const char *filename, const char *mode) {
 127     FILE *handle;
 128     return (fopen_s(&handle, filename, mode) != 0) ? NULL : handle;
 129 }
 130
 131 size_t platform_fread(void *ptr, size_t size, size_t count, FILE *stream) {
 132     return fread_s(ptr, size, size, count, stream);
 133 }
 134
 135 size_t platform_fwrite(const void *ptr, size_t size, size_t count, FILE *stream) {
 136     return fwrite(ptr, size, count, stream);
 137 }
 138
 139 int platform_fflush(FILE *stream) {
 140     return fflush(stream);
 141 }
 142
 143 int platform_vfprintf(FILE *stream, const char *format, va_list arg) {
 144     return vfprintf_s(stream, format, arg);
 145 }
 146
 147 int platform_fclose(FILE *stream) {
 148     return fclose(stream);
 149 }
 150
 151 int platform_ferror(FILE *stream) {
 152     return ferror(stream);
 153 }
 154
 155 int platform_fgetc(FILE *stream) {
 156     return fgetc(stream);
 157 }
 158
 159 int platform_fputs(const char *str, FILE *stream) {
 160     return fputs(str, stream);
 161 }
 162
 163 int platform_fseek(FILE *stream, long offset, int origin) {
 164     return fseek(stream, offset, origin);
 165 }
 166
 167 long platform_ftell(FILE *stream) {
 168     return ftell(stream);
 169 }
 170
 171 int platform_mkdir(const char *path, int mode) {
 172     return mkdir(path, mode);
 173 }
 174
 175 DIR *platform_opendir(const char *path) {
 176     DIR *dir = (DIR*)mem_a(sizeof(DIR) + strlen(path));
 177     if (!dir)
 178         return NULL;
 179
 180     platform_strncpy(dir->dd_name, path, strlen(path));
 181     return dir;
 182 }
 183
 184 int platform_closedir(DIR *dir) {
 185     FindClose((HANDLE)dir->dd_handle);
 186     mem_d((void*)dir);
 187     return 0;
 188 }
 189
 190 struct dirent *platform_readdir(DIR *dir) {
 191     WIN32_FIND_DATA info;
 192     struct dirent  *data;
 193     int             ret;
 194
 195     if (!dir->dd_handle) {
 196         char *dirname;
 197         if (*dir->dd_name) {
 198             size_t n = strlen(dir->dd_name);
 199             if ((dir = (char*)mem_a(n+5))) {
 200                 platform_strncpy(dirname,     dir->dd_name, n);
 201                 platform_strncpy(dirname + n, "\\*.*",      4);
 202             }
 203         } else {
 204             if (!(dirname = util_strdup("\\*.*")))
 205                 return NULL;
 206         }
 207
 208         dir->dd_handle = (long)FindFirstFile(dirname, &info);
 209         mem_d(dirname);
 210         ret = !(!dir->dd_handle);
 211     } else if (dir->dd_handle != -11) {
 212         ret = FindNextFile((HANDLE)dir->dd_handle, &info);
 213     } else {
 214         ret = 0;
 215     }
 216
 217     if (!ret)
 218         return NULL;
 219
 220     if ((data = (struct dirent*)mem_a(sizeof(struct dirent)))) {
 221         platform_strncpy(data->d_name, info.cFileName, FILENAME_MAX - 1);
 222         data->d_name[FILENAME_MAX - 1] = '\0';
 223         data->d_namelen                = strlen(data->d_name);
 224     }
 225
 226     return data;
 227 }
 228
 229 int platform_istty(int fd) {
 230     return _isatty(fd);
 231 }