FX_BOOL considered harmful, part 2.
[pdfium.git] / core / src / fxcrt / fx_extension.cpp
1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "../../include/fxcrt/fx_basic.h"
8 #include "../../include/fxcrt/fx_ext.h"
9 #include "extension.h"
10 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
11 #include <wincrypt.h>
12 #else
13 #include <ctime>
14 #endif
15
16 FX_HFILE FX_File_Open(const CFX_ByteStringC& fileName, FX_DWORD dwMode)
17 {
18     IFXCRT_FileAccess* pFA = FXCRT_FileAccess_Create();
19     if (pFA && !pFA->Open(fileName, dwMode)) {
20         pFA->Release();
21         return NULL;
22     }
23     return (FX_HFILE)pFA;
24 }
25 FX_HFILE FX_File_Open(const CFX_WideStringC& fileName, FX_DWORD dwMode)
26 {
27     IFXCRT_FileAccess* pFA = FXCRT_FileAccess_Create();
28     if (pFA && !pFA->Open(fileName, dwMode)) {
29         pFA->Release();
30         return NULL;
31     }
32     return (FX_HFILE)pFA;
33 }
34 void FX_File_Close(FX_HFILE hFile)
35 {
36     FXSYS_assert(hFile != NULL);
37     ((IFXCRT_FileAccess*)hFile)->Close();
38     ((IFXCRT_FileAccess*)hFile)->Release();
39 }
40 FX_FILESIZE FX_File_GetSize(FX_HFILE hFile)
41 {
42     FXSYS_assert(hFile != NULL);
43     return ((IFXCRT_FileAccess*)hFile)->GetSize();
44 }
45 FX_FILESIZE FX_File_GetPosition(FX_HFILE hFile)
46 {
47     FXSYS_assert(hFile != NULL);
48     return ((IFXCRT_FileAccess*)hFile)->GetPosition();
49 }
50 FX_FILESIZE FX_File_SetPosition(FX_HFILE hFile, FX_FILESIZE pos)
51 {
52     FXSYS_assert(hFile != NULL);
53     return ((IFXCRT_FileAccess*)hFile)->SetPosition(pos);
54 }
55 size_t FX_File_Read(FX_HFILE hFile, void* pBuffer, size_t szBuffer)
56 {
57     FXSYS_assert(hFile != NULL);
58     return ((IFXCRT_FileAccess*)hFile)->Read(pBuffer, szBuffer);
59 }
60 size_t FX_File_ReadPos(FX_HFILE hFile, void* pBuffer, size_t szBuffer, FX_FILESIZE pos)
61 {
62     FXSYS_assert(hFile != NULL);
63     return ((IFXCRT_FileAccess*)hFile)->ReadPos(pBuffer, szBuffer, pos);
64 }
65 size_t FX_File_Write(FX_HFILE hFile, const void* pBuffer, size_t szBuffer)
66 {
67     FXSYS_assert(hFile != NULL);
68     return ((IFXCRT_FileAccess*)hFile)->Write(pBuffer, szBuffer);
69 }
70 size_t FX_File_WritePos(FX_HFILE hFile, const void* pBuffer, size_t szBuffer, FX_FILESIZE pos)
71 {
72     FXSYS_assert(hFile != NULL);
73     return ((IFXCRT_FileAccess*)hFile)->WritePos(pBuffer, szBuffer, pos);
74 }
75 bool FX_File_Flush(FX_HFILE hFile)
76 {
77     FXSYS_assert(hFile != NULL);
78     return ((IFXCRT_FileAccess*)hFile)->Flush();
79 }
80 bool FX_File_Truncate(FX_HFILE hFile, FX_FILESIZE szFile)
81 {
82     FXSYS_assert(hFile != NULL);
83     return ((IFXCRT_FileAccess*)hFile)->Truncate(szFile);
84 }
85 IFX_FileStream* FX_CreateFileStream(const FX_CHAR* filename, FX_DWORD dwModes)
86 {
87     IFXCRT_FileAccess* pFA = FXCRT_FileAccess_Create();
88     if (!pFA) {
89         return NULL;
90     }
91     if (!pFA->Open(filename, dwModes)) {
92         pFA->Release();
93         return NULL;
94     }
95     return new CFX_CRTFileStream(pFA);
96 }
97 IFX_FileStream* FX_CreateFileStream(const FX_WCHAR* filename, FX_DWORD dwModes)
98 {
99     IFXCRT_FileAccess* pFA = FXCRT_FileAccess_Create();
100     if (!pFA) {
101         return NULL;
102     }
103     if (!pFA->Open(filename, dwModes)) {
104         pFA->Release();
105         return NULL;
106     }
107     return new CFX_CRTFileStream(pFA);
108 }
109 IFX_FileWrite* FX_CreateFileWrite(const FX_CHAR* filename)
110 {
111     return FX_CreateFileStream(filename, FX_FILEMODE_Truncate);
112 }
113 IFX_FileWrite* FX_CreateFileWrite(const FX_WCHAR* filename)
114 {
115     return FX_CreateFileStream(filename, FX_FILEMODE_Truncate);
116 }
117 IFX_FileRead* FX_CreateFileRead(const FX_CHAR* filename)
118 {
119     return FX_CreateFileStream(filename, FX_FILEMODE_ReadOnly);
120 }
121 IFX_FileRead* FX_CreateFileRead(const FX_WCHAR* filename)
122 {
123     return FX_CreateFileStream(filename, FX_FILEMODE_ReadOnly);
124 }
125 IFX_MemoryStream* FX_CreateMemoryStream(uint8_t* pBuffer, size_t dwSize, bool bTakeOver)
126 {
127     return new CFX_MemoryStream(pBuffer, dwSize, bTakeOver);
128 }
129 IFX_MemoryStream* FX_CreateMemoryStream(bool bConsecutive)
130 {
131     return new CFX_MemoryStream(bConsecutive);
132 }
133 #ifdef __cplusplus
134 extern "C" {
135 #endif
136 FX_FLOAT FXSYS_tan(FX_FLOAT a)
137 {
138     return (FX_FLOAT)tan(a);
139 }
140 FX_FLOAT FXSYS_logb(FX_FLOAT b, FX_FLOAT x)
141 {
142     return FXSYS_log(x) / FXSYS_log(b);
143 }
144 FX_FLOAT FXSYS_strtof(const FX_CHAR* pcsStr, int32_t iLength, int32_t *pUsedLen)
145 {
146     FXSYS_assert(pcsStr != NULL);
147     if (iLength < 0) {
148         iLength = (int32_t)FXSYS_strlen(pcsStr);
149     }
150     CFX_WideString ws = CFX_WideString::FromLocal(pcsStr, iLength);
151     return FXSYS_wcstof(ws.c_str(), iLength, pUsedLen);
152 }
153 FX_FLOAT FXSYS_wcstof(const FX_WCHAR* pwsStr, int32_t iLength, int32_t *pUsedLen)
154 {
155     FXSYS_assert(pwsStr != NULL);
156     if (iLength < 0) {
157         iLength = (int32_t)FXSYS_wcslen(pwsStr);
158     }
159     if (iLength == 0) {
160         return 0.0f;
161     }
162     int32_t iUsedLen = 0;
163     bool bNegtive = false;
164     switch (pwsStr[iUsedLen]) {
165         case '-':
166             bNegtive = true;
167         case '+':
168             iUsedLen++;
169             break;
170     }
171     FX_FLOAT fValue = 0.0f;
172     while (iUsedLen < iLength) {
173         FX_WCHAR wch = pwsStr[iUsedLen];
174         if (wch >= L'0' && wch <= L'9') {
175             fValue = fValue * 10.0f + (wch - L'0');
176         } else {
177             break;
178         }
179         iUsedLen++;
180     }
181     if (iUsedLen < iLength && pwsStr[iUsedLen] == L'.') {
182         FX_FLOAT fPrecise = 0.1f;
183         while (++iUsedLen < iLength) {
184             FX_WCHAR wch = pwsStr[iUsedLen];
185             if (wch >= L'0' && wch <= L'9') {
186                 fValue += (wch - L'0') * fPrecise;
187                 fPrecise *= 0.1f;
188             } else {
189                 break;
190             }
191         }
192     }
193     if (pUsedLen) {
194         *pUsedLen = iUsedLen;
195     }
196     return bNegtive ? -fValue : fValue;
197 }
198 FX_WCHAR* FXSYS_wcsncpy(FX_WCHAR* dstStr, const FX_WCHAR* srcStr, size_t count)
199 {
200     FXSYS_assert(dstStr != NULL && srcStr != NULL && count > 0);
201     for (size_t i = 0; i < count; ++i)
202         if ((dstStr[i] = srcStr[i]) == L'\0') {
203             break;
204         }
205     return dstStr;
206 }
207 int32_t FXSYS_wcsnicmp(const FX_WCHAR* s1, const FX_WCHAR* s2, size_t count)
208 {
209     FXSYS_assert(s1 != NULL && s2 != NULL && count > 0);
210     FX_WCHAR wch1 = 0, wch2 = 0;
211     while (count-- > 0) {
212         wch1 = (FX_WCHAR)FXSYS_tolower(*s1++);
213         wch2 = (FX_WCHAR)FXSYS_tolower(*s2++);
214         if (wch1 != wch2) {
215             break;
216         }
217     }
218     return wch1 - wch2;
219 }
220 int32_t FXSYS_strnicmp(const FX_CHAR* s1, const FX_CHAR* s2, size_t count)
221 {
222     FXSYS_assert(s1 != NULL && s2 != NULL && count > 0);
223     FX_CHAR ch1 = 0, ch2 = 0;
224     while (count-- > 0) {
225         ch1 = (FX_CHAR)FXSYS_tolower(*s1++);
226         ch2 = (FX_CHAR)FXSYS_tolower(*s2++);
227         if (ch1 != ch2) {
228             break;
229         }
230     }
231     return ch1 - ch2;
232 }
233 FX_DWORD FX_HashCode_String_GetA(const FX_CHAR* pStr, int32_t iLength, bool bIgnoreCase)
234 {
235     FXSYS_assert(pStr != NULL);
236     if (iLength < 0) {
237         iLength = (int32_t)FXSYS_strlen(pStr);
238     }
239     const FX_CHAR* pStrEnd = pStr + iLength;
240     FX_DWORD dwHashCode = 0;
241     if (bIgnoreCase) {
242         while (pStr < pStrEnd) {
243             dwHashCode = 31 * dwHashCode + FXSYS_tolower(*pStr++);
244         }
245     } else {
246         while (pStr < pStrEnd) {
247             dwHashCode = 31 * dwHashCode + *pStr ++;
248         }
249     }
250     return dwHashCode;
251 }
252 FX_DWORD FX_HashCode_String_GetW(const FX_WCHAR* pStr, int32_t iLength, bool bIgnoreCase)
253 {
254     FXSYS_assert(pStr != NULL);
255     if (iLength < 0) {
256         iLength = (int32_t)FXSYS_wcslen(pStr);
257     }
258     const FX_WCHAR* pStrEnd = pStr + iLength;
259     FX_DWORD dwHashCode = 0;
260     if (bIgnoreCase) {
261         while (pStr < pStrEnd) {
262             dwHashCode = 1313 * dwHashCode + FXSYS_tolower(*pStr++);
263         }
264     } else {
265         while (pStr < pStrEnd) {
266             dwHashCode = 1313 * dwHashCode + *pStr ++;
267         }
268     }
269     return dwHashCode;
270 }
271 #ifdef __cplusplus
272 }
273 #endif
274 #ifdef __cplusplus
275 extern "C" {
276 #endif
277 void* FX_Random_MT_Start(FX_DWORD dwSeed)
278 {
279     FX_LPMTRANDOMCONTEXT pContext = FX_Alloc(FX_MTRANDOMCONTEXT, 1);
280     pContext->mt[0] = dwSeed;
281     FX_DWORD &i = pContext->mti;
282     FX_DWORD* pBuf = pContext->mt;
283     for (i = 1; i < MT_N; i ++) {
284         pBuf[i] = (1812433253UL * (pBuf[i - 1] ^ (pBuf[i - 1] >> 30)) + i);
285     }
286     pContext->bHaveSeed = true;
287     return pContext;
288 }
289 FX_DWORD FX_Random_MT_Generate(void* pContext)
290 {
291     FXSYS_assert(pContext != NULL);
292     FX_LPMTRANDOMCONTEXT pMTC = (FX_LPMTRANDOMCONTEXT)pContext;
293     FX_DWORD v;
294     static FX_DWORD mag[2] = {0, MT_Matrix_A};
295     FX_DWORD &mti = pMTC->mti;
296     FX_DWORD* pBuf = pMTC->mt;
297     if ((int)mti < 0 || mti >= MT_N) {
298         if (mti > MT_N && !pMTC->bHaveSeed) {
299             return 0;
300         }
301         FX_DWORD kk;
302         for (kk = 0; kk < MT_N - MT_M; kk ++) {
303             v = (pBuf[kk] & MT_Upper_Mask) | (pBuf[kk + 1] & MT_Lower_Mask);
304             pBuf[kk] = pBuf[kk + MT_M] ^ (v >> 1) ^ mag[v & 1];
305         }
306         for (; kk < MT_N - 1; kk ++) {
307             v = (pBuf[kk] & MT_Upper_Mask) | (pBuf[kk + 1] & MT_Lower_Mask);
308             pBuf[kk] = pBuf[kk + (MT_M - MT_N)] ^ (v >> 1) ^ mag[v & 1];
309         }
310         v = (pBuf[MT_N - 1] & MT_Upper_Mask) | (pBuf[0] & MT_Lower_Mask);
311         pBuf[MT_N - 1] = pBuf[MT_M - 1] ^ (v >> 1) ^ mag[v & 1];
312         mti = 0;
313     }
314     v = pBuf[mti ++];
315     v ^= (v >> 11);
316     v ^= (v << 7) & 0x9d2c5680UL;
317     v ^= (v << 15) & 0xefc60000UL;
318     v ^= (v >> 18);
319     return v;
320 }
321 void FX_Random_MT_Close(void* pContext)
322 {
323     FXSYS_assert(pContext != NULL);
324     FX_Free(pContext);
325 }
326 void FX_Random_GenerateMT(FX_DWORD* pBuffer, int32_t iCount)
327 {
328     FX_DWORD dwSeed;
329 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
330     if (!FX_GenerateCryptoRandom(&dwSeed, 1)) {
331         FX_Random_GenerateBase(&dwSeed, 1);
332     }
333 #else
334     FX_Random_GenerateBase(&dwSeed, 1);
335 #endif
336     void* pContext = FX_Random_MT_Start(dwSeed);
337     while (iCount -- > 0) {
338         *pBuffer ++ = FX_Random_MT_Generate(pContext);
339     }
340     FX_Random_MT_Close(pContext);
341 }
342 void FX_Random_GenerateBase(FX_DWORD* pBuffer, int32_t iCount)
343 {
344 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
345     SYSTEMTIME st1, st2;
346     ::GetSystemTime(&st1);
347     do {
348         ::GetSystemTime(&st2);
349     } while (FXSYS_memcmp(&st1, &st2, sizeof(SYSTEMTIME)) == 0);
350     FX_DWORD dwHash1 = FX_HashCode_String_GetA((const FX_CHAR*)&st1, sizeof(st1), true);
351     FX_DWORD dwHash2 = FX_HashCode_String_GetA((const FX_CHAR*)&st2, sizeof(st2), true);
352     ::srand((dwHash1 << 16) | (FX_DWORD)dwHash2);
353 #else
354     time_t tmLast = time(NULL), tmCur;
355     while ((tmCur = time(NULL)) == tmLast);
356     ::srand((tmCur << 16) | (tmLast & 0xFFFF));
357 #endif
358     while (iCount -- > 0) {
359         *pBuffer ++ = (FX_DWORD)((::rand() << 16) | (::rand() & 0xFFFF));
360     }
361 }
362 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
363 bool FX_GenerateCryptoRandom(FX_DWORD* pBuffer, int32_t iCount)
364 {
365     HCRYPTPROV hCP = NULL;
366     if (!::CryptAcquireContext(&hCP, NULL, NULL, PROV_RSA_FULL, 0) || hCP == NULL) {
367         return false;
368     }
369     ::CryptGenRandom(hCP, iCount * sizeof(FX_DWORD), (uint8_t*)pBuffer);
370     ::CryptReleaseContext(hCP, 0);
371     return true;
372 }
373 #endif
374 void FX_Random_GenerateCrypto(FX_DWORD* pBuffer, int32_t iCount)
375 {
376 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
377     FX_GenerateCryptoRandom(pBuffer, iCount);
378 #else
379     FX_Random_GenerateBase(pBuffer, iCount);
380 #endif
381 }
382 #ifdef __cplusplus
383 }
384 #endif