summaryrefslogtreecommitdiffabout
Unidiff
Diffstat (more/less context) (ignore whitespace changes)
-rw-r--r--lib/util.cc7
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
11namespace opkele { 11namespace 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}