校外远程开机校园网内的电脑

1
#include <winsock2.h>
2
#include <ws2tcpip.h>
3
#include <windows.h>
4
#include <iphlpapi.h>
5
#include <stdio.h>
6
#include <string>
7
#include <vector>
8
#include <sstream>
9

10
#pragma comment(lib, "ws2_32.lib")
11
#pragma comment(lib, "iphlpapi.lib")
12

13
// ============================================
14
// 1. 获取 WiFi 适配器的全局 IPv6 地址
15
// ============================================
16
static std::string GetWiFiIPv6() {
17
    // 使用栈内存，避免堆分配开销
18
    BYTE buffer[15000]{};
19
    ULONG bufferSize = sizeof(buffer);
20
    PIP_ADAPTER_ADDRESSES pAdapter = (PIP_ADAPTER_ADDRESSES)buffer;
21

22
    // 指定 AF_INET6 并跳过无关数据，让系统底层开销最小化
23
    DWORD ret = GetAdaptersAddresses(
24
        AF_INET6,
25
        GAA_FLAG_SKIP_ANYCAST | GAA_FLAG_SKIP_MULTICAST | GAA_FLAG_SKIP_DNS_SERVER,
26
        nullptr,
27
        pAdapter,
28
        &bufferSize
29
    );
30

31
    if (ret != NO_ERROR) return "";
32

33
    for (PIP_ADAPTER_ADDRESSES p = pAdapter; p != nullptr; p = p->Next) {
34
        // 匹配 WiFi 适配器名称
35
        std::wstring name(p->FriendlyName);
36
        if (name.find(L"WiFi") == std::wstring::npos &&
37
            name.find(L"Wireless") == std::wstring::npos &&
38
            name.find(L"WLAN") == std::wstring::npos) {
39
            continue;
40
        }
41

42
        // 遍历该适配器的单播地址
43
        for (PIP_ADAPTER_UNICAST_ADDRESS pAddr = p->FirstUnicastAddress; pAddr != nullptr; pAddr = pAddr->Next) {
44
            sockaddr_in6* sa6 = (sockaddr_in6*)pAddr->Address.lpSockaddr;
45

46
            // 跳过 fe80 开头的链路本地地址，只返回全局 IPv6
47
            if (IN6_IS_ADDR_LINKLOCAL(&sa6->sin6_addr)) {
48
                continue;
49
            }
50

51
            // 找到第一个全局 IPv6 地址，转换并返回
52
            char ipStr[INET6_ADDRSTRLEN];
53
            inet_ntop(AF_INET6, &sa6->sin6_addr, ipStr, sizeof(ipStr));
54
            return std::string(ipStr);
55
        }
56
    }
57
    return "";
58
}
59

60
// ============================================
61
// 2. HTTP 响应构造
62
// ============================================
63
static std::string BuildHttpResponse(const std::string& ipv6) {
64
    std::ostringstream response;
65
    response << "HTTP/1.1 200 OK\r\n"
66
        << "Content-Type: application/json\r\n"
67
        << "Connection: close\r\n"
68
        << "\r\n"
69
        << "{\"wifi_ipv6\":\"" << ipv6 << "\"}\r\n";
70
    return response.str();
71
}
72

73
// ============================================
74
// 3. 提取 HTTP 请求的方法 (Method) 和路径 (Path)
75
// ============================================
76
static bool ParseHttpRequest(const char* buffer, int len, std::string& method, std::string& path) {
77
    if (len < 5) return false; // 至少需要 "GET / "
78

79
    // 1. 提取 Method (到第一个空格为止)
80
    const char* methodStart = buffer;
81
    const char* methodEnd = methodStart;
82
    while (methodEnd < buffer + len && *methodEnd != ' ') methodEnd++;
83
    if (methodEnd >= buffer + len) return false;
84
    method = std::string(methodStart, methodEnd - methodStart);
85

86
    // 2. 提取 Path (Method 后的空格之后，到下一个空格/回车/换行之前)
87
    const char* pathStart = methodEnd + 1;
88
    const char* pathEnd = pathStart;
89
    while (pathEnd < buffer + len && *pathEnd != ' ' && *pathEnd != '\r' && *pathEnd != '\n') pathEnd++;
90
    if (pathEnd <= pathStart) return false;
91
    path = std::string(pathStart, pathEnd - pathStart);
92

93
    return true;
94
}
95

96
// 辅助函数：为当前进程启用关机权限
97
static bool EnableShutdownPrivilege() {
98
    HANDLE hToken;
99
    TOKEN_PRIVILEGES tkp;
100

101
    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) {
102
        return false;
103
    }
104

