clusterGetRandomName() generalized into getRandomHexChars() so that we can use it for the run_id field as well.
This commit is contained in:
parent
bab3ce5a09
commit
d05dba2e54
@ -19,20 +19,6 @@ int clusterAddSlot(clusterNode *n, int slot);
|
|||||||
* Initialization
|
* Initialization
|
||||||
* -------------------------------------------------------------------------- */
|
* -------------------------------------------------------------------------- */
|
||||||
|
|
||||||
void clusterGetRandomName(char *p) {
|
|
||||||
FILE *fp = fopen("/dev/urandom","r");
|
|
||||||
char *charset = "0123456789abcdef";
|
|
||||||
int j;
|
|
||||||
|
|
||||||
if (fp == NULL || fread(p,REDIS_CLUSTER_NAMELEN,1,fp) == 0) {
|
|
||||||
for (j = 0; j < REDIS_CLUSTER_NAMELEN; j++)
|
|
||||||
p[j] = rand();
|
|
||||||
}
|
|
||||||
for (j = 0; j < REDIS_CLUSTER_NAMELEN; j++)
|
|
||||||
p[j] = charset[p[j] & 0x0F];
|
|
||||||
fclose(fp);
|
|
||||||
}
|
|
||||||
|
|
||||||
int clusterLoadConfig(char *filename) {
|
int clusterLoadConfig(char *filename) {
|
||||||
FILE *fp = fopen(filename,"r");
|
FILE *fp = fopen(filename,"r");
|
||||||
char *line;
|
char *line;
|
||||||
@ -304,7 +290,7 @@ clusterNode *createClusterNode(char *nodename, int flags) {
|
|||||||
if (nodename)
|
if (nodename)
|
||||||
memcpy(node->name, nodename, REDIS_CLUSTER_NAMELEN);
|
memcpy(node->name, nodename, REDIS_CLUSTER_NAMELEN);
|
||||||
else
|
else
|
||||||
clusterGetRandomName(node->name);
|
getRandomHexChars(node->name, REDIS_CLUSTER_NAMELEN);
|
||||||
node->flags = flags;
|
node->flags = flags;
|
||||||
memset(node->slots,0,sizeof(node->slots));
|
memset(node->slots,0,sizeof(node->slots));
|
||||||
node->numslaves = 0;
|
node->numslaves = 0;
|
||||||
|
@ -58,6 +58,8 @@
|
|||||||
#define REDIS_REPL_TIMEOUT 60
|
#define REDIS_REPL_TIMEOUT 60
|
||||||
#define REDIS_REPL_PING_SLAVE_PERIOD 10
|
#define REDIS_REPL_PING_SLAVE_PERIOD 10
|
||||||
|
|
||||||
|
#define REDIS_RUN_ID_SIZE 40
|
||||||
|
|
||||||
/* Protocol and I/O related defines */
|
/* Protocol and I/O related defines */
|
||||||
#define REDIS_MAX_QUERYBUF_LEN (1024*1024*1024) /* 1GB max query buffer. */
|
#define REDIS_MAX_QUERYBUF_LEN (1024*1024*1024) /* 1GB max query buffer. */
|
||||||
#define REDIS_IOBUF_LEN (1024*16) /* Generic I/O buffer size */
|
#define REDIS_IOBUF_LEN (1024*16) /* Generic I/O buffer size */
|
||||||
@ -813,6 +815,7 @@ dictType hashDictType;
|
|||||||
/* Utils */
|
/* Utils */
|
||||||
long long ustime(void);
|
long long ustime(void);
|
||||||
long long mstime(void);
|
long long mstime(void);
|
||||||
|
void getRandomHexChars(char *p, unsigned int len);
|
||||||
|
|
||||||
/* networking.c -- Networking and Client related operations */
|
/* networking.c -- Networking and Client related operations */
|
||||||
redisClient *createClient(int fd);
|
redisClient *createClient(int fd);
|
||||||
|
48
src/util.c
48
src/util.c
@ -5,6 +5,8 @@
|
|||||||
#include <ctype.h>
|
#include <ctype.h>
|
||||||
#include <limits.h>
|
#include <limits.h>
|
||||||
#include <math.h>
|
#include <math.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <sys/time.h>
|
||||||
|
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
@ -327,6 +329,52 @@ int d2string(char *buf, size_t len, double value) {
|
|||||||
return len;
|
return len;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* Generate the Redis "Run ID", a SHA1-sized random number that identifies a
|
||||||
|
* given execution of Redis, so that if you are talking with an instance
|
||||||
|
* having run_id == A, and you reconnect and it has run_id == B, you can be
|
||||||
|
* sure that it is either a different instance or it was restarted. */
|
||||||
|
void getRandomHexChars(char *p, unsigned int len) {
|
||||||
|
FILE *fp = fopen("/dev/urandom","r");
|
||||||
|
char *charset = "0123456789abcdef";
|
||||||
|
unsigned int j;
|
||||||
|
|
||||||
|
if (fp == NULL || fread(p,len,1,fp) == 0) {
|
||||||
|
/* If we can't read from /dev/urandom, do some reasonable effort
|
||||||
|
* in order to create some entropy, since this function is used to
|
||||||
|
* generate run_id and cluster instance IDs */
|
||||||
|
char *x = p;
|
||||||
|
unsigned int l = len;
|
||||||
|
struct timeval tv;
|
||||||
|
pid_t pid = getpid();
|
||||||
|
|
||||||
|
/* Use time and PID to fill the initial array. */
|
||||||
|
gettimeofday(&tv,NULL);
|
||||||
|
if (l >= sizeof(tv.tv_usec)) {
|
||||||
|
memcpy(x,&tv.tv_usec,sizeof(tv.tv_usec));
|
||||||
|
l -= sizeof(tv.tv_usec);
|
||||||
|
x += sizeof(tv.tv_usec);
|
||||||
|
}
|
||||||
|
if (l >= sizeof(tv.tv_sec)) {
|
||||||
|
memcpy(x,&tv.tv_sec,sizeof(tv.tv_sec));
|
||||||
|
l -= sizeof(tv.tv_sec);
|
||||||
|
x += sizeof(tv.tv_sec);
|
||||||
|
}
|
||||||
|
if (l >= sizeof(pid)) {
|
||||||
|
memcpy(x,&pid,sizeof(pid));
|
||||||
|
l -= sizeof(pid);
|
||||||
|
x += sizeof(pid);
|
||||||
|
}
|
||||||
|
/* Finally xor it with rand() output, that was already seeded with
|
||||||
|
* time() at startup. */
|
||||||
|
for (j = 0; j < len; j++)
|
||||||
|
p[j] ^= rand();
|
||||||
|
}
|
||||||
|
/* Turn it into hex digits taking just 4 bits out of 8 for every byte. */
|
||||||
|
for (j = 0; j < len; j++)
|
||||||
|
p[j] = charset[p[j] & 0x0F];
|
||||||
|
fclose(fp);
|
||||||
|
}
|
||||||
|
|
||||||
#ifdef UTIL_TEST_MAIN
|
#ifdef UTIL_TEST_MAIN
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user