Solsmith/Variable.cpp at master · HighBe/Solsmith · 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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#include "Variable.h"
using namespace std;

//确定生成的变量的name
string getName(int n) {
    string res;
    if (n / 26 == 1) {
        res += "a";
    }
    res += ('a' + n % 26);
    return res;
}
//因为solidity的uint256,int256是2的255次方，c++中无法用数字表示，只能直接生成字符串。生成的数都是正数，负数额外添加负号。
//2的255次方可以用64位16进制数来表示。

string getUInt256() {

    // string res = "0x";
    //随机生成的数字
    // for(int i = 0;i < 5;i ++){
    //     res += '0';
    // }
    // for (int i = 5; i < 64; i++) {
    //     res += '0' + rand() % 10;
    // }
    string res = to_string(rand() % 255);
    return res;
}
string getInt256() {

    // string res = "0x";
    // //随机生成的数字
    // for(int i = 0;i < 5;i ++){
    //     res += '0';
    // }
    // for (int i = 5; i < 64; i++) {
    //     res += '0' + rand() % 10;
    // }
    string res = to_string(rand() % 255);
    return res;
}

string getAdress() {

    string res = "0x";
    //随机生成的数字，后n位是有效的，前64-n全是0。
    int n = rand() % 40;
    for (int i = 0; i < 40 - n; i++) {
        res += '0';
    }
    for (int i = n; i < 40; i++) {
        res += '0' + rand() % 10;
    }
    return res;
}


string genVariable(vector<Variable>& VarList, int i, bool isLocal) {
    string res;//记录生成的代码。
    string value;
    int typec = rand() % 2;
    string tmpname = getName(i);
    //如果是局部变量在名称前加一个l_的前缀
    if (isLocal) {
        tmpname = "l_" + tmpname;
    }
    switch (typec)
    {
    case 0:
        value = getUInt256();
        VarList.emplace_back(Variable("uint256", tmpname, stoi(value)));
        //生成文本
        res += "\n\tuint256 ";
        if (!isLocal) res += " public ";
        res += tmpname + " = " + value;

        break;
    case 1:
        value = getInt256();
        //如果是0，生成负数，否则为正数
        res += "\n\tint256 ";
        if (!isLocal) res += " public ";
        res += tmpname + " = ";
        if (rand() % 2 == 0) value = "-" + value;
        res += getInt256();
        VarList.emplace_back(Variable("int256", tmpname, stoi(value)));
        break;
    case 2:
        VarList.emplace_back(Variable("bool", tmpname, 2));
        if (rand() % 2 == 0) res += "\n\tbool " + tmpname + " = " + "false";
        else res += "\n\tbool " + tmpname + " = " + "true";
        break;
    case 3://address 使用40个16进制数表示
        VarList.emplace_back(Variable("address", tmpname, 3));
        res += "\n\taddress " + tmpname + " = " + getAdress();
        break;
    case 4:
        VarList.emplace_back(Variable("bytes32", getName(i), 4));
        res += "\n\tbytes32 " + tmpname + " = " + getInt256();
        break;
    }
    res += ";\n";
    return res;
}

string genArray(vector<Variable> &VarList,int i){
    string res = "";
    int tmpl = rand() % 3 + 3;
    res += "\tuint[] Array" + getName(i);

    // 给数组赋初值
    res += " = [";
    for(int i = 0;i < tmpl; ++ i){
        if(i != 0) {
            res += "," + to_string(rand() % 65);
        }else{
            res += to_string(rand() % 65);
        }
    }
    res += "];\n";
    VarList.emplace_back(Variable("uint256","Array"+getName(i),tmpl,0));
    return res;
}

string genConstVariable(vector<Variable> &VarList,int i) {
    string res = "uint256 constant const_" + getName(i) + " = ";
    if (rand() % 2 == 1) {
        res += to_string(rand() % 50) + ";\n";
    }
    else {
        res += "0xfffffffffffffffffffffffffffffffffffffffffffffffe;\n";
    }
    VarList.emplace_back(Variable("uint256", "const_" + getName(i)));
    return res;
}