105
    LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid);
106
    tkp.PrivilegeCount = 1;
107
    tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
108

109
    BOOL result = AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);
110
    CloseHandle(hToken);
111

112
    return (result && GetLastError() == ERROR_SUCCESS);
113
}
114

115
int main() {
116
    // 初始化 Winsock
117
    WSADATA wsaData;
118
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
119
        printf("WSAStartup failed\n");
120
        return 1;
121
    }
122

123
    // 创建 IPv6 Socket（双栈模式）
124
    SOCKET listenSocket = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
125
    if (listenSocket == INVALID_SOCKET) {
126
        printf("socket failed\n");
127
        WSACleanup();
128
        return 1;
129
    }
130

131
    // 禁用 IPV6_V6ONLY，允许 IPv4 映射连接
132
    int v6Only = 0;
133
    setsockopt(listenSocket, IPPROTO_IPV6, IPV6_V6ONLY, (char*)&v6Only, sizeof(v6Only));
134

135
    // 绑定到所有接口的 9001 端口
136
    sockaddr_in6 serverAddr = {};
137
    serverAddr.sin6_family = AF_INET6;
138
    serverAddr.sin6_addr = in6addr_any;
139
    serverAddr.sin6_port = htons(9001);
140

141
    if (bind(listenSocket, (sockaddr*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) {
142
        printf("bind failed: %d\n", WSAGetLastError());
143
        closesocket(listenSocket);
144
        WSACleanup();
145
        return 1;
146
    }
147

148
    if (listen(listenSocket, SOMAXCONN) == SOCKET_ERROR) {
149
        printf("listen failed\n");
150
        closesocket(listenSocket);
151
        WSACleanup();
152
        return 1;
153
    }
154

155
    // 主循环
156
    while (true) {
157
        sockaddr_storage clientAddr = {};
158
        int addrLen = sizeof(clientAddr);
159
        SOCKET clientSocket = accept(listenSocket, (sockaddr*)&clientAddr, &addrLen);
160

161
        if (clientSocket == INVALID_SOCKET) continue;
162

163
        // 设置接收超时为 5 秒 防止客户端连接后不发数据导致 recv 永久阻塞，从而引发 Socket 句柄泄漏
164
        DWORD timeout = 5000;
165
        setsockopt(clientSocket, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(timeout));
166

167
        // 接收请求并提取路径
168
        char buffer[1024]{};
169
        int recvLen = recv(clientSocket, buffer, sizeof(buffer) - 1, 0);
170

171
        if (recvLen > 0) {
172
            std::string method, path;
173
            if (ParseHttpRequest(buffer, recvLen, method, path)) {
174
                // 【路由 1】获取 IPv6 地址
175
                if (method == "GET" && path == "/ipv6") {
176
                    std::string wifiIPv6 = GetWiFiIPv6();
177
                    if (wifiIPv6.empty()) wifiIPv6 = "not_found";
178
                    std::string response = BuildHttpResponse(wifiIPv6);
179
                    send(clientSocket, response.c_str(), (int)response.length(), 0);
180
                }
181
                // 【路由 2】处理 POST /shutdown 关机请求
182
                else if (method == "POST" && path == "/shutdown") {
183
                    const char* resp = "HTTP/1.1 200 OK\r\nContent-Type: application/json\r\nConnection: close\r\n\r\n{\"result\":0}";
184
                    send(clientSocket, resp, (int)strlen(resp), 0);
185
                    // 1. 立即关闭当前客户端连接，释放资源
186
                    closesocket(clientSocket);
187
                    // 2. 尝试赋予关机权限
188
                    EnableShutdownPrivilege();
189
                    // 换回 InitiateSystemShutdownW
190
                    if (!InitiateSystemShutdownW(
191
                        NULL,   // lpMachineName
192
                        NULL,   // lpMessage
193
                        0,      // dwTimeout
194
                        TRUE,   // bForceAppsClosed
195
                        FALSE   // bRebootAfterShutdown
196
                    )) {
197
                        printf("Failed to initiate shutdown! Error: %d\n", GetLastError());
198
                    }
199
                    continue;
200
                }
201
                // 【默认】404 Not Found
202
                else {
203
                    std::string response = "HTTP/1.1 404 Not Found\r\nContent-Length: 0\r\nConnection: close\r\n\r\n";
204
                    send(clientSocket, response.c_str(), (int)response.length(), 0);
205
                }
206
            }
207
        }
208
        // 无论成功、失败还是超时，都确保关闭连接释放资源
209
        closesocket(clientSocket);
210
    }
211

212
    closesocket(listenSocket);
213
    WSACleanup();
214
    return 0;
215
}