SmokeDetectorProtocol/main.c at main · Burnside999/SmokeDetectorProtocol · GitHub

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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#include "protocol.h"

#define TEST_ALL

#if defined(__linux__)
#include <sys/resource.h>
size_t get_memory_usage() {
    FILE *file = fopen("/proc/self/status", "r");
    if (!file)
        return 0;

    char line[128];
    size_t memory = 0;
    while (fgets(line, sizeof(line), file)) {
        if (strncmp(line, "VmRSS:", 6) == 0) {
            sscanf(line + 6, "%zu", &memory);
            break;
        }
    }
    fclose(file);
    return memory;
}
#endif

/*
    封装一个读指令
*/
void MakeRead(int op, int address, byte **output, int *len) {
    int length = 6;
    *len = length;
    *output = malloc(length * sizeof(byte));
    (*output)[0] = 0x40;
    (*output)[1] = (byte)length;
    (*output)[2] = 0x00;
    (*output)[3] = (byte)op;
    int sum = 0;
    for (int i = 0; i < length - 2; i++) {
        sum += (byte)(*output)[i];
    }
    (*output)[4] = sum & 0xff;
    (*output)[5] = 0x23;
}

/*
    封装一个写指令
*/
void MakeWrite(int op, int address, byte **output, int *len, int param) {
    int length = 6 + DecodeLength[op];
    *len = length;
    *output = malloc(length * sizeof(byte));
    (*output)[0] = 0x40;
    (*output)[1] = (byte)length;
    (*output)[2] = 0x00;
    (*output)[3] = (byte)op;
    ParseI2B(param, (*output) + 4, DecodeLength[op]);
    int sum = 0;
    for (int i = 0; i < length - 2; i++) {
        sum += (byte)(*output)[i];
    }
    (*output)[length - 2] = sum & 0xff;
    (*output)[length - 1] = 0x23;
}

int param4 = 0; // 参数 4
int temp = 18;  // 环境温度

int main() {
    InitProtocol();
    RegisterCallback(22, BasicWrite, &param4, false); // 写参数 4 的回调
    RegisterCallback(17, BasicRead, &temp, false);    // 读环境温度的回调
    // 模拟测试

    // 测试 1 - 写参数 4
    printf("Original Param4: %d\n", param4);
    byte *Package = NULL;
    int len = 0;
    MakeWrite(22, 0, &Package, &len, 42); // 0 是测试用地址，将 param4 修改为 42
    parse(Package, len);
    printf("After Param4: %d\n", param4); // param4 == 42 即通过测试
    free(Package);
    Package = NULL;

    // 测试 2 - 读环境温度
    MakeRead(17, 0, &Package, &len); // 0 是测试用地址，读取环境温度
    parse(Package, len);             // 回码中输出正确即通过测试，即0x40 0x08 0x00 0x11 0x00
                                     // 0x12 0x6b 0x23
    free(Package);
    Package = NULL;

    // 测试 3 - 内存泄露测试
#if defined(__linux__) && defined(TEST_ALL)
    size_t InitMemory = get_memory_usage();
    MakeWrite(22, 0, &Package, &len, 42);
    for (int i = 0; i < 10000; i++) {
        parse(Package, len);
        printf("Epoch %d, Memory Used: %ld KB.\n", i, get_memory_usage());
    }
    printf("Before: %ld KB.\nAfter:  %ld KB.\n", InitMemory, get_memory_usage());
    free(Package);
    Package = NULL;
#else
    printf("Skip Memory Test.\n");
#endif
    return 0;
}