-rw-r--r-- | src/eyetil.cc | 9 |
1 files changed, 0 insertions, 9 deletions
diff --git a/src/eyetil.cc b/src/eyetil.cc index 5bceec7..5489d7b 100644 --- a/src/eyetil.cc +++ b/src/eyetil.cc | |||
@@ -1,203 +1,194 @@ | |||
1 | #include <stdlib.h> | 1 | #include <stdlib.h> |
2 | #include <sys/stat.h> | 2 | #include <sys/stat.h> |
3 | #include <syslog.h> | 3 | #include <syslog.h> |
4 | #include <iostream> | 4 | #include <iostream> |
5 | #include <cassert> | 5 | #include <cassert> |
6 | #include <stdexcept> | 6 | #include <stdexcept> |
7 | #include <algorithm> | 7 | #include <algorithm> |
8 | #include <numeric> | 8 | #include <numeric> |
9 | #include <openssl/md5.h> | 9 | #include <openssl/md5.h> |
10 | #include "eyetil.h" | 10 | #include "eyetil.h" |
11 | 11 | ||
12 | #include "config.h" | 12 | #include "config.h" |
13 | #ifdef HAVE_LIBUUID | 13 | #ifdef HAVE_LIBUUID |
14 | # include <uuid/uuid.h> | 14 | # include <uuid/uuid.h> |
15 | #endif | 15 | #endif |
16 | 16 | ||
17 | binary_t& binary_t::from_hex(const std::string& h) { | 17 | binary_t& binary_t::from_hex(const std::string& h) { |
18 | std::string::size_type hs = h.length(); | 18 | std::string::size_type hs = h.length(); |
19 | if(hs&1) | 19 | if(hs&1) |
20 | throw std::runtime_error("odd number of characters in hexadecimal number"); | 20 | throw std::runtime_error("odd number of characters in hexadecimal number"); |
21 | size_t rvs = hs>>1; | 21 | size_t rvs = hs>>1; |
22 | resize(rvs); | 22 | resize(rvs); |
23 | const unsigned char *hp = (const unsigned char*)h.data(); | 23 | const unsigned char *hp = (const unsigned char*)h.data(); |
24 | iterator oi=begin(); | 24 | iterator oi=begin(); |
25 | char t[3] = { 0,0,0 }; | 25 | char t[3] = { 0,0,0 }; |
26 | for(size_t i=0;i<rvs;++i) { | 26 | for(size_t i=0;i<rvs;++i) { |
27 | t[0]=*(hp++); t[1]=*(hp++); | 27 | t[0]=*(hp++); t[1]=*(hp++); |
28 | *(oi++) = static_cast<binary_t::value_type>(0xff&strtol(t,0,16)); | 28 | *(oi++) = static_cast<binary_t::value_type>(0xff&strtol(t,0,16)); |
29 | } | 29 | } |
30 | return *this; | 30 | return *this; |
31 | } | 31 | } |
32 | 32 | ||
33 | binary_t& binary_t::from_data(const void *d,size_t s) { | 33 | binary_t& binary_t::from_data(const void *d,size_t s) { |
34 | resize(s); | 34 | resize(s); |
35 | std::copy((const unsigned char*)d,(const unsigned char *)d+s, | 35 | std::copy((const unsigned char*)d,(const unsigned char *)d+s, |
36 | begin() ); | 36 | begin() ); |
37 | return *this; | 37 | return *this; |
38 | } | 38 | } |
39 | 39 | ||
40 | binary_t& binary_t::make_nonce() { | 40 | binary_t& binary_t::make_nonce() { |
41 | #ifdef HAVE_LIBUUID | 41 | #ifdef HAVE_LIBUUID |
42 | uuid_t uuid; | 42 | uuid_t uuid; |
43 | uuid_generate(uuid); | 43 | uuid_generate(uuid); |
44 | from_data((unsigned char*)uuid,sizeof(uuid)); | 44 | from_data((unsigned char*)uuid,sizeof(uuid)); |
45 | #else | 45 | #else |
46 | resize(16); | 46 | resize(16); |
47 | std::generate_n(begin(),16,rand); | 47 | std::generate_n(begin(),16,rand); |
48 | #endif /* HAVE_LIBUUID */ | 48 | #endif /* HAVE_LIBUUID */ |
49 | return *this; | 49 | return *this; |
50 | } | 50 | } |
51 | 51 | ||
52 | std::string binary_t::hex() const { | 52 | std::string binary_t::hex() const { |
53 | std::string rv; | 53 | std::string rv; |
54 | rv.reserve((size()<<1)+1); | 54 | rv.reserve((size()<<1)+1); |
55 | char t[3] = {0,0,0}; | 55 | char t[3] = {0,0,0}; |
56 | for(const_iterator i=begin(),ie=end();i!=ie;++i) { | 56 | for(const_iterator i=begin(),ie=end();i!=ie;++i) { |
57 | size_t rc = snprintf(t,sizeof(t),"%02x",*i); | 57 | size_t rc = snprintf(t,sizeof(t),"%02x",*i); |
58 | assert(rc<sizeof(t)); | 58 | assert(rc<sizeof(t)); |
59 | rv += t; | 59 | rv += t; |
60 | } | 60 | } |
61 | return rv; | 61 | return rv; |
62 | } | 62 | } |
63 | 63 | ||
64 | binary_t binary_t::md5() const { | 64 | binary_t binary_t::md5() const { |
65 | binary_t rv(MD5_DIGEST_LENGTH); | 65 | binary_t rv(MD5_DIGEST_LENGTH); |
66 | if(!MD5( | 66 | if(!MD5( |
67 | (const unsigned char*)&(front()),size(), | 67 | (const unsigned char*)&(front()),size(), |
68 | (unsigned char*)&(rv.front()) )) | 68 | (unsigned char*)&(rv.front()) )) |
69 | throw std::runtime_error("failed to md5()"); | 69 | throw std::runtime_error("failed to md5()"); |
70 | return rv; | 70 | return rv; |
71 | } | 71 | } |
72 | 72 | ||
73 | void md5_digester::init() { | 73 | void md5_digester::init() { |
74 | if(!MD5_Init(&ctx)) throw std::runtime_error("failed to MD5_Init()"); | 74 | if(!MD5_Init(&ctx)) throw std::runtime_error("failed to MD5_Init()"); |
75 | } | 75 | } |
76 | void md5_digester::update(const void *d,size_t l) { | 76 | void md5_digester::update(const void *d,size_t l) { |
77 | if(!MD5_Update(&ctx,d,l)) throw std::runtime_error("failed to MD5_Update()"); | 77 | if(!MD5_Update(&ctx,d,l)) throw std::runtime_error("failed to MD5_Update()"); |
78 | } | 78 | } |
79 | binary_t md5_digester::final() { | 79 | binary_t md5_digester::final() { |
80 | binary_t rv(MD5_DIGEST_LENGTH); | 80 | binary_t rv(MD5_DIGEST_LENGTH); |
81 | if(!MD5_Final((unsigned char*)&(rv.front()), &ctx)) | 81 | if(!MD5_Final((unsigned char*)&(rv.front()), &ctx)) |
82 | throw std::runtime_error("failed to MD5_Final()"); | 82 | throw std::runtime_error("failed to MD5_Final()"); |
83 | return rv; | 83 | return rv; |
84 | } | 84 | } |
85 | 85 | ||
86 | uint16_t block512_t::tcpcksum(block512_t& data) { | 86 | uint16_t block512_t::tcpcksum(block512_t& data) { |
87 | uint32_t sum = std::accumulate(data.data,data.data+words,0); | 87 | uint32_t sum = std::accumulate(data.data,data.data+words,0); |
88 | while(uint32_t hw = sum>>16) sum = (sum&0xffff)+hw; | 88 | while(uint32_t hw = sum>>16) sum = (sum&0xffff)+hw; |
89 | return 0xffff&~sum; | 89 | return 0xffff&~sum; |
90 | } | 90 | } |
91 | 91 | ||
92 | void integrity_digester::update(const void *d_,size_t s) { | 92 | void integrity_digester::update(const void *d_,size_t s) { |
93 | uint8_t *d=(uint8_t*)d_; | 93 | uint8_t *d=(uint8_t*)d_; |
94 | if(data_size) { | 94 | if(data_size) { |
95 | int l = sizeof(data)-data_size; | 95 | int l = sizeof(data)-data_size; |
96 | if(l>s) { | 96 | if(l>s) { |
97 | memmove(data.dptr(data_size),d,s); data_size+=s; return; | 97 | memmove(data.dptr(data_size),d,s); data_size+=s; return; |
98 | } | 98 | } |
99 | memmove(data.dptr(data_size),d,l); d+=l; s-=l; | 99 | memmove(data.dptr(data_size),d,l); d+=l; s-=l; |
100 | md5.update<uint16_t>( data.tcpcksum(data) ); | 100 | md5.update<uint16_t>( data.tcpcksum(data) ); |
101 | } | 101 | } |
102 | if(s<sizeof(data)) { | 102 | if(s<sizeof(data)) { |
103 | memmove(data.dptr(0),d,s); data_size=s; return; | 103 | memmove(data.dptr(0),d,s); data_size=s; return; |
104 | } | 104 | } |
105 | size_t bb=s/sizeof(block512_t); | 105 | size_t bb=s/sizeof(block512_t); |
106 | std::transform((block512_t*)d,((block512_t*)d)+bb, | 106 | std::transform((block512_t*)d,((block512_t*)d)+bb, |
107 | md5.updater<uint16_t>(),block512_t::tcpcksum); | 107 | md5.updater<uint16_t>(),block512_t::tcpcksum); |
108 | size_t ss=bb*sizeof(block512_t); | 108 | size_t ss=bb*sizeof(block512_t); |
109 | d+=ss; s-=ss; | 109 | d+=ss; s-=ss; |
110 | assert(s<sizeof(block512_t)); | 110 | assert(s<sizeof(block512_t)); |
111 | if(s) memmove(data.dptr(0),d,data_size=s); | 111 | if(s) memmove(data.dptr(0),d,data_size=s); |
112 | } | 112 | } |
113 | 113 | ||
114 | binary_t integrity_digester::final(const std::string& ukey) { | 114 | binary_t integrity_digester::final(const std::string& ukey) { |
115 | assert(!data_size); | 115 | assert(!data_size); |
116 | md5.update( binary_t(ukey) ); | 116 | md5.update( binary_t(ukey) ); |
117 | return md5.final(); | 117 | return md5.final(); |
118 | } | 118 | } |
119 | 119 | ||
120 | static void make_path_for_template(const std::string& p,mode_t m) { | 120 | static void make_path_for_template(const std::string& p,mode_t m) { |
121 | struct stat st; | 121 | struct stat st; |
122 | std::string pp; | 122 | std::string pp; |
123 | for(std::string::size_type sl=p.find('/',1); | 123 | for(std::string::size_type sl=p.find('/',1); |
124 | sl!=std::string::npos; | 124 | sl!=std::string::npos; |
125 | sl=p.find('/',sl+1)) { | 125 | sl=p.find('/',sl+1)) { |
126 | if(stat( (pp=p.substr(0,sl)).c_str() ,&st) | 126 | if(stat( (pp=p.substr(0,sl)).c_str() ,&st) |
127 | || !S_ISDIR(st.st_mode)) { | 127 | || !S_ISDIR(st.st_mode)) { |
128 | if(mkdir(pp.c_str(),m)) | 128 | if(mkdir(pp.c_str(),m)) |
129 | throw std::runtime_error("failed to mkdir()"); | 129 | throw std::runtime_error("failed to mkdir()"); |
130 | } | 130 | } |
131 | } | 131 | } |
132 | } | 132 | } |
133 | 133 | ||
134 | tmpdir_t::tmpdir_t(const std::string& dt) : dir(dt) { | 134 | tmpdir_t::tmpdir_t(const std::string& dt) : dir(dt) { |
135 | make_path_for_template(dt,0777); | 135 | make_path_for_template(dt,0777); |
136 | if(!mkdtemp((char*)dir.data())) | 136 | if(!mkdtemp((char*)dir.data())) |
137 | throw std::runtime_error("failed to mkdtmp()"); | 137 | throw std::runtime_error("failed to mkdtmp()"); |
138 | } | 138 | } |
139 | tmpdir_t::~tmpdir_t() { | 139 | tmpdir_t::~tmpdir_t() { |
140 | assert(!dir.empty()); | 140 | assert(!dir.empty()); |
141 | if(rmdir(dir.c_str())) { | 141 | if(rmdir(dir.c_str())) { |
142 | syslog(LOG_WARNING,"Failed to remove '%s' directory",dir.c_str()); | 142 | syslog(LOG_WARNING,"Failed to remove '%s' directory",dir.c_str()); |
143 | } | 143 | } |
144 | } | 144 | } |
145 | 145 | ||
146 | std::string tmpdir_t::get_file(const std::string& f) { | 146 | std::string tmpdir_t::get_file(const std::string& f) { |
147 | std::string::size_type ls = f.rfind('/'); | 147 | std::string::size_type ls = f.rfind('/'); |
148 | return dir+'/'+( | 148 | return dir+'/'+( |
149 | (ls==std::string::npos) | 149 | (ls==std::string::npos) |
150 | ? f | 150 | ? f |
151 | : f.substr(ls+1) | 151 | : f.substr(ls+1) |
152 | ); | 152 | ); |
153 | } | 153 | } |
154 | 154 | ||
155 | tarchive_t::tarchive_t(const char *fn) : a(archive_read_new()), e(0) { | 155 | tarchive_t::tarchive_t(const char *fn) : a(archive_read_new()), e(0) { |
156 | if(!a) throw std::runtime_error("failed to archive_read_new()"); | 156 | if(!a) throw std::runtime_error("failed to archive_read_new()"); |
157 | if(archive_read_support_format_tar(a)) { | 157 | if(archive_read_support_format_tar(a)) { |
158 | archive_read_finish(a); | 158 | archive_read_finish(a); |
159 | throw std::runtime_error("failed to archive_read_support_format_tar()"); | 159 | throw std::runtime_error("failed to archive_read_support_format_tar()"); |
160 | } | 160 | } |
161 | if(archive_read_open_filename(a,fn,16384)) { | 161 | if(archive_read_open_filename(a,fn,16384)) { |
162 | archive_read_finish(a); | 162 | archive_read_finish(a); |
163 | throw std::runtime_error("failed to archive_read_open_memory()"); | 163 | throw std::runtime_error("failed to archive_read_open_memory()"); |
164 | } | 164 | } |
165 | } | 165 | } |
166 | tarchive_t::~tarchive_t() { | 166 | tarchive_t::~tarchive_t() { |
167 | assert(a); | 167 | assert(a); |
168 | archive_read_finish(a); | 168 | archive_read_finish(a); |
169 | } | 169 | } |
170 | 170 | ||
171 | bool tarchive_t::read_next_header() { | 171 | bool tarchive_t::read_next_header() { |
172 | assert(a); | 172 | assert(a); |
173 | return archive_read_next_header(a,&e)==ARCHIVE_OK; | 173 | return archive_read_next_header(a,&e)==ARCHIVE_OK; |
174 | } | 174 | } |
175 | 175 | ||
176 | std::string tarchive_t::entry_pathname() { | 176 | std::string tarchive_t::entry_pathname() { |
177 | assert(a); assert(e); | 177 | assert(a); assert(e); |
178 | return archive_entry_pathname(e); | 178 | return archive_entry_pathname(e); |
179 | } | 179 | } |
180 | 180 | ||
181 | bool tarchive_t::read_data_into_fd(int fd) { | 181 | bool tarchive_t::read_data_into_fd(int fd) { |
182 | assert(a); | 182 | assert(a); |
183 | return archive_read_data_into_fd(a,fd)==ARCHIVE_OK; | 183 | return archive_read_data_into_fd(a,fd)==ARCHIVE_OK; |
184 | } | 184 | } |
185 | 185 | ||
186 | |||
187 | binary_t integrity_digest(const void *ptr,size_t size,const std::string& ukey) { | ||
188 | md5_digester rv; | ||
189 | std::transform( (block512_t*)ptr, ((block512_t*)ptr)+size/sizeof(block512_t), | ||
190 | rv.updater<uint16_t>(), block512_t::tcpcksum ); | ||
191 | rv.update( binary_t(ukey) ); | ||
192 | return rv.final(); | ||
193 | } | ||
194 | |||
195 | mimewrite_tarfile::mimewrite_tarfile(tmpdir_t& d) { | 186 | mimewrite_tarfile::mimewrite_tarfile(tmpdir_t& d) { |
196 | f.open((fn=d.get_file("the-tarfile.tar")).c_str(),std::ios_base::in|std::ios_base::out|std::ios_base::trunc|std::ios_base::binary); | 187 | f.open((fn=d.get_file("the-tarfile.tar")).c_str(),std::ios_base::in|std::ios_base::out|std::ios_base::trunc|std::ios_base::binary); |
197 | } | 188 | } |
198 | mimewrite_tarfile::~mimewrite_tarfile() { | 189 | mimewrite_tarfile::~mimewrite_tarfile() { |
199 | unlink(fn.c_str()); | 190 | unlink(fn.c_str()); |
200 | } | 191 | } |
201 | int mimewrite_tarfile::write(const char *buf,size_t len) { | 192 | int mimewrite_tarfile::write(const char *buf,size_t len) { |
202 | return f.write(buf,len) ? (idigest.update(buf,len),SOAP_OK) : SOAP_ERR; | 193 | return f.write(buf,len) ? (idigest.update(buf,len),SOAP_OK) : SOAP_ERR; |
203 | } | 194 | } |