gvsafronov/futriix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Wrapper around fwrite to make API consistent

Browse Source
This commit is contained in:
Pieter Noordhuis 2010-11-21 16:12:25 +01:00
parent f73971ae53
commit 47ddeec2dd
1 changed files with 25 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
src/rdb.c
View File

@ -9,15 +9,22 @@
#include <arpa/inet.h> #include <arpa/inet.h>
#include <sys/stat.h> #include <sys/stat.h>
/* Convenience wrapper around fwrite, that returns the number of bytes written
* to the file instead of the number of objects (see fwrite(3)) and -1 in the
* case of an error. It also supports a NULL *fp to skip writing altogether
* instead of writing to /dev/null. */
static int rdbWriteRaw(FILE *fp, void *p, size_t len) {
if (fp != NULL && fwrite(p,len,1,fp) == 0) return -1;
return len;
}
int rdbSaveType(FILE *fp, unsigned char type) { int rdbSaveType(FILE *fp, unsigned char type) {
if (fwrite(&type,1,1,fp) == 0) return -1; return rdbWriteRaw(fp,&type,1);
return 1; /* bytes written */
} }
int rdbSaveTime(FILE *fp, time_t t) { int rdbSaveTime(FILE *fp, time_t t) {
int32_t t32 = (int32_t) t; int32_t t32 = (int32_t) t;
if (fwrite(&t32,4,1,fp) == 0) return -1; return rdbWriteRaw(fp,&t32,4);
return 4; /* bytes written */
} }
/* check rdbLoadLen() comments for more info */ /* check rdbLoadLen() comments for more info */
@ -28,20 +35,20 @@ int rdbSaveLen(FILE *fp, uint32_t len) {
if (len < (1<<6)) { if (len < (1<<6)) {
/* Save a 6 bit len */ /* Save a 6 bit len */
buf[0] = (len&0xFF)|(REDIS_RDB_6BITLEN<<6); buf[0] = (len&0xFF)|(REDIS_RDB_6BITLEN<<6);
if (fwrite(buf,1,1,fp) == 0) return -1; if (rdbWriteRaw(fp,buf,1) == -1) return -1;
nwritten = 1; nwritten = 1;
} else if (len < (1<<14)) { } else if (len < (1<<14)) {
/* Save a 14 bit len */ /* Save a 14 bit len */
buf[0] = ((len>>8)&0xFF)|(REDIS_RDB_14BITLEN<<6); buf[0] = ((len>>8)&0xFF)|(REDIS_RDB_14BITLEN<<6);
buf[1] = len&0xFF; buf[1] = len&0xFF;
if (fwrite(buf,2,1,fp) == 0) return -1; if (rdbWriteRaw(fp,buf,2) == -1) return -1;
nwritten = 2; nwritten = 2;
} else { } else {
/* Save a 32 bit len */ /* Save a 32 bit len */
buf[0] = (REDIS_RDB_32BITLEN<<6); buf[0] = (REDIS_RDB_32BITLEN<<6);
if (fwrite(buf,1,1,fp) == 0) return -1; if (rdbWriteRaw(fp,buf,1) == -1) return -1;
len = htonl(len); len = htonl(len);
if (fwrite(&len,4,1,fp) == 0) return -1; if (rdbWriteRaw(fp,&len,4) == -1) return -1;
nwritten = 1+4; nwritten = 1+4;
} }
return nwritten; return nwritten;
@ -111,8 +118,8 @@ int rdbSaveLzfStringObject(FILE *fp, unsigned char *s, size_t len) {
} }
/* Data compressed! Let's save it on disk */ /* Data compressed! Let's save it on disk */
byte = (REDIS_RDB_ENCVAL<<6)|REDIS_RDB_ENC_LZF; byte = (REDIS_RDB_ENCVAL<<6)|REDIS_RDB_ENC_LZF;
if (fwrite(&byte,1,1,fp) == 0) goto writeerr; if ((n = rdbWriteRaw(fp,&byte,1)) == -1) goto writeerr;
nwritten += 1; nwritten += n;
if ((n = rdbSaveLen(fp,comprlen)) == -1) goto writeerr; if ((n = rdbSaveLen(fp,comprlen)) == -1) goto writeerr;
nwritten += n; nwritten += n;
@ -120,8 +127,8 @@ int rdbSaveLzfStringObject(FILE *fp, unsigned char *s, size_t len) {
if ((n = rdbSaveLen(fp,len)) == -1) goto writeerr; if ((n = rdbSaveLen(fp,len)) == -1) goto writeerr;
nwritten += n; nwritten += n;
if (fwrite(out,comprlen,1,fp) == 0) goto writeerr; if ((n = rdbWriteRaw(fp,out,comprlen)) == -1) goto writeerr;
nwritten += comprlen; nwritten += n;
zfree(out); zfree(out);
return nwritten; return nwritten;
@ -141,7 +148,7 @@ int rdbSaveRawString(FILE *fp, unsigned char *s, size_t len) {
if (len <= 11) { if (len <= 11) {
unsigned char buf[5]; unsigned char buf[5];
if ((enclen = rdbTryIntegerEncoding((char*)s,len,buf)) > 0) { if ((enclen = rdbTryIntegerEncoding((char*)s,len,buf)) > 0) {
if (fwrite(buf,enclen,1,fp) == 0) return -1; if (rdbWriteRaw(fp,buf,enclen) == -1) return -1;
return enclen; return enclen;
} }
} }
@ -159,7 +166,7 @@ int rdbSaveRawString(FILE *fp, unsigned char *s, size_t len) {
if ((n = rdbSaveLen(fp,len)) == -1) return -1; if ((n = rdbSaveLen(fp,len)) == -1) return -1;
nwritten += n; nwritten += n;
if (len > 0) { if (len > 0) {
if (fwrite(s,len,1,fp) == 0) return -1; if (rdbWriteRaw(fp,s,len) == -1) return -1;
nwritten += len; nwritten += len;
} }
return nwritten; return nwritten;
@ -171,16 +178,15 @@ int rdbSaveLongLongAsStringObject(FILE *fp, long long value) {
int n, nwritten = 0; int n, nwritten = 0;
int enclen = rdbEncodeInteger(value,buf); int enclen = rdbEncodeInteger(value,buf);
if (enclen > 0) { if (enclen > 0) {
if (fwrite(buf,enclen,1,fp) == 0) return -1; return rdbWriteRaw(fp,buf,enclen);
nwritten = enclen;
} else { } else {
/* Encode as string */ /* Encode as string */
enclen = ll2string((char*)buf,32,value); enclen = ll2string((char*)buf,32,value);
redisAssert(enclen < 32); redisAssert(enclen < 32);
if ((n = rdbSaveLen(fp,enclen)) == -1) return -1; if ((n = rdbSaveLen(fp,enclen)) == -1) return -1;
nwritten += n; nwritten += n;
if (fwrite(buf,enclen,1,fp) == 0) return -1; if ((n = rdbWriteRaw(fp,buf,enclen)) == -1) return -1;
nwritten += enclen; nwritten += n;
} }
return nwritten; return nwritten;
} }
@ -236,8 +242,7 @@ int rdbSaveDoubleValue(FILE *fp, double val) {
buf[0] = strlen((char*)buf+1); buf[0] = strlen((char*)buf+1);
len = buf[0]+1; len = buf[0]+1;
} }
if (fwrite(buf,len,1,fp) == 0) return -1; return rdbWriteRaw(fp,buf,len);
return len;
} }
/* Save a Redis object. */ /* Save a Redis object. */