1 /************************************************************************* 2 * 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * Copyright 2000, 2010 Oracle and/or its affiliates. 6 * 7 * OpenOffice.org - a multi-platform office productivity suite 8 * 9 * This file is part of OpenOffice.org. 10 * 11 * OpenOffice.org is free software: you can redistribute it and/or modify 12 * it under the terms of the GNU Lesser General Public License version 3 13 * only, as published by the Free Software Foundation. 14 * 15 * OpenOffice.org is distributed in the hope that it will be useful, 16 * but WITHOUT ANY WARRANTY; without even the implied warranty of 17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 18 * GNU Lesser General Public License version 3 for more details 19 * (a copy is included in the LICENSE file that accompanied this code). 20 * 21 * You should have received a copy of the GNU Lesser General Public License 22 * version 3 along with OpenOffice.org. If not, see 23 * <http://www.openoffice.org/license.html> 24 * for a copy of the LGPLv3 License. 25 * 26 ************************************************************************/ 27 28 // MARKER(update_precomp.py): autogen include statement, do not remove 29 #include "precompiled_xmlsecurity.hxx" 30 31 #include <stdio.h> 32 #include "helper.hxx" 33 34 #include "libxml/tree.h" 35 #include "libxml/parser.h" 36 #ifndef XMLSEC_NO_XSLT 37 #include "libxslt/xslt.h" 38 #endif 39 40 #include "securityenvironment_nssimpl.hxx" 41 #include "xmlelementwrapper_xmlsecimpl.hxx" 42 43 #include "nspr.h" 44 #include "prtypes.h" 45 46 #include "pk11func.h" 47 #include "cert.h" 48 #include "cryptohi.h" 49 #include "certdb.h" 50 #include "nss.h" 51 52 #include "xmlsec/strings.h" 53 #include "xmlsec/xmltree.h" 54 55 #include <rtl/ustring.hxx> 56 #include <cppuhelper/servicefactory.hxx> 57 58 #include <com/sun/star/lang/XComponent.hpp> 59 #include <com/sun/star/beans/PropertyValue.hpp> 60 #include <com/sun/star/xml/wrapper/XXMLElementWrapper.hpp> 61 #include <com/sun/star/xml/wrapper/XXMLDocumentWrapper.hpp> 62 #include <com/sun/star/xml/crypto/XXMLEncryption.hpp> 63 #include <com/sun/star/xml/crypto/XXMLEncryptionTemplate.hpp> 64 #include <com/sun/star/xml/crypto/XXMLSecurityContext.hpp> 65 #include <com/sun/star/xml/crypto/XSecurityEnvironment.hpp> 66 67 using namespace ::rtl ; 68 using namespace ::cppu ; 69 using namespace ::com::sun::star::uno ; 70 using namespace ::com::sun::star::io ; 71 using namespace ::com::sun::star::ucb ; 72 using namespace ::com::sun::star::beans ; 73 using namespace ::com::sun::star::document ; 74 using namespace ::com::sun::star::lang ; 75 using namespace ::com::sun::star::registry ; 76 using namespace ::com::sun::star::xml::wrapper ; 77 using namespace ::com::sun::star::xml::crypto ; 78 79 int SAL_CALL main( int argc, char **argv ) 80 { 81 CERTCertDBHandle* certHandle ; 82 PK11SlotInfo* slot = NULL ; 83 PK11SymKey* symKey = NULL ; 84 xmlDocPtr doc = NULL ; 85 xmlNodePtr tplNode ; 86 xmlNodePtr tarNode ; 87 FILE* dstFile = NULL ; 88 89 if( argc != 7 ) { 90 fprintf( stderr, "Usage: %s < CertDir > <file_url of template> <file_url of result> <target element name> <target element namespace> <rdb file>\n\n" , argv[0] ) ; 91 return 1 ; 92 } 93 94 //Init libxml and libxslt libraries 95 xmlInitParser(); 96 LIBXML_TEST_VERSION 97 xmlLoadExtDtdDefaultValue = XML_DETECT_IDS | XML_COMPLETE_ATTRS; 98 xmlSubstituteEntitiesDefault(1); 99 100 #ifndef XMLSEC_NO_XSLT 101 xmlIndentTreeOutput = 1; 102 #endif // XMLSEC_NO_XSLT 103 104 //Initialize NSPR and NSS 105 PR_Init( PR_SYSTEM_THREAD, PR_PRIORITY_NORMAL, 1 ) ; 106 PK11_SetPasswordFunc( PriPK11PasswordFunc ) ; 107 if( NSS_Init( argv[1] ) != SECSuccess ) { 108 fprintf( stderr , "### cannot intialize NSS!\n" ) ; 109 goto done ; 110 } 111 112 certHandle = CERT_GetDefaultCertDB() ; 113 slot = PK11_GetInternalKeySlot() ; 114 115 symKey = PK11_KeyGen( slot , CKM_DES3_CBC, NULL, 128, NULL ) ; 116 if( symKey == NULL ) { 117 fprintf( stderr , "### cannot create symmetric key!\n" ) ; 118 goto done ; 119 } 120 121 //Load XML document 122 doc = xmlParseFile( argv[2] ) ; 123 if( doc == NULL || xmlDocGetRootElement( doc ) == NULL ) { 124 fprintf( stderr , "### Cannot load template xml document!\n" ) ; 125 goto done ; 126 } 127 128 //Find the encryption template 129 tplNode = xmlSecFindNode( xmlDocGetRootElement( doc ), xmlSecNodeEncryptedData, xmlSecEncNs ) ; 130 if( tplNode == NULL ) { 131 fprintf( stderr , "### Cannot find the encryption template!\n" ) ; 132 goto done ; 133 } 134 135 //Find the encryption template 136 tarNode = xmlSecFindNode( xmlDocGetRootElement( doc ), ( const unsigned char*)argv[4], ( const unsigned char*)argv[5] ) ; 137 if( tarNode == NULL ) { 138 fprintf( stderr , "### Cannot find the encryption target!\n" ) ; 139 goto done ; 140 } 141 142 try { 143 Reference< XMultiComponentFactory > xManager = NULL ; 144 Reference< XComponentContext > xContext = NULL ; 145 146 xManager = serviceManager( xContext , OUString::createFromAscii( "local" ), OUString::createFromAscii( argv[6] ) ) ; 147 148 //Create encryption template 149 Reference< XInterface > tplElement = 150 xManager->createInstanceWithContext( OUString::createFromAscii( "com.sun.star.xml.security.bridge.xmlsec.XMLElementWrapper_XmlSecImpl" ) , xContext ) ; 151 OSL_ENSURE( tplElement.is() , 152 "Encryptor - " 153 "Cannot get service instance of \"xsec.XMLElementWrapper\"" ) ; 154 155 Reference< XXMLElementWrapper > xTplElement( tplElement , UNO_QUERY ) ; 156 OSL_ENSURE( xTplElement.is() , 157 "Encryptor - " 158 "Cannot get interface of \"XXMLElementWrapper\" from service \"xsec.XMLElementWrapper\"" ) ; 159 160 Reference< XUnoTunnel > xTplEleTunnel( xTplElement , UNO_QUERY ) ; 161 OSL_ENSURE( xTplEleTunnel.is() , 162 "Encryptor - " 163 "Cannot get interface of \"XUnoTunnel\" from service \"xsec.XMLElementWrapper\"" ) ; 164 165 XMLElementWrapper_XmlSecImpl* pTplElement = ( XMLElementWrapper_XmlSecImpl* )xTplEleTunnel->getSomething( XMLElementWrapper_XmlSecImpl::getUnoTunnelImplementationId() ) ; 166 OSL_ENSURE( pTplElement != NULL , 167 "Encryptor - " 168 "Cannot get implementation of \"xsec.XMLElementWrapper\"" ) ; 169 170 pTplElement->setNativeElement( tplNode ) ; 171 172 //Create encryption target element 173 Reference< XInterface > tarElement = 174 xManager->createInstanceWithContext( OUString::createFromAscii( "com.sun.star.xml.security.bridge.xmlsec.XMLElementWrapper_XmlSecImpl" ) , xContext ) ; 175 OSL_ENSURE( tarElement.is() , 176 "Encryptor - " 177 "Cannot get service instance of \"xsec.XMLElementWrapper\"" ) ; 178 179 Reference< XXMLElementWrapper > xTarElement( tarElement , UNO_QUERY ) ; 180 OSL_ENSURE( xTarElement.is() , 181 "Encryptor - " 182 "Cannot get interface of \"XXMLElementWrapper\" from service \"xsec.XMLElementWrapper\"" ) ; 183 184 Reference< XUnoTunnel > xTarEleTunnel( xTarElement , UNO_QUERY ) ; 185 OSL_ENSURE( xTarEleTunnel.is() , 186 "Encryptor - " 187 "Cannot get interface of \"XUnoTunnel\" from service \"xsec.XMLElementWrapper\"" ) ; 188 189 XMLElementWrapper_XmlSecImpl* pTarElement = ( XMLElementWrapper_XmlSecImpl* )xTarEleTunnel->getSomething( XMLElementWrapper_XmlSecImpl::getUnoTunnelImplementationId() ) ; 190 OSL_ENSURE( pTarElement != NULL , 191 "Encryptor - " 192 "Cannot get implementation of \"xsec.XMLElementWrapper\"" ) ; 193 194 pTarElement->setNativeElement( tarNode ) ; 195 196 197 //Build XML Encryption template 198 Reference< XInterface > enctpl = 199 xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.crypto.XMLEncryptionTemplate"), xContext ) ; 200 OSL_ENSURE( enctpl.is() , 201 "Encryptor - " 202 "Cannot get service instance of \"xsec.XMLEncryptionTemplate\"" ) ; 203 204 Reference< XXMLEncryptionTemplate > xTemplate( enctpl , UNO_QUERY ) ; 205 OSL_ENSURE( xTemplate.is() , 206 "Encryptor - " 207 "Cannot get interface of \"XXMLEncryptionTemplate\" from service \"xsec.XMLEncryptionTemplate\"" ) ; 208 209 //Import the encryption template 210 xTemplate->setTemplate( xTplElement ) ; 211 xTemplate->setTarget( xTarElement ) ; 212 213 //Create security environment 214 //Build Security Environment 215 Reference< XInterface > xsecenv = 216 xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.security.bridge.xmlsec.SecurityEnvironment_NssImpl"), xContext ) ; 217 OSL_ENSURE( xsecenv.is() , 218 "Encryptor - " 219 "Cannot get service instance of \"xsec.SecurityEnvironment\"" ) ; 220 221 Reference< XSecurityEnvironment > xSecEnv( xsecenv , UNO_QUERY ) ; 222 OSL_ENSURE( xSecEnv.is() , 223 "Encryptor - " 224 "Cannot get interface of \"XSecurityEnvironment\" from service \"xsec.SecurityEnvironment\"" ) ; 225 226 //Setup key slot and certDb 227 Reference< XUnoTunnel > xEnvTunnel( xsecenv , UNO_QUERY ) ; 228 OSL_ENSURE( xEnvTunnel.is() , 229 "Encryptor - " 230 "Cannot get interface of \"XUnoTunnel\" from service \"xsec.SecurityEnvironment\"" ) ; 231 232 SecurityEnvironment_NssImpl* pSecEnv = ( SecurityEnvironment_NssImpl* )xEnvTunnel->getSomething( SecurityEnvironment_NssImpl::getUnoTunnelId() ) ; 233 OSL_ENSURE( pSecEnv != NULL , 234 "Encryptor - " 235 "Cannot get implementation of \"xsec.SecurityEnvironment\"" ) ; 236 237 pSecEnv->setCryptoSlot( slot ) ; 238 pSecEnv->setCertDb( certHandle ) ; 239 pSecEnv->adoptSymKey( symKey ) ; 240 241 242 //Build XML Security Context 243 Reference< XInterface > xmlsecctx = 244 xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.security.bridge.xmlsec.XMLSecurityContext_NssImpl"), xContext ) ; 245 OSL_ENSURE( xmlsecctx.is() , 246 "Encryptor - " 247 "Cannot get service instance of \"xsec.XMLSecurityContext\"" ) ; 248 249 Reference< XXMLSecurityContext > xSecCtx( xmlsecctx , UNO_QUERY ) ; 250 OSL_ENSURE( xSecCtx.is() , 251 "Encryptor - " 252 "Cannot get interface of \"XXMLSecurityContext\" from service \"xsec.XMLSecurityContext\"" ) ; 253 254 xSecCtx->setSecurityEnvironment( xSecEnv ) ; 255 256 //Get encrypter 257 Reference< XInterface > xmlencrypter = 258 xManager->createInstanceWithContext( OUString::createFromAscii("com.sun.star.xml.security.bridge.xmlsec.XMLEncryption_NssImpl"), xContext ) ; 259 OSL_ENSURE( xmlencrypter.is() , 260 "Encryptor - " 261 "Cannot get service instance of \"xsec.XMLEncryption\"" ) ; 262 263 Reference< XXMLEncryption > xEncrypter( xmlencrypter , UNO_QUERY ) ; 264 OSL_ENSURE( xEncrypter.is() , 265 "Encryptor - " 266 "Cannot get interface of \"XXMLEncryption\" from service \"xsec.XMLEncryption\"" ) ; 267 268 //perform encryption 269 xTemplate = xEncrypter->encrypt( xTemplate , xSecCtx ) ; 270 OSL_ENSURE( xTemplate.is() , 271 "Encryptor - " 272 "Cannot encrypt the xml document" ) ; 273 } catch( Exception& e ) { 274 fprintf( stderr , "Error Message: %s\n" , OUStringToOString( e.Message , RTL_TEXTENCODING_ASCII_US ).getStr() ) ; 275 goto done ; 276 } 277 278 dstFile = fopen( argv[3], "w" ) ; 279 if( dstFile == NULL ) { 280 fprintf( stderr , "### Can not open file %s\n", argv[3] ) ; 281 goto done ; 282 } 283 284 //Save result 285 xmlDocDump( dstFile, doc ) ; 286 287 done: 288 if( dstFile != NULL ) 289 fclose( dstFile ) ; 290 291 if( symKey != NULL ) { 292 PK11_FreeSymKey( symKey ) ; 293 } 294 295 if( slot != NULL ) 296 PK11_FreeSlot( slot ) ; 297 298 PK11_LogoutAll() ; 299 NSS_Shutdown() ; 300 301 /* Shutdown libxslt/libxml */ 302 #ifndef XMLSEC_NO_XSLT 303 xsltCleanupGlobals(); 304 #endif /* XMLSEC_NO_XSLT */ 305 xmlCleanupParser(); 306 307 return 0; 308 } 309 310