fluencelabs/redis
1
0
Fork 0
mirror of https://github.com/fluencelabs/redis synced 2025-06-23 14:01:34 +00:00
Code Issues Projects Releases Wiki Activity

Add checksum computation to rio.c

Browse Source
This commit is contained in:
antirez
2012-04-09 12:33:09 +02:00
parent 88c1d9550d
commit 736b7c3f04
3 changed files with 43 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/rio.h
View File

@ -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