MINOR: server: Be more strict when reading the version of a server-state file
Now, we read a full line and expects to found an integer only on it. And if
the line is empty or truncated, an error is returned. If the version is not
valid, an error is also returned. This way, the first line is no longer
partially read.
diff --git a/src/server.c b/src/server.c
index bcfdbb2..b9fa115 100644
--- a/src/server.c
+++ b/src/server.c
@@ -3049,20 +3049,26 @@
* Note that this should be the first read on <f>
*/
static int srv_state_get_version(FILE *f) {
- char buf[2];
- int ret;
+ char mybuf[SRV_STATE_LINE_MAXLEN];
+ char *endptr;
+ long int vsn;
/* first character of first line of the file must contain the version of the export */
- if (fgets(buf, 2, f) == NULL) {
+ if (fgets(mybuf, SRV_STATE_LINE_MAXLEN, f) == NULL)
+ return 0;
+
+ vsn = strtol(mybuf, &endptr, 10);
+ if (endptr == mybuf || *endptr != '\n') {
+ /* Empty or truncated line */
return 0;
}
- ret = atoi(buf);
- if ((ret < SRV_STATE_FILE_VERSION_MIN) ||
- (ret > SRV_STATE_FILE_VERSION_MAX))
+ if (vsn < SRV_STATE_FILE_VERSION_MIN || vsn > SRV_STATE_FILE_VERSION_MAX) {
+ /* Wrong version number */
return 0;
+ }
- return ret;
+ return vsn;
}