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