File: test_utils.c

package info (click to toggle)
pcp 7.1.0-1
  • links: PTS
  • area: main
  • in suites: forky, sid
  • size: 252,748 kB
  • sloc: ansic: 1,483,656; sh: 182,366; xml: 160,462; cpp: 83,813; python: 24,980; perl: 18,327; yacc: 6,877; lex: 2,864; makefile: 2,738; awk: 165; fortran: 60; java: 52
file content (65 lines) | stat: -rw-r--r-- 1,874 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#include "test_utils.h"

#include "cluster.h"

#include <stdlib.h>
#include <string.h>

static int server_version_major;
static int server_version_minor;

/* Helper to extract version information. */
#define VALKEY_VERSION_FIELD "valkey_version:"
#define REDIS_VERSION_FIELD "redis_version:"
void load_valkey_version(valkeyClusterContext *cc) {
    valkeyClusterNodeIterator ni;
    valkeyClusterNode *node;
    char *eptr, *s, *e;
    valkeyReply *reply = NULL;

    valkeyClusterInitNodeIterator(&ni, cc);
    if ((node = valkeyClusterNodeNext(&ni)) == NULL)
        goto abort;

    reply = valkeyClusterCommandToNode(cc, node, "INFO");
    if (reply == NULL || cc->err || reply->type != VALKEY_REPLY_STRING)
        goto abort;
    if ((s = strstr(reply->str, VALKEY_VERSION_FIELD)) != NULL)
        s += strlen(VALKEY_VERSION_FIELD);
    else if ((s = strstr(reply->str, REDIS_VERSION_FIELD)) != NULL)
        s += strlen(REDIS_VERSION_FIELD);
    else
        goto abort;

    /* We need a field terminator and at least 'x.y.z' (5) bytes of data */
    if ((e = strstr(s, "\r\n")) == NULL || (e - s) < 5)
        goto abort;

    /* Extract version info */
    server_version_major = strtol(s, &eptr, 10);
    if (*eptr != '.')
        goto abort;
    server_version_minor = strtol(eptr + 1, NULL, 10);

    freeReplyObject(reply);
    return;

abort:
    freeReplyObject(reply);
    fprintf(stderr, "Error: Cannot get Valkey version, aborting..\n");
    exit(1);
}

/* Helper to verify Valkey version information. */
int valkey_version_less_than(int major, int minor) {
    if (server_version_major == 0) {
        fprintf(stderr, "Error: Valkey version not loaded, aborting..\n");
        exit(1);
    }

    if (server_version_major < major)
        return 1;
    if (server_version_major == major && server_version_minor < minor)
        return 1;
    return 0;
}