]> WPIA git - cassiopeia.git/blob - src/util.cpp
upd: clean up valgrind stuff
[cassiopeia.git] / src / util.cpp
1 #include "util.h"
2
3 #include <sys/stat.h>
4
5 #include <fstream>
6 #include <iostream>
7 #include <sstream>
8 #include <time.h>
9 #include <stdexcept>
10
11 void writeFile( const std::string& name, const std::string& content ) {
12     std::ofstream file( name );
13     file << content;
14
15     //! \FIXME: Error checking
16 }
17
18 std::string readFile( const std::string& name ) {
19     std::ifstream t( name );
20     std::string res = std::string( std::istreambuf_iterator<char>( t ), std::istreambuf_iterator<char>() );
21
22     return res;
23 }
24
25 std::string writeBackFile( const std::string& serial, const std::string& cert, const std::string& keydir ) {
26     errno = 0;
27
28     std::string filename = keydir;
29
30     if( 0 != mkdir( filename.c_str(), 0755 ) ) {
31         if( EEXIST != errno ) {
32             throw std::runtime_error("Storage location could not be determined");
33         }
34
35         //! \FIXME: Check this is a directory
36     }
37
38     filename += "/crt";
39
40     if( 0 != mkdir( filename.c_str(), 0755 ) ) {
41         if( EEXIST != errno ) {
42             return "";
43         }
44
45         //! \FIXME: Check this is a directory
46     }
47
48     std::string first;
49
50     if( serial.length() < 3 ) {
51         first = "0";
52     } else {
53         first = serial.substr( 0, serial.length() - 3 );
54     }
55
56     filename += "/" + first;
57
58     if( 0 != mkdir( filename.c_str(), 0755 ) ) {
59         if( EEXIST != errno ) {
60             return "";
61         }
62
63         //! \FIXME: Check this is a directory
64     }
65
66     filename += "/" + serial + ".crt";
67     writeFile( filename, cert );
68
69     return filename;
70 }
71
72 bool isDigit( char c ) {
73     return ( c >= '0' ) && ( c <= '9' );
74 }
75
76 std::pair<bool, time_t> parseDate( const std::string& date ) {
77     if( date.size() != 10 || date[4] != '-' || date[7] != '-' ) {
78         return std::pair<bool, time_t>( false, 0 );
79     }
80
81     if( !isDigit( date[0] )
82             || !isDigit( date[1] )
83             || !isDigit( date[2] )
84             || !isDigit( date[3] )
85             || !isDigit( date[5] )
86             || !isDigit( date[6] )
87             || !isDigit( date[8] )
88             || !isDigit( date[9] ) ) {
89         return std::pair<bool, time_t>( false, 0 );
90     }
91
92     std::tm t;
93     t.tm_sec = 0;
94     t.tm_min = 0;
95     t.tm_hour = 0;
96     t.tm_year = std::stoi( date.substr( 0, 4 ) ) - 1900;
97     t.tm_mon = std::stoi( date.substr( 5, 2 ) ) - 1;
98     t.tm_mday = std::stoi( date.substr( 8, 2 ) );
99     t.tm_wday = 0;   /* Day of the week (0-6, Sunday = 0) */
100     t.tm_yday = 0;   /* Day in the year (0-365, 1 Jan = 0) */
101     t.tm_isdst = 0;  /* Daylight saving time */
102
103     setenv( "TZ", "UTC", 1 );
104     tzset();
105     std::time_t res = mktime( &t );
106     char check[11];
107     std::size_t siz = strftime( check, 11, "%Y-%m-%d", &t );
108
109     if( siz != 10 ) {
110         return std::pair<bool, time_t>( false, 0 ); // NO-COVERAGE (by contract of strftime)
111     }
112
113     std::string checkS( check, siz );
114
115     if( checkS != date ) {
116         return { false, 0 };
117     }
118
119     return std::pair<bool, time_t>( true, res );
120 }
121
122 std::pair<bool, time_t> addMonths( std::time_t t, int32_t count ) {
123     std::tm* parsed = gmtime( &t );
124
125     if( !parsed || count <= 0 || count > 24 ) { // FIXED MAX-Validity-Length
126         return std::pair<bool, time_t>( false, 0 );
127     }
128
129     parsed->tm_mon += count;
130     int oldday = parsed->tm_mday;
131     setenv( "TZ", "UTC", 1 );
132     tzset();
133     std::time_t res = mktime( parsed );
134
135     if( parsed->tm_mday != oldday ) {
136         parsed->tm_mday = 0;
137         res = mktime( parsed );
138     }
139
140     return std::pair<bool, time_t>( true, res );
141 }
142
143 std::pair<bool, time_t> parseMonthInterval( std::time_t t, const std::string& date ) {
144     if( date[date.size() - 1] != 'm' ) {
145         return  std::pair<bool, time_t>( false, 0 );
146     }
147
148     try {
149         size_t end = 0;
150         int num = std::stoi( date.substr( 0, date.size() - 1 ) , &end );
151
152         if( end != date.size() - 1 ) {
153             return  std::pair<bool, time_t>( false, 0 );
154         }
155
156         return addMonths( t, num );
157     } catch( const std::invalid_argument& a ) {
158         return std::pair<bool, time_t>( false, 0 );
159     } catch( const std::out_of_range& a ) {
160         return std::pair<bool, time_t>( false, 0 );
161     }
162 }
163
164 std::pair<bool, time_t> parseYearInterval( std::time_t t, const std::string& date ) {
165     if( date[date.size() - 1] != 'y' ) {
166         return  std::pair<bool, time_t>( false, 0 );
167     }
168
169     try {
170         size_t end = 0;
171         int num = std::stoi( date.substr( 0, date.size() - 1 ), &end );
172
173         if( end != date.size() - 1 ) {
174             return  std::pair<bool, time_t>( false, 0 );
175         }
176
177         return addMonths( t, num * 12 );
178     } catch( std::invalid_argument& a ) {
179         return std::pair<bool, time_t>( false, 0 );
180     } catch( std::out_of_range& a ) {
181         return std::pair<bool, time_t>( false, 0 );
182     }
183 }
184
185 std::unique_ptr<std::ofstream> openLogfile( const std::string &name ) {
186     struct stat buffer;
187     std::string tname = name;
188     int ctr = 2;
189
190     while( stat( tname.c_str(), &buffer ) == 0 ) {
191         tname = name + "_" + std::to_string( ctr++ );
192     }
193
194     auto res = std::make_unique<std::ofstream>( tname );
195
196     if( ! res->good() ) {
197         throw std::runtime_error( std::string("Failed to open file for logging: " ) + name );
198     }
199
200     return res;
201 }
202
203 std::string timestamp(){
204     time_t c_time;
205     if( time( &c_time ) == -1 ) {
206         throw std::runtime_error( "Error while fetching time?" );
207     }
208     return std::to_string( c_time );
209 }