Initial commit.
[pdfium.git] / core / src / fpdfapi / fpdf_basic_module.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/fxcodec/fx_codec.h"\r
8 #include "../../include/fpdfapi/fpdf_module.h"\r
9 static CPDF_ModuleMgr*  g_FPDFAPI_pDefaultMgr = NULL;\r
10 CPDF_ModuleMgr* CPDF_ModuleMgr::Get()\r
11 {\r
12     return g_FPDFAPI_pDefaultMgr;\r
13 }\r
14 void CPDF_ModuleMgr::Create()\r
15 {\r
16     g_FPDFAPI_pDefaultMgr = FX_NEW CPDF_ModuleMgr;\r
17     g_FPDFAPI_pDefaultMgr->Initialize();\r
18 }\r
19 void CPDF_ModuleMgr::Destroy()\r
20 {\r
21     if (g_FPDFAPI_pDefaultMgr) {\r
22         delete g_FPDFAPI_pDefaultMgr;\r
23     }\r
24     g_FPDFAPI_pDefaultMgr = NULL;\r
25 }\r
26 CPDF_ModuleMgr::CPDF_ModuleMgr()\r
27 {\r
28     m_pCodecModule = NULL;\r
29     m_pPageModule = NULL;\r
30     m_pRenderModule = NULL;\r
31     m_FileBufSize = 512;\r
32 }\r
33 void CPDF_ModuleMgr::Initialize()\r
34 {\r
35     InitModules();\r
36     m_FileBufSize = 512;\r
37 }\r
38 void CPDF_ModuleMgr::InitModules()\r
39 {\r
40     m_pCodecModule = NULL;\r
41     m_pPageModule = FX_NEW CPDF_PageModuleDef;\r
42     m_pRenderModule = FX_NEW CPDF_RenderModuleDef;\r
43 }\r
44 CPDF_ModuleMgr::~CPDF_ModuleMgr()\r
45 {\r
46     if (m_pPageModule) {\r
47         delete m_pPageModule;\r
48     }\r
49     if (m_pRenderModule) {\r
50         delete m_pRenderModule;\r
51     }\r
52 }\r
53 void CPDF_ModuleMgr::SetDownloadCallback(FX_BOOL (*callback)(FX_LPCSTR module_name))\r
54 {\r
55     m_pDownloadCallback = callback;\r
56 }\r
57 FX_BOOL CPDF_ModuleMgr::DownloadModule(FX_LPCSTR module_name)\r
58 {\r
59     if (m_pDownloadCallback == NULL) {\r
60         return FALSE;\r
61     }\r
62     return m_pDownloadCallback(module_name);\r
63 }\r
64 static CFX_ByteString _GetPath(const CFX_ByteString& folder, FX_LPCSTR name)\r
65 {\r
66     FX_STRSIZE folder_len = folder.GetLength();\r
67 #if _FX_OS_ == _FX_SYMBIAN_ || _FXM_PLATFORM_  == _FXM_PLATFORM_WINDOWS_\r
68     if (folder[folder_len - 1] == '\\') {\r
69         return folder + name;\r
70     } else {\r
71         return (folder + "\\") + name;\r
72     }\r
73 #else\r
74     if (folder[folder_len - 1] == '/') {\r
75         return folder + name;\r
76     } else {\r
77         return (folder + "/") + name;\r
78     }\r
79 #endif\r
80 }\r
81 void CPDF_ModuleMgr::SetModulePath(FX_LPCSTR module_name, FX_LPCSTR path)\r
82 {\r
83     if (module_name == NULL || module_name[0] == 0) {\r
84         m_DefaultModulePath = path;\r
85     } else {\r
86         m_ModulePathList.SetAt(module_name, FX_NEW CFX_ByteString(path, -1));\r
87     }\r
88 }\r
89 CFX_ByteString CPDF_ModuleMgr::GetModuleFilePath(FX_LPCSTR module_name, FX_LPCSTR name)\r
90 {\r
91     CFX_ByteString* pPath = NULL;\r
92     if (m_ModulePathList.Lookup(module_name, (FX_LPVOID&)pPath)) {\r
93         return _GetPath(*pPath, name);\r
94     }\r
95     if (!m_DefaultModulePath.IsEmpty()) {\r
96         return _GetPath(m_DefaultModulePath, name);\r
97     }\r
98 #if _FXM_PLATFORM_  == _FXM_PLATFORM_WINDOWS_\r
99     FX_WCHAR app_path[260];\r
100     ::GetModuleFileNameW(NULL, (LPWSTR)app_path, 260);\r
101     FX_INTPTR len = FXSYS_wcslen(app_path);\r
102     for (FX_INTPTR i = len; i >= 0; i --)\r
103         if (app_path[i] == '\\') {\r
104             app_path[i] = 0;\r
105             break;\r
106         }\r
107     CFX_ByteString path = CFX_ByteString::FromUnicode(app_path);\r
108     path += '\\';\r
109     path += name;\r
110     return path;\r
111 #else\r
112     return name;\r
113 #endif\r
114 }\r
115 void CPDF_ModuleMgr::NotifyModuleAvailable(FX_LPCSTR module_name)\r
116 {\r
117     if (FXSYS_strcmp(module_name, ADDIN_NAME_CJK) == 0) {\r
118         m_pPageModule->NotifyCJKAvailable();\r
119     } else if (FXSYS_strcmp(module_name, ADDIN_NAME_DECODER) == 0) {\r
120         m_pRenderModule->NotifyDecoderAvailable();\r
121     }\r
122 }\r
123 void CPDF_ModuleMgr::RegisterSecurityHandler(FX_LPCSTR filter, CPDF_SecurityHandler * (*CreateHandler)(void* param), void* param)\r
124 {\r
125     if (CreateHandler == NULL) {\r
126         m_SecurityHandlerMap.RemoveKey(filter);\r
127     } else {\r
128         m_SecurityHandlerMap.SetAt(filter, (void*)CreateHandler);\r
129     }\r
130     if (param) {\r
131         m_SecurityHandlerMap.SetAt(FX_BSTRC("_param_") + filter, param);\r
132     }\r
133 }\r
134 void CPDF_ModuleMgr::SetPrivateData(FX_LPVOID module_id, FX_LPVOID pData, PD_CALLBACK_FREEDATA callback)\r
135 {\r
136     m_privateData.SetPrivateData(module_id, pData, callback);\r
137 }\r
138 FX_LPVOID CPDF_ModuleMgr::GetPrivateData(FX_LPVOID module_id)\r
139 {\r
140     return m_privateData.GetPrivateData(module_id);\r
141 }\r
142 CPDF_SecurityHandler* CPDF_ModuleMgr::CreateSecurityHandler(FX_LPCSTR filter)\r
143 {\r
144     CPDF_SecurityHandler* (*CreateHandler)(void*) = NULL;\r
145     if (!m_SecurityHandlerMap.Lookup(filter, (void*&)CreateHandler)) {\r
146         return NULL;\r
147     }\r
148     if (CreateHandler == NULL) {\r
149         return NULL;\r
150     }\r
151     void* param = NULL;\r
152     m_SecurityHandlerMap.Lookup(FX_BSTRC("_param_") + filter, param);\r
153     return CreateHandler(param);\r
154 }\r
155 ICodec_FaxModule* CPDF_ModuleMgr::GetFaxModule()\r
156 {\r
157     return m_pCodecModule ? m_pCodecModule->GetFaxModule() : NULL;\r
158 }\r
159 ICodec_JpegModule* CPDF_ModuleMgr::GetJpegModule()\r
160 {\r
161     return m_pCodecModule ? m_pCodecModule->GetJpegModule() : NULL;\r
162 }\r
163 ICodec_JpxModule* CPDF_ModuleMgr::GetJpxModule()\r
164 {\r
165     return m_pCodecModule ? m_pCodecModule->GetJpxModule() : NULL;\r
166 }\r
167 ICodec_Jbig2Module* CPDF_ModuleMgr::GetJbig2Module()\r
168 {\r
169     return m_pCodecModule ? m_pCodecModule->GetJbig2Module() : NULL;\r
170 }\r
171 ICodec_IccModule* CPDF_ModuleMgr::GetIccModule()\r
172 {\r
173     return m_pCodecModule ? m_pCodecModule->GetIccModule() : NULL;\r
174 }\r
175 ICodec_FlateModule* CPDF_ModuleMgr::GetFlateModule()\r
176 {\r
177     return m_pCodecModule ? m_pCodecModule->GetFlateModule() : NULL;\r
178 }\r