xmlsecurity/tools/standalone/mscsfit/decrypter.cxx

   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