diff options
author | antirez <antirez@gmail.com> | 2012-04-09 12:33:09 +0200 |
---|---|---|
committer | antirez <antirez@gmail.com> | 2012-04-09 12:33:09 +0200 |
commit | 736b7c3f04ff34a2487559f01e06218ea8afb8be (patch) | |
tree | 332519d87dc774c87f0953c900091e77b3e17902 /src/rio.h | |
parent | 88c1d9550d198fd7df426b19ea67e9c51c92a811 (diff) | |
download | redis-736b7c3f04ff34a2487559f01e06218ea8afb8be.tar.gz |
Add checksum computation to rio.c
Diffstat (limited to 'src/rio.h')
-rw-r--r-- | src/rio.h | 33 |
1 files changed, 31 insertions, 2 deletions
@@ -2,6 +2,7 @@ #define __REDIS_RIO_H #include <stdio.h> +#include <stdint.h> #include "sds.h" struct _rio { @@ -11,6 +12,14 @@ struct _rio { size_t (*read)(struct _rio *, void *buf, size_t len); size_t (*write)(struct _rio *, const void *buf, size_t len); off_t (*tell)(struct _rio *); + /* The update_cksum method if not NULL is used to compute the checksum of all the + * data that was read or written so far. The method should be designed so that + * can be called with the current checksum, and the buf and len fields pointing + * to the new block of data to add to the checksum computation. */ + void (*update_cksum)(struct _rio *, void *buf, size_t len); + + /* The current checksum */ + uint64_t cksum; /* Backend-specific vars. */ union { @@ -26,8 +35,26 @@ struct _rio { typedef struct _rio rio; -#define rioWrite(rio,buf,len) ((rio)->write((rio),(buf),(len))) -#define rioRead(rio,buf,len) ((rio)->read((rio),(buf),(len))) +/* The following functions are our interface with the stream. They'll call the + * actual implementation of read / write / tell, and will update the checksum + * if needed. */ + +inline size_t rioWrite(rio *r, const void *buf, size_t len) { + if (r->udpate_cksum) r->update_cksum(r,buf,len); + return r->write(r,buf,len); +} + +inline size_t rioRead(rio *r, void *buf, size_t len) { + if (r->read(r,buf,len) == 1) { + if (r->udpate_cksum) r->update_cksum(r,buf,len); + return 1; + } + return 0; +} + +inline off_t rioTell(rio *r) { + return r->tell(r); +} void rioInitWithFile(rio *r, FILE *fp); void rioInitWithBuffer(rio *r, sds s); @@ -37,4 +64,6 @@ size_t rioWriteBulkString(rio *r, const char *buf, size_t len); size_t rioWriteBulkLongLong(rio *r, long long l); size_t rioWriteBulkDouble(rio *r, double d); +void rioGenericUpdateChecksum(rio *r, const void *buf, size_t len); + #endif |