summaryrefslogtreecommitdiffabout
Unidiff
Diffstat (more/less context) (ignore whitespace changes)
-rw-r--r--src/eyetil.cc28
-rw-r--r--src/eyetil.h11
2 files changed, 39 insertions, 0 deletions
diff --git a/src/eyetil.cc b/src/eyetil.cc
index cd22eea..6ccc4ae 100644
--- a/src/eyetil.cc
+++ b/src/eyetil.cc
@@ -89,6 +89,34 @@ uint16_t block512_t::tcpcksum(block512_t& data) {
89 return 0xffff&~sum; 89 return 0xffff&~sum;
90} 90}
91 91
92void integrity_digester::update(const void *d_,size_t s) {
93 uint8_t *d=(uint8_t*)d_;
94 if(data_size) {
95 int l = sizeof(data)-data_size;
96 if(l>s) {
97 memmove(data.dptr(data_size),d,s); data_size+=s; return;
98 }
99 memmove(data.dptr(data_size),d,l); d+=l; s-=l;
100 md5.update<uint16_t>( data.tcpcksum(data) );
101 }
102 if(s<sizeof(data)) {
103 memmove(data.dptr(0),d,s); data_size=s; return;
104 }
105 size_t bb=s/sizeof(block512_t);
106 std::transform((block512_t*)d,((block512_t*)d)+bb,
107 md5.updater<uint16_t>(),block512_t::tcpcksum);
108 size_t ss=bb*sizeof(block512_t);
109 d+=ss; s-=ss;
110 assert(s<sizeof(block512_t));
111 if(s) memmove(data.dptr(0),d,data_size=s);
112}
113
114binary_t integrity_digester::final(const std::string& ukey) {
115 assert(!data_size);
116 md5.update( binary_t(ukey) );
117 return md5.final();
118}
119
92static void make_path_for_template(const std::string& p,mode_t m) { 120static void make_path_for_template(const std::string& p,mode_t m) {
93 struct stat st; 121 struct stat st;
94 std::string pp; 122 std::string pp;
diff --git a/src/eyetil.h b/src/eyetil.h
index 7517ba6..8784cb4 100644
--- a/src/eyetil.h
+++ b/src/eyetil.h
@@ -74,6 +74,17 @@ struct block512_t {
74}; 74};
75#pragma pack() 75#pragma pack()
76 76
77struct integrity_digester {
78 md5_digester md5;
79 size_t data_size;
80 block512_t data;
81
82 integrity_digester() : data_size(0) { }
83 void update(const void *d,size_t s);
84 binary_t final(const std::string& ukey);
85};
86
87
77class tmpdir_t { 88class tmpdir_t {
78 public: 89 public:
79 std::string dir; 90 std::string dir;