author | Michael Krelin <hacker@klever.net> | 2007-11-21 12:09:06 (UTC) |
---|---|---|
committer | Michael Krelin <hacker@klever.net> | 2007-11-21 12:09:06 (UTC) |
commit | d31def2b6dc0fb5c4a17918d608d9f5107372a74 (patch) (unidiff) | |
tree | a52663a3db03f75d647e6dfea5fe173cdd37d528 | |
parent | f0877a6af785a36ef769114816c71a838295d902 (diff) | |
download | libopkele-d31def2b6dc0fb5c4a17918d608d9f5107372a74.zip libopkele-d31def2b6dc0fb5c4a17918d608d9f5107372a74.tar.gz libopkele-d31def2b6dc0fb5c4a17918d608d9f5107372a74.tar.bz2 |
keep compiler happy and check more error conditions
Thanks Marcus Rueckert for pointing this out
Signed-off-by: Michael Krelin <hacker@klever.net>
-rw-r--r-- | lib/util.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/lib/util.cc b/lib/util.cc index 94f6f53..94e09ed 100644 --- a/lib/util.cc +++ b/lib/util.cc | |||
@@ -7,51 +7,52 @@ | |||
7 | #include <curl/curl.h> | 7 | #include <curl/curl.h> |
8 | #include "opkele/util.h" | 8 | #include "opkele/util.h" |
9 | #include "opkele/exception.h" | 9 | #include "opkele/exception.h" |
10 | 10 | ||
11 | namespace opkele { | 11 | namespace opkele { |
12 | using namespace std; | 12 | using namespace std; |
13 | 13 | ||
14 | namespace util { | 14 | namespace util { |
15 | 15 | ||
16 | /* | 16 | /* |
17 | * base64 | 17 | * base64 |
18 | */ | 18 | */ |
19 | string encode_base64(const void *data,size_t length) { | 19 | string encode_base64(const void *data,size_t length) { |
20 | BIO *b64 = 0, *bmem = 0; | 20 | BIO *b64 = 0, *bmem = 0; |
21 | try { | 21 | try { |
22 | b64 = BIO_new(BIO_f_base64()); | 22 | b64 = BIO_new(BIO_f_base64()); |
23 | if(!b64) | 23 | if(!b64) |
24 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new() base64 encoder"); | 24 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new() base64 encoder"); |
25 | BIO_set_flags(b64,BIO_FLAGS_BASE64_NO_NL); | 25 | BIO_set_flags(b64,BIO_FLAGS_BASE64_NO_NL); |
26 | bmem = BIO_new(BIO_s_mem()); | 26 | bmem = BIO_new(BIO_s_mem()); |
27 | BIO_set_flags(b64,BIO_CLOSE); | 27 | BIO_set_flags(b64,BIO_CLOSE); |
28 | if(!bmem) | 28 | if(!bmem) |
29 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new() memory buffer"); | 29 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new() memory buffer"); |
30 | BIO_push(b64,bmem); | 30 | BIO_push(b64,bmem); |
31 | if(BIO_write(b64,data,length)!=length) | 31 | if(((size_t)BIO_write(b64,data,length))!=length) |
32 | throw exception_openssl(OPKELE_CP_ "failed to BIO_write()"); | 32 | throw exception_openssl(OPKELE_CP_ "failed to BIO_write()"); |
33 | BIO_flush(b64); | 33 | if(BIO_flush(b64)!=1) |
34 | throw exception_openssl(OPKELE_CP_ "failed to BIO_flush()"); | ||
34 | char *rvd; | 35 | char *rvd; |
35 | long rvl = BIO_get_mem_data(bmem,&rvd); | 36 | long rvl = BIO_get_mem_data(bmem,&rvd); |
36 | string rv(rvd,rvl); | 37 | string rv(rvd,rvl); |
37 | BIO_free_all(b64); | 38 | BIO_free_all(b64); |
38 | return rv; | 39 | return rv; |
39 | }catch(...) { | 40 | }catch(...) { |
40 | if(b64) BIO_free_all(b64); | 41 | if(b64) BIO_free_all(b64); |
41 | throw; | 42 | throw; |
42 | } | 43 | } |
43 | } | 44 | } |
44 | 45 | ||
45 | void decode_base64(const string& data,vector<unsigned char>& rv) { | 46 | void decode_base64(const string& data,vector<unsigned char>& rv) { |
46 | BIO *b64 = 0, *bmem = 0; | 47 | BIO *b64 = 0, *bmem = 0; |
47 | rv.clear(); | 48 | rv.clear(); |
48 | try { | 49 | try { |
49 | bmem = BIO_new_mem_buf((void*)data.data(),data.size()); | 50 | bmem = BIO_new_mem_buf((void*)data.data(),data.size()); |
50 | if(!bmem) | 51 | if(!bmem) |
51 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new_mem_buf()"); | 52 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new_mem_buf()"); |
52 | b64 = BIO_new(BIO_f_base64()); | 53 | b64 = BIO_new(BIO_f_base64()); |
53 | if(!b64) | 54 | if(!b64) |
54 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new() base64 decoder"); | 55 | throw exception_openssl(OPKELE_CP_ "failed to BIO_new() base64 decoder"); |
55 | BIO_set_flags(b64,BIO_FLAGS_BASE64_NO_NL); | 56 | BIO_set_flags(b64,BIO_FLAGS_BASE64_NO_NL); |
56 | BIO_push(b64,bmem); | 57 | BIO_push(b64,bmem); |
57 | unsigned char tmp[512]; | 58 | unsigned char tmp[512]; |
@@ -123,40 +124,40 @@ namespace opkele { | |||
123 | tm_t.tm_mon--; | 124 | tm_t.tm_mon--; |
124 | tm_t.tm_year-=1900; | 125 | tm_t.tm_year-=1900; |
125 | time_t rv = mktime(&tm_t); | 126 | time_t rv = mktime(&tm_t); |
126 | if(rv==(time_t)-1) | 127 | if(rv==(time_t)-1) |
127 | throw failed_conversion(OPKELE_CP_ "failed to mktime()"); | 128 | throw failed_conversion(OPKELE_CP_ "failed to mktime()"); |
128 | return rv; | 129 | return rv; |
129 | } | 130 | } |
130 | 131 | ||
131 | /* | 132 | /* |
132 | * | 133 | * |
133 | */ | 134 | */ |
134 | 135 | ||
135 | string url_encode(const string& str) { | 136 | string url_encode(const string& str) { |
136 | char * t = curl_escape(str.c_str(),str.length()); | 137 | char * t = curl_escape(str.c_str(),str.length()); |
137 | if(!t) | 138 | if(!t) |
138 | throw failed_conversion(OPKELE_CP_ "failed to curl_escape()"); | 139 | throw failed_conversion(OPKELE_CP_ "failed to curl_escape()"); |
139 | string rv(t); | 140 | string rv(t); |
140 | curl_free(t); | 141 | curl_free(t); |
141 | return rv; | 142 | return rv; |
142 | } | 143 | } |
143 | 144 | ||
144 | string long_to_string(long l) { | 145 | string long_to_string(long l) { |
145 | char rv[32]; | 146 | char rv[32]; |
146 | int r=snprintf(rv,sizeof(rv),"%ld",l); | 147 | int r=snprintf(rv,sizeof(rv),"%ld",l); |
147 | if(r<0 || r>=sizeof(rv)) | 148 | if(r<0 || r>=(int)sizeof(rv)) |
148 | throw failed_conversion(OPKELE_CP_ "failed to snprintf()"); | 149 | throw failed_conversion(OPKELE_CP_ "failed to snprintf()"); |
149 | return rv; | 150 | return rv; |
150 | } | 151 | } |
151 | 152 | ||
152 | long string_to_long(const string& s) { | 153 | long string_to_long(const string& s) { |
153 | char *endptr = 0; | 154 | char *endptr = 0; |
154 | long rv = strtol(s.c_str(),&endptr,10); | 155 | long rv = strtol(s.c_str(),&endptr,10); |
155 | if((!endptr) || endptr==s.c_str()) | 156 | if((!endptr) || endptr==s.c_str()) |
156 | throw failed_conversion(OPKELE_CP_ "failed to strtol()"); | 157 | throw failed_conversion(OPKELE_CP_ "failed to strtol()"); |
157 | return rv; | 158 | return rv; |
158 | } | 159 | } |
159 | 160 | ||
160 | } | 161 | } |
161 | 162 | ||
162 | } | 163 | } |