1 /** -- C++ Source File -- **/
2 
3 // MARKER(update_precomp.py): autogen include statement, do not remove
4 #include "precompiled_xmlsecurity.hxx"
5 #include <stdio.h>
6 #include "helper.hxx"
7 
8 #include "libxml/tree.h"
9 #include "libxml/parser.h"
10 #ifndef XMLSEC_NO_XSLT
11 #include "libxslt/xslt.h"
12 #endif
13 
14 
15 #include "securityenvironment_mscryptimpl.hxx"
16 #include "xmlelementwrapper_xmlsecimpl.hxx"
17 
18 #include "nspr.h"
19 #include "prtypes.h"
20 
21 #include "pk11func.h"
22 #include "cert.h"
23 #include "cryptohi.h"
24 #include "certdb.h"
25 #include "nss.h"
26 
27 #include "xmlsec/strings.h"
28 #include "xmlsec/xmltree.h"
29 
30 #include <rtl/ustring.hxx>
31 #include <cppuhelper/bootstrap.hxx>
32 #include <cppuhelper/servicefactory.hxx>
33 
34 #include <com/sun/star/beans/PropertyValue.hpp>
35 #include <com/sun/star/xml/wrapper/XXMLElementWrapper.hpp>
36 #include <com/sun/star/xml/wrapper/XXMLDocumentWrapper.hpp>
37 #include <com/sun/star/xml/crypto/XXMLEncryption.hpp>
38 #include <com/sun/star/xml/crypto/XXMLEncryptionTemplate.hpp>
39 #include <com/sun/star/xml/crypto/XXMLSecurityContext.hpp>
40 #include <com/sun/star/xml/crypto/XSecurityEnvironment.hpp>
41 
42 
43 using namespace ::rtl ;
44 using namespace ::cppu ;
45 using namespace ::com::sun::star::uno ;
46 using namespace ::com::sun::star::io ;
47 using namespace ::com::sun::star::ucb ;
48 using namespace ::com::sun::star::beans ;
49 using namespace ::com::sun::star::document ;
50 using namespace ::com::sun::star::lang ;
51 using namespace ::com::sun::star::registry ;
52 using namespace ::com::sun::star::xml::wrapper ;
53 using namespace ::com::sun::star::xml::crypto ;
54 
55 
56 int SAL_CALL main( int argc, char **argv )
57 {
58 	CERTCertDBHandle*	certHandle = NULL ;
59 	PK11SlotInfo*		slot = NULL ;
60 	xmlDocPtr			doc = NULL ;
61 	xmlNodePtr			tplNode ;
62 	xmlNodePtr			tarNode ;
63 	FILE*				dstFile = NULL ;
64 
65 
66 	if( argc != 5 ) {
67 		fprintf( stderr, "Usage: %s < CertDir > <input file_url> <output file_url> <rdb file>\n\n" , argv[0] ) ;
68 		return 1 ;
69 	}
70 
71 	//Init libxml and libxslt libraries
72 	xmlInitParser();
73 	LIBXML_TEST_VERSION
74 	xmlLoadExtDtdDefaultValue = XML_DETECT_IDS | XML_COMPLETE_ATTRS;
75 	xmlSubstituteEntitiesDefault(1);
76 
77 	#ifndef XMLSEC_NO_XSLT
78 	xmlIndentTreeOutput = 1;
79 	#endif // XMLSEC_NO_XSLT
80 
81 
82 	//Initialize NSPR and NSS
83 	PR_Init( PR_SYSTEM_THREAD, PR_PRIORITY_NORMAL, 1 ) ;
84 	PK11_SetPasswordFunc( PriPK11PasswordFunc ) ;
85 	if( NSS_Init( argv[1] ) != SECSuccess ) {
86 		fprintf( stderr , "### cannot intialize NSS!\n" ) ;
87 		goto done ;
88 	}
89 
90 	certHandle = CERT_GetDefaultCertDB() ;
91 	slot = PK11_GetInternalKeySlot() ;
92 
93 	//Load XML document
94 	doc = xmlParseFile( argv[2] ) ;
95 	if( doc == NULL || xmlDocGetRootElement( doc ) == NULL ) {
96 		fprintf( stderr , "### Cannot load template xml document!\n" ) ;
97 		goto done ;
98 	}
99 
100 	//Find the encryption template
101 	tplNode = xmlSecFindNode( xmlDocGetRootElement( doc ), xmlSecNodeEncryptedData, xmlSecEncNs ) ;
102 	if( tplNode == NULL ) {
103 		fprintf( stderr , "### Cannot find the encryption template!\n" ) ;
104 		goto done ;
105 	}
106 
107 
108 	try {
109 		Reference< XMultiComponentFactory > xManager = NULL ;
110 		Reference< XComponentContext > xContext = NULL ;
111 
112 		xManager = serviceManager( xContext , OUString::createFromAscii( "local" ), OUString::createFromAscii( argv[4] ) ) ;
113 
114 		//Create encryption template
115 		Reference< XInterface > tplElement =
116 			xManager->createInstanceWithContext( OUString::createFromAscii( "com.sun.star.xml.xsec.XMLElementWrapper" ) , xContext ) ;
117 		OSL_ENSURE( tplElement.is() ,
118 			"Decryptor - "
119 			"Cannot get service instance of \"xsec.XMLElementWrapper\"" ) ;
120 
121 		Reference< XXMLElementWrapper > xTplElement( tplElement , UNO_QUERY ) ;
122 		OSL_ENSURE( xTplElement.is() ,
123 			"Decryptor - "
124 			"Cannot get interface of \"XXMLElementWrapper\" from service \"xsec.XMLElementWrapper\"" ) ;
125 
126 		Reference< XUnoTunnel > xTplEleTunnel( xTplElement , UNO_QUERY ) ;
127 		OSL_ENSURE( xTplEleTunnel.is() ,
128 			"Decryptor - "
129 			"Cannot get interface of \"XUnoTunnel\" from service \"xsec.XMLElementWrapper\"" ) ;
130 
131 		XMLElementWrapper_XmlSecImpl* pTplElement = ( XMLElementWrapper_XmlSecImpl* )xTplEleTunnel->getSomething( XMLElementWrapper_XmlSecImpl::getUnoTunnelImplementationId() ) ;
132 		OSL_ENSURE( pTplElement != NULL ,
133 			"Decryptor - "
134 			"Cannot get implementation of \"xsec.XMLElementWrapper\"" ) ;
135 
136 		pTplElement->setNativeElement( tplNode ) ;
137 
138 		//Build XML Encryption template
139 		Reference< XInterface > enctpl =
140 			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.xsec.XMLEncryptionTemplate"), xContext ) ;
141 		OSL_ENSURE( enctpl.is() ,
142 			"Decryptor - "
143 			"Cannot get service instance of \"xsec.XMLEncryptionTemplate\"" ) ;
144 
145 		Reference< XXMLEncryptionTemplate > xTemplate( enctpl , UNO_QUERY ) ;
146 		OSL_ENSURE( xTemplate.is() ,
147 			"Decryptor - "
148 			"Cannot get interface of \"XXMLEncryptionTemplate\" from service \"xsec.XMLEncryptionTemplate\"" ) ;
149 
150 		//Import the encryption template
151 		xTemplate->setTemplate( xTplElement ) ;
152 
153 		//Create security environment
154 		//Build Security Environment
155 		Reference< XInterface > xsecenv =
156 			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.xsec.SecurityEnvironment"), xContext ) ;
157 		OSL_ENSURE( xsecenv.is() ,
158 			"Decryptor - "
159 			"Cannot get service instance of \"xsec.SecurityEnvironment\"" ) ;
160 
161 		Reference< XSecurityEnvironment > xSecEnv( xsecenv , UNO_QUERY ) ;
162 		OSL_ENSURE( xSecEnv.is() ,
163 			"Decryptor - "
164 			"Cannot get interface of \"XSecurityEnvironment\" from service \"xsec.SecurityEnvironment\"" ) ;
165 
166 		//Setup key slot and certDb
167 		Reference< XUnoTunnel > xEnvTunnel( xsecenv , UNO_QUERY ) ;
168 		OSL_ENSURE( xEnvTunnel.is() ,
169 			"Decryptor - "
170 			"Cannot get interface of \"XUnoTunnel\" from service \"xsec.SecurityEnvironment\"" ) ;
171 
172 		SecurityEnvironment_XmlSecImpl* pSecEnv = ( SecurityEnvironment_XmlSecImpl* )xEnvTunnel->getSomething( SecurityEnvironment_XmlSecImpl::getUnoTunnelId() ) ;
173 		OSL_ENSURE( pSecEnv != NULL ,
174 			"Decryptor - "
175 			"Cannot get implementation of \"xsec.SecurityEnvironment\"" ) ;
176 
177 		pSecEnv->setCryptoSlot( slot ) ;
178 		pSecEnv->setCertDb( certHandle ) ;
179 
180 
181 		//Build XML Security Context
182 		Reference< XInterface > xmlsecctx =
183 			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.xsec.XMLSecurityContext"), xContext ) ;
184 		OSL_ENSURE( xmlsecctx.is() ,
185 			"Decryptor - "
186 			"Cannot get service instance of \"xsec.XMLSecurityContext\"" ) ;
187 
188 		Reference< XXMLSecurityContext > xSecCtx( xmlsecctx , UNO_QUERY ) ;
189 		OSL_ENSURE( xSecCtx.is() ,
190 			"Decryptor - "
191 			"Cannot get interface of \"XXMLSecurityContext\" from service \"xsec.XMLSecurityContext\"" ) ;
192 
193 		xSecCtx->setSecurityEnvironment( xSecEnv ) ;
194 
195 
196 		//Get encrypter
197 		Reference< XInterface > xmlencrypter =
198 			xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.xsec.XMLEncryption"), xContext ) ;
199 		OSL_ENSURE( xmlencrypter.is() ,
200 			"Decryptor - "
201 			"Cannot get service instance of \"xsec.XMLEncryption\"" ) ;
202 
203 		Reference< XXMLEncryption > xEncrypter( xmlencrypter , UNO_QUERY ) ;
204 		OSL_ENSURE( xEncrypter.is() ,
205 			"Decryptor - "
206 			"Cannot get interface of \"XXMLEncryption\" from service \"xsec.XMLEncryption\"" ) ;
207 
208 
209 		//Perform decryption
210 		Reference< XXMLElementWrapper> xDecrRes = xEncrypter->decrypt( xTemplate , xSecCtx ) ;
211 		OSL_ENSURE( xDecrRes.is() ,
212 			"Decryptor - "
213 			"Cannot decrypt the xml document" ) ;
214 	} catch( Exception& e ) {
215 		fprintf( stderr , "Error Message: %s\n" , OUStringToOString( e.Message , RTL_TEXTENCODING_ASCII_US ).getStr() ) ;
216 		goto done ;
217 	}
218 
219 	dstFile = fopen( argv[3], "w" ) ;
220 	if( dstFile == NULL ) {
221 		fprintf( stderr , "### Can not open file %s\n", argv[3] ) ;
222 		goto done ;
223 	}
224 
225 	//Save result
226 	xmlDocDump( dstFile, doc ) ;
227 
228 done:
229 	if( dstFile != NULL )
230 		fclose( dstFile ) ;
231 
232 	if( slot != NULL )
233 		PK11_FreeSlot( slot ) ;
234 
235 	PK11_LogoutAll() ;
236 	NSS_Shutdown() ;
237 
238 	/* Shutdown libxslt/libxml */
239 	#ifndef XMLSEC_NO_XSLT
240 	xsltCleanupGlobals();
241 	#endif /* XMLSEC_NO_XSLT */
242 	xmlCleanupParser();
243 
244 	return 0;
245 }
246 
247