259 lines
6.5 KiB
C
259 lines
6.5 KiB
C
/*
|
||
说明：存放接口及常用结构
|
||
*/
|
||
#ifndef _COMMON_H
|
||
#define _COMMON_H
|
||
|
||
// 确定编译环境
|
||
#define IN_LINUX
|
||
|
||
#include <stdlib.h>
|
||
#include <stdio.h>
|
||
#include <string.h>
|
||
|
||
#ifdef IN_LINUX
|
||
//#include <sys/io.h>
|
||
#include <sys/types.h>
|
||
#include <unistd.h>
|
||
#include <pthread.h>
|
||
#include <sys/socket.h>
|
||
#include <netinet/in.h>
|
||
#include <sys/errno.h>
|
||
#include <sys/stat.h>
|
||
#include <fcntl.h>
|
||
#else
|
||
#include <io.h>
|
||
#include <direct.h>
|
||
#include <windows.h>
|
||
#include <winsock.h>
|
||
#include <basetsd.h>
|
||
#endif
|
||
|
||
#pragma comment(lib,"ws2_32.lib")
|
||
|
||
#define MAX_BUFFER_SIZE 1024
|
||
#define MAX_CMDLINE_LENGTH 200
|
||
#define MAX_THREAD_NUM 40
|
||
#define LENGTH_OF_LISTEN_QUEUE 20
|
||
#define MAX_IP_LENGTH 20
|
||
#define MAX_CMDTYPE_LENGTH 20
|
||
#define MAX_RSP_LENGTH 40
|
||
#define BUFSIZE 65535
|
||
|
||
#define CONFIG "config.ini"
|
||
|
||
//工具的版本
|
||
#define VERSION "1.1.1"
|
||
|
||
//应答消息
|
||
#define RSP_OK "0"
|
||
#define RSP_FAILED "1"
|
||
|
||
#define RET_OK 0
|
||
#define RET_FAILED 1
|
||
|
||
|
||
//支持的命令类型（命令类型的定义字符串不超过 MAX_CMDTYPE_LENGTH 定义的长度）
|
||
#define COPYFILE "copyfile"
|
||
#define CMD "cmd"
|
||
#define CMDWAIT "cmdwait" //等待客户端命令执行完成
|
||
#define SENDFILE "sendfile"
|
||
#define FINDPROC "findproc"
|
||
#define RSP "rsp" //应答消息
|
||
#define CMDECHO "cmdecho"
|
||
|
||
|
||
#define CMD_FALG_WAIT 0
|
||
#define CMD_FALG_NOWAIT 1
|
||
#define CMD_ECHO 2
|
||
|
||
|
||
//服务器端口号，从配置文件中读取
|
||
int g_port;
|
||
|
||
|
||
#ifdef IN_LINUX
|
||
/*---------------------------------
|
||
*说明： 与客户端进行交互
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
void * TalkToClient_linux(void *data);
|
||
|
||
/*---------------------------------
|
||
*说明： 发送数据给客户端
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
buffer ： 需发送给客户端的数据
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int SendDataToClient_linux(int socketID, char *data, int iDataLen);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 接收来自给客户端的数据
|
||
*参数说明： socketID: 已经建立连接的socket句柄
|
||
buffer : 为收到的数据
|
||
*返回值： 0 断开连接
|
||
大于0 收到数据的长度
|
||
-1 失败
|
||
-----------------------------------*/
|
||
int RecvDataFromClient_linux(int socketID, char *buffer, int iBufLen);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 在linux环境下查找指定进程是否存在
|
||
*参数说明：
|
||
*返回值： 0 存在，其他表示不存在
|
||
-----------------------------------*/
|
||
int FindProcess_linux(char *pProcessName);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 初始化服务连接及监听
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int InitServer_linux();
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 开始提供服务
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int StartService_linux(int server_socket);
|
||
|
||
|
||
#else
|
||
/*---------------------------------
|
||
*说明： 与客户端进行交互
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
ULONG __stdcall TalkToClient_windows(void* lpParameter);
|
||
|
||
/*---------------------------------
|
||
*说明： 发送数据给客户端
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
buffer ： 需发送给客户端的数据
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int SendDataToClient_windows(SOCKET socketID, char *data, int iDataLen);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 接收来自给客户端的数据
|
||
*参数说明： socketID: 已经建立连接的socket句柄
|
||
buffer : 为收到的数据，调用者必须手动释放该空间
|
||
*返回值： 0 断开连接
|
||
大于0 收到数据的长度
|
||
-1 失败
|
||
-----------------------------------*/
|
||
int RecvDataFromClient_windows(SOCKET socketID, char *buffer, int iBufLen);
|
||
|
||
/*---------------------------------
|
||
*说明： 在windows环境下查找指定进程是否存在
|
||
*参数说明：
|
||
*返回值： 0 存在，其他表示不存在
|
||
-----------------------------------*/
|
||
int FindProcess_windows(char *pProcessName);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 初始化服务连接及监听
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int InitServer_windows();
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 开始提供服务
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int StartService_windows(SOCKET server_socket);
|
||
|
||
#endif
|
||
|
||
/*---------------------------------
|
||
*说明： 执行拷贝文件到客户端
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
data ： 需要拷贝的文件名（全路径）
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int CopyFileToClient(void* socketID, char *pFile);
|
||
|
||
/*---------------------------------
|
||
*说明： 接收来自客户端的文件
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
file ： 需要保存的文件名（全路径）
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int RecvFileFromClient(void* socketID, char *pFile);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 执行系统命令
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
pCmd ： 执行的命令
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int RunSystemCmd(void* socketID, char *pCmd, int iWaitFlag);
|
||
|
||
/*---------------------------------
|
||
*说明： 解析客户端发送过来的命令
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
data ： 客户端发送过来的命令
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int ParseCmd(void* socketID, char *data);
|
||
|
||
/*---------------------------------
|
||
*说明： 发送应答消息给客户端
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
data ： 发送数据
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int SendRspToClient(void* socketID, char *result);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 等待客户端的应答
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int WaitRspFromClient(void* socketID);
|
||
|
||
/*---------------------------------
|
||
*说明： 解析应答消息
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int ParseRsp(char* strRsp);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 查看指定的程序是否正在运行
|
||
*参数说明： socketID： 已经建立连接的socket句柄
|
||
data ： 进程名
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int FindProcess(void* socketID, char *data);
|
||
|
||
|
||
/*---------------------------------
|
||
*说明： 初始化
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int Init();
|
||
|
||
/*---------------------------------
|
||
*说明： 运行
|
||
*参数说明：
|
||
*返回值： 0成功，否则失败
|
||
-----------------------------------*/
|
||
int Run();
|
||
#endif
|