Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Newer
Older
100644 187 lines (164 sloc) 4.275 kB
b4bd052 @pietern tool to check if AOF is valid
pietern authored
1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <string.h>
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
4 #include <unistd.h>
b4bd052 @pietern tool to check if AOF is valid
pietern authored
5 #include <sys/stat.h>
6 #include "config.h"
7
8 #define ERROR(...) { \
9 char __buf[1024]; \
10 sprintf(__buf, __VA_ARGS__); \
11 sprintf(error, "0x%08lx: %s", epos, __buf); \
12 }
13
14 static char error[1024];
15 static long epos;
16
17 int consumeNewline(char *buf) {
18 if (strncmp(buf,"\r\n",2) != 0) {
19 ERROR("Expected \\r\\n, got: %02x%02x",buf[0],buf[1]);
20 return 0;
21 }
22 return 1;
23 }
24
25 int readLong(FILE *fp, char prefix, long *target) {
26 char buf[128], *eptr;
27 epos = ftell(fp);
28 if (fgets(buf,sizeof(buf),fp) == NULL) {
29 return 0;
30 }
31 if (buf[0] != prefix) {
32 ERROR("Expected prefix '%c', got: '%c'",buf[0],prefix);
33 return 0;
34 }
35 *target = strtol(buf+1,&eptr,10);
36 return consumeNewline(eptr);
37 }
38
39 int readBytes(FILE *fp, char *target, long length) {
40 long real;
41 epos = ftell(fp);
42 real = fread(target,1,length,fp);
43 if (real != length) {
44 ERROR("Expected to read %ld bytes, got %ld bytes",length,real);
45 return 0;
46 }
47 return 1;
48 }
49
50 int readString(FILE *fp, char** target) {
51 long len;
52 *target = NULL;
53 if (!readLong(fp,'$',&len)) {
54 return 0;
55 }
56
57 /* Increase length to also consume \r\n */
58 len += 2;
59 *target = (char*)malloc(len);
60 if (!readBytes(fp,*target,len)) {
61 free(*target);
62 *target = NULL;
63 return 0;
64 }
65 if (!consumeNewline(*target+len-2)) {
66 free(*target);
67 *target = NULL;
68 return 0;
69 }
70 (*target)[len-2] = '\0';
71 return 1;
72 }
73
74 int readArgc(FILE *fp, long *target) {
75 return readLong(fp,'*',target);
76 }
77
78 long process(FILE *fp) {
79 long argc, pos = 0;
80 int i, multi = 0;
81 char *str;
82
83 while(1) {
84 if (!multi) pos = ftell(fp);
85 if (!readArgc(fp, &argc)) {
86 break;
87 }
88
89 for (i = 0; i < argc; i++) {
90 if (!readString(fp,&str)) {
91 break;
92 }
93 if (i == 0) {
94 if (strcasecmp(str, "multi") == 0) {
95 if (multi++) {
96 ERROR("Unexpected MULTI");
97 break;
98 }
99 } else if (strcasecmp(str, "exec") == 0) {
100 if (--multi) {
101 ERROR("Unexpected EXEC");
102 break;
103 }
104 }
105 }
106 free(str);
107 }
108
109 /* Check if loop was finished */
110 if (i < argc) {
111 if (str) free(str);
112 break;
113 }
114 }
115
116 if (feof(fp) && multi && strlen(error) == 0) {
117 ERROR("Reached EOF before reading EXEC for MULTI");
118 }
119
120 if (strlen(error) > 0) {
121 printf("%s\n", error);
122 }
123
124 return pos;
125 }
126
127 int main(int argc, char **argv) {
128 /* expect the first argument to be the dump file */
129 if (argc <= 1) {
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
130 printf("Usage: %s [--fix] <file.aof>\n", argv[0]);
b4bd052 @pietern tool to check if AOF is valid
pietern authored
131 exit(0);
132 }
133
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
134 char *filename;
135 int fix = 0;
136 if (argc == 3) {
137 if (strcmp(argv[1],"--fix") != 0) {
138 printf("Invalid argument: %s\n", argv[1]);
139 exit(1);
140 }
141 fix = 1;
142 filename = argv[2];
143 } else if (argc == 2) {
144 filename = argv[1];
145 } else {
146 printf("Invalid argument");
147 exit(1);
148 }
149
150 FILE *fp = fopen(filename,"r+");
b4bd052 @pietern tool to check if AOF is valid
pietern authored
151 if (fp == NULL) {
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
152 printf("Cannot open file: %s\n", filename);
b4bd052 @pietern tool to check if AOF is valid
pietern authored
153 exit(1);
154 }
155
156 struct redis_stat sb;
157 if (redis_fstat(fileno(fp),&sb) == -1) {
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
158 printf("Cannot stat file: %s\n", filename);
b4bd052 @pietern tool to check if AOF is valid
pietern authored
159 exit(1);
160 }
161
162 long size = sb.st_size;
163 if (size == 0) {
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
164 printf("Empty file: %s\n", filename);
b4bd052 @pietern tool to check if AOF is valid
pietern authored
165 exit(1);
166 }
167
168 long pos = process(fp);
169 if (pos < size) {
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
170 if (fix) {
171 if (ftruncate(fileno(fp), pos) == -1) {
172 printf("Could not truncate AOF to size %ld\n", pos);
173 exit(1);
174 } else {
175 printf("AOF succesfully truncated to %ld bytes\n", pos);
176 }
177 } else {
178 printf("First invalid operation at offset %ld\n", pos);
179 }
b4bd052 @pietern tool to check if AOF is valid
pietern authored
180 } else {
cb8ae3c @pietern allow AOF to be fixed by truncating to the portion of the file that i…
pietern authored
181 printf("AOF is valid\n");
b4bd052 @pietern tool to check if AOF is valid
pietern authored
182 }
183
184 fclose(fp);
185 return 0;
186 }
Something went wrong with that request. Please try again.