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