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