UpModule/ExistingModInfo.cpp at main · KongQBin/UpModule · 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
#include "ExistingModInfo.h"

ExistingModInfo::ExistingModInfo()
{
    if(!uname(&m_system_info))
    {
        printf("System Name: %s\n", m_system_info.sysname);
        printf("Node Name: %s\n", m_system_info.nodename);
        printf("Release: %s\n", m_system_info.release);
        printf("Version: %s\n", m_system_info.version);
        printf("Machine: %s\n", m_system_info.machine);
#ifdef _GNU_SOURCE
        printf("Domain Name: %s\n", m_system_info.domainname);
#endif
    }
    queryKMods();
    initAllSymbolVerInfo();
}

ExistingModInfo::~ExistingModInfo()
{

}

void ExistingModInfo::printKMods()
{
    for(auto it = m_kmods.begin(); it != m_kmods.end(); ++it)
    {
        printf("%-32s : %s\n",it->first.c_str(),it->second.c_str());
//        cout << it->first << " : " << it->second << endl;
    }
    return;
}

string ExistingModInfo::getVermagic()
{
    return m_vermagic;
}

unsigned long ExistingModInfo::findSym(string symname)
{
    unsigned long ret = 0;
    auto it = m_va_infos.find(symname);
    if(it != m_va_infos.end()) ret = it->second;
    return ret;
}

void ExistingModInfo::queryKMods()
{
    list<string> md;
    string modulespath = "/usr/lib/modules/";
    modulespath += (string(m_system_info.release) + "/");
    ifstream modules(modulespath+string("modules.dep"));
    if (modules.is_open()) {
        string line;
        while (getline(modules, line)) {
            // 分割出驱动和依赖
            list<string> mod_dep = splitString(line,':');
            if(mod_dep.size() > 1)  // 如果有依赖，那么进一步进行分割
            {
                list<string> tmp;
                tmp.push_back(mod_dep.front());
                for(auto it = ++mod_dep.begin(); it != mod_dep.end(); ++it)
                {
                    list<string> n = splitString(*it,' ');
                    tmp.insert(tmp.end(),n.begin(),n.end());
                }
                mod_dep = tmp;
            }
            md.insert(md.end(),mod_dep.begin(),mod_dep.end());
        }
        modules.close(); // 关闭文件
    } else cerr << "Unable to open file." << endl;

    md.sort();
    for(auto it = md.begin(); it != md.end(); ++it)
    {
        auto it2 = next(it);
        if(*it == *it2) continue;
        string path = modulespath + *it;
        if(access(path.c_str(),F_OK)) continue;
        list<string> plist = splitString(path,'/');
        m_kmods.insert(make_pair(plist.back(),path));
    }
}

void ExistingModInfo::initAllSymbolVerInfo()
{
    for(auto it = m_kmods.begin(); it != m_kmods.end(); ++it)
    {
        ElfSymInfos infos(it->second,false,false);
        // 获取版本魔术
        string newVermagic = infos.getVermagic();
        if(m_vermagic.empty())
            m_vermagic = newVermagic;
        else if(m_vermagic != newVermagic)
        {
            printf("Warning : old magic = %s new magic = %s\n",
                   m_vermagic.c_str(),newVermagic.c_str());
        }
        else {}
        // 获取现有驱动的所有符号的CRC信息
        map<string, unsigned long> tmp(move(infos.getSymVerInfo()));
        for(auto it2 = tmp.begin(); it2 != tmp.end(); ++it2)
        {
            auto index = m_va_infos.find(it2->first);
            if(index != m_va_infos.end())
            {
                // 发现冲突的CRC校验码
                if(index->second != it2->second)
                {
                    printf("Has repetitive crc name is %-32s %lu and %lu\n",it2->first,it2->second,index->second);
                    m_va_infos.insert(make_pair(it2->first,it2->second));
                }
            }
            else
                m_va_infos.insert(make_pair(it2->first,it2->second));
        }
    }
    printf("init %s symbol size = %d\n",m_va_infos.size() ? "success" : "fail", m_va_infos.size());
//    if(m_va_infos.size())
//    {
//        for(auto it : m_va_infos)
//        {
//            printf("\tname = %-32s crc = %lx\n",it.first.c_str(),it.second);
//        }
//    }
}