-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcthandler.cpp
More file actions
110 lines (100 loc) · 3.2 KB
/
Copy pathcthandler.cpp
File metadata and controls
110 lines (100 loc) · 3.2 KB
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
// cthandler.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <WinSock.h>
#include "simpleDll.h"
#include <iostream>
using namespace std;
#pragma comment(lib,"simpleDll.lib")
#pragma comment(lib, "ws2_32.lib")
#pragma comment( linker, "/subsystem:\"windows\" /entry:\"mainCRTStartup\"" ) // 设置连接器选项
//去掉控制台
int InitSocket(string ip,string port)
{
WSADATA wsa;
//初始化套接字DLL
if(WSAStartup(MAKEWORD(2,2),&wsa)!=0)
{
MessageBox(NULL, "网络无响应，请检查网络是否畅通？","",MB_OK);
wlog("CloudTerm\\cthandler.log",true,"socket initialize failed.\n");
return -1;
}
//创建套接字
SOCKET sock;
if((sock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP))==INVALID_SOCKET)
{
MessageBox(NULL, "网络无响应，请检查网络是否畅通？","",MB_OK);
wlog("CloudTerm\\cthandler.log",true,"create socket failed.\n");
return -1;
}
struct sockaddr_in serverAddress;
memset(&serverAddress,0,sizeof(sockaddr_in));
serverAddress.sin_family=AF_INET;
serverAddress.sin_addr.S_un.S_addr = inet_addr(ip.c_str());
serverAddress.sin_port = htons(atoi(port.c_str()));
//建立和服务器的连接
if(connect(sock,(sockaddr*)&serverAddress,sizeof(serverAddress))==SOCKET_ERROR)
{
MessageBox(NULL, "网络无响应，请检查网络是否畅通？","",MB_OK);
wlog("CloudTerm\\cthandler.log",true,"connect failed.\n");
return -1;
}
//printf("Message from %s: %s\n",inet_ntoa(serverAddress.sin_addr),buf);
return sock;
}
int _tmain(int argc, _TCHAR* argv[])
{
if (argc<2)
{
wlog("CloudTerm\\cthandler.log",true,"%s:start arvc wrong\n",argv[0]);
return -1;
}
wlog("CloudTerm\\cthandler.log",true,"%s:start\n",argv[0]);
TCHAR lpchbuf[64]={'\0'};
DWORD dSize=64;
if(!GetUserName(lpchbuf,&dSize))
{
wlog("CloudTerm\\cthandler.log",true,"GetUserNameError %d\n",GetLastError());
return -1;
}
//wlog("CloudTerm\\cthandler.log",true,"username %s\n",lpchbuf);
TCHAR pChPipeName[128]={'\0'};
sprintf(pChPipeName,"\\\\.\\pipe\\CTLIS%s",lpchbuf);
wlog("CloudTerm\\cthandler.log",true,"pchpipename %s\n",pChPipeName);
label: if (WaitNamedPipe(pChPipeName, NMPWAIT_WAIT_FOREVER) == FALSE)
{
wlog("CloudTerm\\cthandler.log",true,"ERROR:waitnamedpipe %d\n",GetLastError());
if(GetLastError()==ERROR_PIPE_BUSY)
{
goto label;
}
else
{
if(GetLastError()==ERROR_FILE_NOT_FOUND)MessageBox(NULL,"请首先登录ct",argv[0],NULL);
return -1;
}
}
HANDLE hPipe = CreateFile(pChPipeName, GENERIC_READ | GENERIC_WRITE,0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hPipe == INVALID_HANDLE_VALUE)
{
wlog("CloudTerm\\cthandler.log",true,"ERROR：createfile %d\n",GetLastError());
if(GetLastError()==ERROR_FILE_NOT_FOUND)
MessageBox(NULL,"请首先登录ct",argv[0],NULL);
return -1;
}
// 向管道写入数据
DWORD WriteNum=0;
memset(lpchbuf,0,64);
sprintf(lpchbuf,"%s",argv[1]);
if (WriteFile(hPipe, lpchbuf, strlen(lpchbuf),
&WriteNum, NULL) == FALSE)
{
wlog("CloudTerm\\cthandler.log",true,"ERROR write file %d\n",GetLastError());
return -1;
} else {
//wlog("CloudTerm\\cthandler.log",true,"write %s\n",lpchbuf);
}
CloseHandle(hPipe); // 关闭管道句柄
wlog("CloudTerm\\cthandler.log",true,"%s:complete\n",argv[0]);
return 0;
}