724dcf49036af8f724aa8e39f20e704a6cad1fc9
[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 IFX_FileStream* FX_CreateFileStream(const FX_CHAR* filename, FX_DWORD dwModes)
17 {
18     IFXCRT_FileAccess* pFA = FXCRT_FileAccess_Create();
19     if (!pFA) {
20         return NULL;
21     }
22     if (!pFA->Open(filename, dwModes)) {
23         pFA->Release();
24         return NULL;
25     }
26     return new CFX_CRTFileStream(pFA);
27 }
28 IFX_FileStream* FX_CreateFileStream(const FX_WCHAR* filename, FX_DWORD dwModes)
29 {
30     IFXCRT_FileAccess* pFA = FXCRT_FileAccess_Create();
31     if (!pFA) {
32         return NULL;
33     }
34     if (!pFA->Open(filename, dwModes)) {
35         pFA->Release();
36         return NULL;
37     }
38     return new CFX_CRTFileStream(pFA);
39 }
40 IFX_FileWrite* FX_CreateFileWrite(const FX_CHAR* filename)
41 {
42     return FX_CreateFileStream(filename, FX_FILEMODE_Truncate);
43 }
44 IFX_FileWrite* FX_CreateFileWrite(const FX_WCHAR* filename)
45 {
46     return FX_CreateFileStream(filename, FX_FILEMODE_Truncate);
47 }
48 IFX_FileRead* FX_CreateFileRead(const FX_CHAR* filename)
49 {
50     return FX_CreateFileStream(filename, FX_FILEMODE_ReadOnly);
51 }
52 IFX_FileRead* FX_CreateFileRead(const FX_WCHAR* filename)
53 {
54     return FX_CreateFileStream(filename, FX_FILEMODE_ReadOnly);
55 }
56 IFX_MemoryStream* FX_CreateMemoryStream(uint8_t* pBuffer, size_t dwSize, FX_BOOL bTakeOver)
57 {
58     return new CFX_MemoryStream(pBuffer, dwSize, bTakeOver);
59 }
60 IFX_MemoryStream* FX_CreateMemoryStream(FX_BOOL bConsecutive)
61 {
62     return new CFX_MemoryStream(bConsecutive);
63 }
64 #ifdef __cplusplus
65 extern "C" {
66 #endif
67 FX_FLOAT FXSYS_tan(FX_FLOAT a)
68 {
69     return (FX_FLOAT)tan(a);
70 }
71 FX_FLOAT FXSYS_logb(FX_FLOAT b, FX_FLOAT x)
72 {
73     return FXSYS_log(x) / FXSYS_log(b);
74 }
75 FX_FLOAT FXSYS_strtof(const FX_CHAR* pcsStr, int32_t iLength, int32_t *pUsedLen)
76 {
77     FXSYS_assert(pcsStr != NULL);
78     if (iLength < 0) {
79         iLength = (int32_t)FXSYS_strlen(pcsStr);
80     }
81     CFX_WideString ws = CFX_WideString::FromLocal(pcsStr, iLength);
82     return FXSYS_wcstof(ws.c_str(), iLength, pUsedLen);
83 }
84 FX_FLOAT FXSYS_wcstof(const FX_WCHAR* pwsStr, int32_t iLength, int32_t *pUsedLen)
85 {
86     FXSYS_assert(pwsStr != NULL);
87     if (iLength < 0) {
88         iLength = (int32_t)FXSYS_wcslen(pwsStr);
89     }
90     if (iLength == 0) {
91         return 0.0f;
92     }
93     int32_t iUsedLen = 0;
94     FX_BOOL bNegtive = FALSE;
95     switch (pwsStr[iUsedLen]) {
96         case '-':
97             bNegtive = TRUE;
98         case '+':
99             iUsedLen++;
100             break;
101     }
102     FX_FLOAT fValue = 0.0f;
103     while (iUsedLen < iLength) {
104         FX_WCHAR wch = pwsStr[iUsedLen];
105         if (wch >= L'0' && wch <= L'9') {
106             fValue = fValue * 10.0f + (wch - L'0');
107         } else {
108             break;
109         }
110         iUsedLen++;
111     }
112     if (iUsedLen < iLength && pwsStr[iUsedLen] == L'.') {
113         FX_FLOAT fPrecise = 0.1f;
114         while (++iUsedLen < iLength) {
115             FX_WCHAR wch = pwsStr[iUsedLen];
116             if (wch >= L'0' && wch <= L'9') {
117                 fValue += (wch - L'0') * fPrecise;
118                 fPrecise *= 0.1f;
119             } else {
120                 break;
121             }
122         }
123     }
124     if (pUsedLen) {
125         *pUsedLen = iUsedLen;
126     }
127     return bNegtive ? -fValue : fValue;
128 }
129 FX_WCHAR* FXSYS_wcsncpy(FX_WCHAR* dstStr, const FX_WCHAR* srcStr, size_t count)
130 {
131     FXSYS_assert(dstStr != NULL && srcStr != NULL && count > 0);
132     for (size_t i = 0; i < count; ++i)
133         if ((dstStr[i] = srcStr[i]) == L'\0') {
134             break;
135         }
136     return dstStr;
137 }
138 int32_t FXSYS_wcsnicmp(const FX_WCHAR* s1, const FX_WCHAR* s2, size_t count)
139 {
140     FXSYS_assert(s1 != NULL && s2 != NULL && count > 0);
141     FX_WCHAR wch1 = 0, wch2 = 0;
142     while (count-- > 0) {
143         wch1 = (FX_WCHAR)FXSYS_tolower(*s1++);
144         wch2 = (FX_WCHAR)FXSYS_tolower(*s2++);
145         if (wch1 != wch2) {
146             break;
147         }
148     }
149     return wch1 - wch2;
150 }
151 int32_t FXSYS_strnicmp(const FX_CHAR* s1, const FX_CHAR* s2, size_t count)
152 {
153     FXSYS_assert(s1 != NULL && s2 != NULL && count > 0);
154     FX_CHAR ch1 = 0, ch2 = 0;
155     while (count-- > 0) {
156         ch1 = (FX_CHAR)FXSYS_tolower(*s1++);
157         ch2 = (FX_CHAR)FXSYS_tolower(*s2++);
158         if (ch1 != ch2) {
159             break;
160         }
161     }
162     return ch1 - ch2;
163 }
164 FX_DWORD FX_HashCode_String_GetA(const FX_CHAR* pStr, int32_t iLength, FX_BOOL bIgnoreCase)
165 {
166     FXSYS_assert(pStr != NULL);
167     if (iLength < 0) {
168         iLength = (int32_t)FXSYS_strlen(pStr);
169     }
170     const FX_CHAR* pStrEnd = pStr + iLength;
171     FX_DWORD dwHashCode = 0;
172     if (bIgnoreCase) {
173         while (pStr < pStrEnd) {
174             dwHashCode = 31 * dwHashCode + FXSYS_tolower(*pStr++);
175         }
176     } else {
177         while (pStr < pStrEnd) {
178             dwHashCode = 31 * dwHashCode + *pStr ++;
179         }
180     }
181     return dwHashCode;
182 }
183 FX_DWORD FX_HashCode_String_GetW(const FX_WCHAR* pStr, int32_t iLength, FX_BOOL bIgnoreCase)
184 {
185     FXSYS_assert(pStr != NULL);
186     if (iLength < 0) {
187         iLength = (int32_t)FXSYS_wcslen(pStr);
188     }
189     const FX_WCHAR* pStrEnd = pStr + iLength;
190     FX_DWORD dwHashCode = 0;
191     if (bIgnoreCase) {
192         while (pStr < pStrEnd) {
193             dwHashCode = 1313 * dwHashCode + FXSYS_tolower(*pStr++);
194         }
195     } else {
196         while (pStr < pStrEnd) {
197             dwHashCode = 1313 * dwHashCode + *pStr ++;
198         }
199     }
200     return dwHashCode;
201 }
202 #ifdef __cplusplus
203 }
204 #endif
205 #ifdef __cplusplus
206 extern "C" {
207 #endif
208 void* FX_Random_MT_Start(FX_DWORD dwSeed)
209 {
210     FX_LPMTRANDOMCONTEXT pContext = FX_Alloc(FX_MTRANDOMCONTEXT, 1);
211     pContext->mt[0] = dwSeed;
212     FX_DWORD &i = pContext->mti;
213     FX_DWORD* pBuf = pContext->mt;
214     for (i = 1; i < MT_N; i ++) {
215         pBuf[i] = (1812433253UL * (pBuf[i - 1] ^ (pBuf[i - 1] >> 30)) + i);
216     }
217     pContext->bHaveSeed = TRUE;
218     return pContext;
219 }
220 FX_DWORD FX_Random_MT_Generate(void* pContext)
221 {
222     FXSYS_assert(pContext != NULL);
223     FX_LPMTRANDOMCONTEXT pMTC = (FX_LPMTRANDOMCONTEXT)pContext;
224     FX_DWORD v;
225     static FX_DWORD mag[2] = {0, MT_Matrix_A};
226     FX_DWORD &mti = pMTC->mti;
227     FX_DWORD* pBuf = pMTC->mt;
228     if ((int)mti < 0 || mti >= MT_N) {
229         if (mti > MT_N && !pMTC->bHaveSeed) {
230             return 0;
231         }
232         FX_DWORD kk;
233         for (kk = 0; kk < MT_N - MT_M; kk ++) {
234             v = (pBuf[kk] & MT_Upper_Mask) | (pBuf[kk + 1] & MT_Lower_Mask);
235             pBuf[kk] = pBuf[kk + MT_M] ^ (v >> 1) ^ mag[v & 1];
236         }
237         for (; kk < MT_N - 1; kk ++) {
238             v = (pBuf[kk] & MT_Upper_Mask) | (pBuf[kk + 1] & MT_Lower_Mask);
239             pBuf[kk] = pBuf[kk + (MT_M - MT_N)] ^ (v >> 1) ^ mag[v & 1];
240         }
241         v = (pBuf[MT_N - 1] & MT_Upper_Mask) | (pBuf[0] & MT_Lower_Mask);
242         pBuf[MT_N - 1] = pBuf[MT_M - 1] ^ (v >> 1) ^ mag[v & 1];
243         mti = 0;
244     }
245     v = pBuf[mti ++];
246     v ^= (v >> 11);
247     v ^= (v << 7) & 0x9d2c5680UL;
248     v ^= (v << 15) & 0xefc60000UL;
249     v ^= (v >> 18);
250     return v;
251 }
252 void FX_Random_MT_Close(void* pContext)
253 {
254     FXSYS_assert(pContext != NULL);
255     FX_Free(pContext);
256 }
257 void FX_Random_GenerateMT(FX_DWORD* pBuffer, int32_t iCount)
258 {
259     FX_DWORD dwSeed;
260 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
261     if (!FX_GenerateCryptoRandom(&dwSeed, 1)) {
262         FX_Random_GenerateBase(&dwSeed, 1);
263     }
264 #else
265     FX_Random_GenerateBase(&dwSeed, 1);
266 #endif
267     void* pContext = FX_Random_MT_Start(dwSeed);
268     while (iCount -- > 0) {
269         *pBuffer ++ = FX_Random_MT_Generate(pContext);
270     }
271     FX_Random_MT_Close(pContext);
272 }
273 void FX_Random_GenerateBase(FX_DWORD* pBuffer, int32_t iCount)
274 {
275 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
276     SYSTEMTIME st1, st2;
277     ::GetSystemTime(&st1);
278     do {
279         ::GetSystemTime(&st2);
280     } while (FXSYS_memcmp(&st1, &st2, sizeof(SYSTEMTIME)) == 0);
281     FX_DWORD dwHash1 = FX_HashCode_String_GetA((const FX_CHAR*)&st1, sizeof(st1), TRUE);
282     FX_DWORD dwHash2 = FX_HashCode_String_GetA((const FX_CHAR*)&st2, sizeof(st2), TRUE);
283     ::srand((dwHash1 << 16) | (FX_DWORD)dwHash2);
284 #else
285     time_t tmLast = time(NULL), tmCur;
286     while ((tmCur = time(NULL)) == tmLast);
287     ::srand((tmCur << 16) | (tmLast & 0xFFFF));
288 #endif
289     while (iCount -- > 0) {
290         *pBuffer ++ = (FX_DWORD)((::rand() << 16) | (::rand() & 0xFFFF));
291     }
292 }
293 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
294 FX_BOOL FX_GenerateCryptoRandom(FX_DWORD* pBuffer, int32_t iCount)
295 {
296     HCRYPTPROV hCP = NULL;
297     if (!::CryptAcquireContext(&hCP, NULL, NULL, PROV_RSA_FULL, 0) || hCP == NULL) {
298         return FALSE;
299     }
300     ::CryptGenRandom(hCP, iCount * sizeof(FX_DWORD), (uint8_t*)pBuffer);
301     ::CryptReleaseContext(hCP, 0);
302     return TRUE;
303 }
304 #endif
305 void FX_Random_GenerateCrypto(FX_DWORD* pBuffer, int32_t iCount)
306 {
307 #if _FXM_PLATFORM_ == _FXM_PLATFORM_WINDOWS_
308     FX_GenerateCryptoRandom(pBuffer, iCount);
309 #else
310     FX_Random_GenerateBase(pBuffer, iCount);
311 #endif
312 }
313 #ifdef __cplusplus
314 }
315 #endif