msvc.c

   1 /*
   2  * Copyright (C) 2012, 2013
   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 #include <string.h>
  24 #include <stdlib.h>
  25
  26 #include "platform.h"
  27
  28 #define CTIME_BUFFER    64
  29 #define GETENV_BUFFER   4096
  30 #define STRERROR_BUFFER 128
  31
  32 static void **platform_mem_pool = NULL;
  33 static void platform_mem_atexit() {
  34     size_t i;
  35     for (i = 0; i < vec_size(platform_mem_pool); i++)
  36         mem_d(platform_mem_pool[i]);
  37     vec_free(platform_mem_pool);
  38 }
  39
  40 static void *platform_mem_allocate(size_t bytes) {
  41     void *mem = NULL;
  42     if (!platform_mem_pool) {
  43         atexit(&platform_mem_atexit);
  44         vec_push(platform_mem_pool, NULL);
  45     }
  46
  47     mem = mem_a(bytes);
  48     vec_push(platform_mem_pool, mem);
  49
  50     return mem;
  51 }
  52
  53 int platform_vsnprintf(char *buffer, size_t bytes, const char *format, va_list arg) {
  54     vsnprintf_s(buffer, bytes, bytes, format, arg);
  55 }
  56
  57 int platform_sscanf(const char *str, const char *format, ...) {
  58     va_list va;
  59     va_start(va, format);
  60     vsscanf_s(str, format, va);
  61     va_end(va);
  62 }
  63
  64 const struct tm *platform_localtime(const time_t *timer) {
  65     struct tm *t;
  66     t = (struct tm*)platform_mem_allocate(sizeof(struct tm));
  67     localtime_s(&t, timer);
  68     return &t;
  69 }
  70
  71 const char *platform_ctime(const time_t *timer) {
  72     char *buffer = (char *)platform_mem_allocate(CTIME_BUFFER);
  73     ctime_s(buffer, CTIME_BUFFER, timer);
  74     return buffer;
  75 }
  76
  77 char *platform_strncat(char *dest, const char *src, size_t num) {
  78     return strncat_s(dest, num, src, _TRUNCATE);
  79 }
  80
  81 const char *platform_tmpnam(char *str) {
  82     return tmpnam_s(str, L_tmpnam);
  83 }
  84
  85 const char *platform_getenv(char *var) {
  86     char  *buffer = (char *)platform_mem_allocate(GETENV_BUFFER);
  87     size_t size;
  88     getenv_s(&size, buffer, GETENV_BUFFER, var);
  89     return buffer;
  90 }
  91
  92 int platform_snprintf(char *src, size_t bytes, const char *format, ...) {
  93     int      rt;
  94     va_list  va;
  95     va_start(va, format);
  96
  97     rt = vsprintf_s(src, bytes, format, va);
  98     va_end  (va);
  99
 100     return rt;
 101 }
 102
 103 /*
 104  * TODO: this isn't exactly 'accurate' for MSVC but it seems to work,
 105  * at least to some extent.
 106  */
 107 int platform_vasprintf(char **dat, const char *fmt, va_list args) {
 108     int   ret;
 109     int   len;
 110     char *tmp = NULL;
 111
 112     if ((len = _vscprintf(fmt, args)) < 0) {
 113         *dat = NULL;
 114         return -1;
 115     }
 116
 117     tmp = (char*)mem_a(len + 1);
 118     if ((ret = _vsnprintf_s(tmp, len+1, len+1, fmt, args)) != len) {
 119         mem_d(tmp);
 120         *dat = NULL;
 121         return -1;
 122     }
 123     *dat = tmp;
 124     return len;
 125 }
 126
 127 char *platform_strcat(char *dest, const char *src) {
 128     strcat_s(dest, strlen(src), src);
 129     return dest;
 130 }
 131
 132 char *platform_strncpy(char *dest, const char *src, size_t num) {
 133     strncpy_s(dest, num, src, num);
 134     return dest;
 135 }
 136
 137 const char *platform_strerror(int err) {
 138     char *buffer = (char*)platform_mem_allocate(STRERROR_BUFFER);
 139     strerror_s(buffer, STRERROR_BUFFER, err);
 140     return buffer;
 141 }
 142
 143 FILE *platform_fopen(const char *filename, const char *mode) {
 144     FILE *handle;
 145     return (fopen_s(&handle, filename, mode) != 0) ? NULL : handle;
 146 }
 147
 148 size_t platform_fread(void *ptr, size_t size, size_t count, FILE *stream) {
 149     return fread_s(ptr, size, size, count, stream);
 150 }
 151
 152 size_t platform_fwrite(const void *ptr, size_t size, size_t count, FILE *stream) {
 153     return fwrite(ptr, size, count, stream);
 154 }
 155
 156 int platform_fflush(FILE *stream) {
 157     return fflush(stream);
 158 }
 159
 160 int platform_vfprintf(FILE *stream, const char *format, va_list arg) {
 161     return vfprintf_s(stream, format, arg);
 162 }
 163
 164 int platform_fclose(FILE *stream) {
 165     return fclose(stream);
 166 }
 167
 168 int platform_ferror(FILE *stream) {
 169     return ferror(stream);
 170 }
 171
 172 int platform_fgetc(FILE *stream) {
 173     return fgetc(stream);
 174 }
 175
 176 int platform_fputs(const char *str, FILE *stream) {
 177     return fputs(str, stream);
 178 }
 179
 180 int platform_fseek(FILE *stream, long offset, int origin) {
 181     return fseek(stream, offset, origin);
 182 }
 183
 184 long platform_ftell(FILE *stream) {
 185     return ftell(stream);
 186 }
 187
 188 int platform_mkdir(const char *path, int mode) {
 189     return mkdir(path, mode);
 190 }
 191
 192 DIR *platform_opendir(const char *path) {
 193     DIR *dir = (DIR*)mem_a(sizeof(DIR) + strlen(path));
 194     if (!dir)
 195         return NULL;
 196
 197     platform_strncpy(dir->dd_name, path, strlen(path));
 198     return dir;
 199 }
 200
 201 int platform_closedir(DIR *dir) {
 202     FindClose((HANDLE)dir->dd_handle);
 203     mem_d((void*)dir);
 204     return 0;
 205 }
 206
 207 struct dirent *platform_readdir(DIR *dir) {
 208     WIN32_FIND_DATA info;
 209     struct dirent  *data;
 210     int             ret;
 211
 212     if (!dir->dd_handle) {
 213         char *dirname;
 214         if (*dir->dd_name) {
 215             size_t n = strlen(dir->dd_name);
 216             if ((dir = (char*)mem_a(n+5))) {
 217                 platform_strncpy(dirname,     dir->dd_name, n);
 218                 platform_strncpy(dirname + n, "\\*.*",      4);
 219             }
 220         } else {
 221             if (!(dirname = util_strdup("\\*.*")))
 222                 return NULL;
 223         }
 224
 225         dir->dd_handle = (long)FindFirstFile(dirname, &info);
 226         mem_d(dirname);
 227         ret = !(!dir->dd_handle);
 228     } else if (dir->dd_handle != -11) {
 229         ret = FindNextFile((HANDLE)dir->dd_handle, &info);
 230     } else {
 231         ret = 0;
 232     }
 233
 234     if (!ret)
 235         return NULL;
 236
 237     if ((data = (struct dirent*)mem_a(sizeof(struct dirent)))) {
 238         platform_strncpy(data->d_name, info.cFileName, FILENAME_MAX - 1);
 239         data->d_name[FILENAME_MAX - 1] = '\0';
 240         data->d_namelen                = strlen(data->d_name);
 241     }
 242
 243     return data;
 244 }
 245
 246 int platform_istty(int fd) {
 247     return _istty(fd);
 248 }