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