src/apps/client.cpp

   1 #include <sys/stat.h>
   2 #include <unistd.h>
   3
   4 #include <iostream>
   5 #include <fstream>
   6 #include <streambuf>
   7 #include <unordered_map>
   8
   9 #include "db/database.h"
  10 #include "db/psql.h"
  11 #include "crypto/simpleOpensslSigner.h"
  12 #include "crypto/remoteSigner.h"
  13 #include "crypto/sslUtil.h"
  14 #include "log/logger.hpp"
  15 #include "util.h"
  16 #include "io/bios.h"
  17 #include "io/slipBio.h"
  18 #include "config.h"
  19 #include <internal/bio.h>
  20
  21 #ifdef NO_DAEMON
  22 #define DAEMON false
  23 #else
  24 #define DAEMON true
  25 #endif
  26
  27 extern std::string keyDir;
  28 extern std::string sqlHost, sqlUser, sqlPass, sqlDB;
  29 extern std::string serialPath;
  30 extern std::unordered_map<std::string, std::shared_ptr<CAConfig>> CAs;
  31
  32 void checkCRLs( std::shared_ptr<Signer> sign ) {
  33
  34     logger::note( "Signing CRLs" );
  35
  36     for( auto& x : CAs ) {
  37         logger::notef( "Checking: %s ...", x.first );
  38
  39         if( !x.second->crlNeedsResign() ) {
  40             logger::warnf( "Skipping Resigning CRL: %s ...", x.second->name );
  41             continue;
  42         }
  43
  44         logger::notef( "Resigning CRL: %s ...", x.second->name );
  45
  46         try {
  47             std::vector<std::string> serials;
  48             std::pair<std::shared_ptr<CRL>, std::string> rev = sign->revoke( x.second, serials );
  49         } catch( const std::exception& e ) {
  50             logger::error( "Exception: ", e.what() );
  51         }
  52     }
  53 }
  54
  55 int main( int argc, const char *argv[] ) {
  56     bool once = false;
  57     bool resetOnly = false;
  58
  59     if( argc == 2 && std::string( "--once" ) == argv[1] ) {
  60         once = true;
  61     }
  62
  63     if( argc == 2 && std::string( "--reset" ) == argv[1] ) {
  64         resetOnly = true;
  65     }
  66
  67     std::string path;
  68
  69 #ifdef NDEBUG
  70     path = "/etc/wpia/cassiopeia/cassiopeia.conf";
  71 #else
  72     path = "config.txt";
  73 #endif
  74
  75     if( parseConfig( path ) != 0 ) {
  76         logger::fatal( "Error: Could not parse the configuration file." );
  77         return -1;
  78     }
  79
  80     if( serialPath == "" ) {
  81         logger::fatal( "Error: no serial device is given!" );
  82         return -1;
  83     }
  84
  85     std::shared_ptr<JobProvider> jp = std::make_shared<PostgresJobProvider>( sqlHost, sqlUser, sqlPass, sqlDB );
  86     std::shared_ptr<BIO> b = openSerial( serialPath );
  87     std::shared_ptr<BIO_METHOD> m( toBio<SlipBIO>(), BIO_meth_free );
  88     std::shared_ptr<BIO> slip1( BIO_new( m.get() ), BIO_free );
  89     static_cast<SlipBIO *>( slip1->ptr )->setTarget( std::make_shared<OpensslBIOWrapper>( b ), false );
  90     auto sign = std::make_shared<RemoteSigner>( slip1, generateSSLContext( false ) );
  91     // std::shared_ptr<Signer> sign( new SimpleOpensslSigner() );
  92
  93     if( resetOnly ) {
  94         std::cout << "Doing BIO reset" << std::endl;
  95         int result = BIO_reset( slip1.get() );
  96         std::cout << "Did BIO reset, result " << result << ", exiting." << std::endl;
  97         return result;
  98     }
  99
 100     time_t lastCRLCheck = 0;
 101
 102     while( true ) {
 103         try {
 104             time_t current;
 105             time( &current );
 106
 107             if( lastCRLCheck + 30 * 60 < current ) {
 108                 // todo set good log TODO FIXME
 109                 auto ostreamFree = []( std::ostream * o ) {
 110                     ( void ) o;
 111                 };
 112                 sign->setLog( std::shared_ptr<std::ostream>( &std::cout, ostreamFree ) );
 113                 checkCRLs( sign );
 114                 lastCRLCheck = current;
 115             }
 116
 117             std::shared_ptr<Job> job;
 118
 119             try {
 120                 job = jp->fetchJob();
 121             } catch( std::exception& e ) {
 122                 logger::errorf( "Exception while fetchJob: %s", e.what() );
 123             }
 124
 125             if( !job ) {
 126                 sleep( 5 );
 127                 continue;
 128             }
 129
 130             std::shared_ptr<std::ofstream> logPtr = openLogfile( std::string( "logs/" ) + job->id + std::string( "_" ) + job->warning + std::string( ".log" ) );
 131
 132             logger::logger_set log_set( {logger::log_target( *logPtr, logger::level::debug )}, logger::auto_register::on );
 133
 134             logger::note( "TASK ID: ", job->id );
 135             logger::note( "TRY:     ", job->warning );
 136             logger::note( "TARGET:  ", job->target );
 137             logger::note( "TASK:    ", job->task );
 138
 139             if( job->task == "sign" ) {
 140                 try {
 141                     std::shared_ptr<TBSCertificate> cert = jp->fetchTBSCert( job );
 142
 143                     if( !cert ) {
 144                         logger::error( "Unable to load CSR" );
 145                         jp->failJob( job );
 146                         continue;
 147                     }
 148
 149                     cert->wishFrom = job->from;
 150                     cert->wishTo = job->to;
 151                     logger::note( "INFO: Message Digest: ", cert->md );
 152                     logger::note( "INFO: Profile ID: ", cert->profile );
 153
 154                     for( auto& SAN : cert->SANs ) {
 155                         logger::notef( "INFO: SAN %s: %s", SAN->type, SAN->content );
 156                     }
 157
 158                     for( auto& AVA : cert->AVAs ) {
 159                         logger::notef( "INFO: AVA %s: %s", AVA->name, AVA->value );
 160                     }
 161
 162                     logger::notef( "FINE: Found the CSR at '%s'", cert->csr );
 163                     cert->csr_content = readFile( keyDir + "/../" + cert->csr );
 164                     logger::note( "FINE: CSR content:\n", cert->csr_content );
 165
 166                     std::shared_ptr<SignedCertificate> res = sign->sign( cert );
 167
 168                     if( !res ) {
 169                         logger::error( "ERROR: The signer failed. No certificate was returned." );
 170                         jp->failJob( job );
 171                         continue;
 172                     }
 173
 174                     logger::note( "FINE: CERTIFICATE LOG:\n", res->log,
 175                                   "FINE: CERTIFICATE:\n", res->certificate );
 176
 177                     std::string fn = writeBackFile( job->target.c_str(), res->certificate, keyDir );
 178
 179                     if( fn.empty() ) {
 180                         logger::error( "ERROR: Writeback of the certificate failed." );
 181                         jp->failJob( job );
 182                         continue;
 183                     }
 184
 185                     res->crt_name = fn;
 186                     jp->writeBack( job, res ); //! \FIXME: Check return value
 187                     logger::note( "FINE: signing done." );
 188
 189                     if( DAEMON ) {
 190                         jp->finishJob( job );
 191                     }
 192
 193                     continue;
 194                 } catch( std::exception& c ) {
 195                     logger::error( "ERROR: ", c.what() );
 196                 }
 197
 198                 try {
 199                     jp->failJob( job );
 200                 } catch( std::exception& c ) {
 201                     logger::error( "ERROR: ", c.what() );
 202                 }
 203             } else if( job->task == "revoke" ) {
 204                 try {
 205                     logger::note( "revoking" );
 206                     auto data = jp->getRevocationInfo( job );
 207                     std::vector<std::string> serials;
 208                     serials.push_back( data.first );
 209                     logger::note( "revoking" );
 210                     std::pair<std::shared_ptr<CRL>, std::string> rev = sign->revoke( CAs.at( data.second ), serials );
 211                     std::string date = rev.second;
 212                     const unsigned char *pos = ( const unsigned char * ) date.data();
 213                     std::shared_ptr<ASN1_TIME> time( d2i_ASN1_TIME( NULL, &pos, date.size() ), ASN1_TIME_free );
 214
 215                     jp->writeBackRevocation( job, timeToString( time ) );
 216                     jp->finishJob( job );
 217                 } catch( const std::exception& c ) {
 218                     logger::error( "Exception: ", c.what() );
 219                 }
 220             } else {
 221                 logger::errorf( "Unknown job type (\"%s\")", job->task );
 222                 jp->failJob( job );
 223             }
 224
 225             if( !DAEMON || once ) {
 226                 return 0;
 227             }
 228         } catch( std::exception& e ) {
 229             logger::errorf( "std::exception in mainloop: %s", e.what() );
 230         }
 231
 232     }
 233 }