I need to test this code on msvc now.
[xonotic/gmqcc.git] / 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 #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     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_tmpnam(char *str) {
80     return tmpnam_s(str, L_tmpnam);
81 }
82
83 const char *platform_getenv(char *var) {
84     char  *buffer = (char *)platform_mem_allocate(GETENV_BUFFER);
85     size_t size;
86     getenv_s(&size, buffer, GETENV_BUFFER, var);
87     return buffer;
88 }
89
90 /*
91  * TODO: this isn't exactly 'accurate' for MSVC but it seems to work,
92  * at least to some extent.
93  */
94 int platform_vasprintf(char **dat, const char *fmt, va_list args) {
95     int   ret;
96     int   len;
97     char *tmp = NULL;
98
99     if ((len = _vscprintf(fmt, args)) < 0) {
100         *dat = NULL;
101         return -1;
102     }
103
104     tmp = (char*)mem_a(len + 1);
105     if ((ret = _vsnprintf_s(tmp, len+1, len+1, fmt, args)) != len) {
106         mem_d(tmp);
107         *dat = NULL;
108         return -1;
109     }
110     *dat = tmp;
111     return len;
112 }
113
114 char *platform_strcat(char *dest, const char *src) {
115     strcat_s(dest, strlen(src), src);
116     return dest;
117 }
118
119 char *platform_strncpy(char *dest, const char *src, size_t num) {
120     strncpy_s(dest, num, src, num);
121     return dest;
122 }
123
124 const char *platform_strerror(int err) {
125     char *buffer = (char*)platform_mem_allocate(STRERROR_BUFFER);
126     strerror_s(buffer, STRERROR_BUFFER, err);
127     return buffer;
128 }
129
130 FILE *platform_fopen(const char *filename, const char *mode) {
131     FILE *handle;
132     return (fopen_s(&handle, filename, mode) != 0) ? NULL : handle;
133 }
134
135 size_t platform_fread(void *ptr, size_t size, size_t count, FILE *stream) {
136     return fread_s(ptr, size, size, count, stream);
137 }
138
139 size_t platform_fwrite(const void *ptr, size_t size, size_t count, FILE *stream) {
140     return fwrite(ptr, size, count, stream);
141 }
142
143 int platform_fflush(FILE *stream) {
144     return fflush(stream);
145 }
146
147 int platform_vfprintf(FILE *stream, const char *format, va_list arg) {
148     return vfprintf_s(stream, format, arg);
149 }
150
151 int platform_fclose(FILE *stream) {
152     return fclose(stream);
153 }
154
155 int platform_ferror(FILE *stream) {
156     return ferror(stream);
157 }
158
159 int platform_fgetc(FILE *stream) {
160     return fgetc(stream);
161 }
162
163 int platform_fputs(const char *str, FILE *stream) {
164     return fputs(str, stream);
165 }
166
167 int platform_fseek(FILE *stream, long offset, int origin) {
168     return fseek(stream, offset, origin);
169 }
170
171 long platform_ftell(FILE *stream) {
172     return ftell(stream);
173 }
174
175 int platform_mkdir(const char *path, int mode) {
176     return mkdir(path, mode);
177 }
178
179 DIR *platform_opendir(const char *path) {
180     DIR *dir = (DIR*)mem_a(sizeof(DIR) + strlen(path));
181     if (!dir)
182         return NULL;
183
184     platform_strncpy(dir->dd_name, path, strlen(path));
185     return dir;
186 }
187
188 int platform_closedir(DIR *dir) {
189     FindClose((HANDLE)dir->dd_handle);
190     mem_d((void*)dir);
191     return 0;
192 }
193
194 struct dirent *platform_readdir(DIR *dir) {
195     WIN32_FIND_DATA info;
196     struct dirent  *data;
197     int             ret;
198
199     if (!dir->dd_handle) {
200         char *dirname;
201         if (*dir->dd_name) {
202             size_t n = strlen(dir->dd_name);
203             if ((dir = (char*)mem_a(n+5))) {
204                 platform_strncpy(dirname,     dir->dd_name, n);
205                 platform_strncpy(dirname + n, "\\*.*",      4);
206             }
207         } else {
208             if (!(dirname = util_strdup("\\*.*")))
209                 return NULL;
210         }
211
212         dir->dd_handle = (long)FindFirstFile(dirname, &info);
213         mem_d(dirname);
214         ret = !(!dir->dd_handle);
215     } else if (dir->dd_handle != -11) {
216         ret = FindNextFile((HANDLE)dir->dd_handle, &info);
217     } else {
218         ret = 0;
219     }
220
221     if (!ret)
222         return NULL;
223
224     if ((data = (struct dirent*)mem_a(sizeof(struct dirent)))) {
225         platform_strncpy(data->d_name, info.cFileName, FILENAME_MAX - 1);
226         data->d_name[FILENAME_MAX - 1] = '\0';
227         data->d_namelen                = strlen(data->d_name);
228     }
229
230     return data;
231 }
232
233 int platform_istty(int fd) {
234     return _isatty(fd);
235 }