开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
SysConfig.cpp
资源名称：p2p_vod.rar　[点击查看]
上传用户：liguizhu
上传日期：2015-11-01
资源大小：2422k
文件大小：9k
源码类别：

P2P编程

开发平台：

Visual C++

SysConfig.cpp：源码内容
  1. /*
  2.  *  Openmysee
  3.  *
  4.  *  This program is free software; you can redistribute it and/or modify
  5.  *  it under the terms of the GNU General Public License as published by
  6.  *  the Free Software Foundation; either version 2 of the License, or
  7.  *  (at your option) any later version.
  8.  *
  9.  *  This program is distributed in the hope that it will be useful,
  10.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.  *  GNU General Public License for more details.
  13.  *
  14.  *  You should have received a copy of the GNU General Public License
  15.  *  along with this program; if not, write to the Free Software
  16.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  *
  18.  */
  19. //
  21. #include "stdafx.h"
  22. #include "GVCapture.h"
  23. #include "SysConfig.h"
  24. #include "ConfigFile.h"
  25. #include "md5.h"
  26. #include ".sysconfig.h"
  27. #include "TestNetDlg.h"
  29. #ifdef _DEBUG
  30. #define new DEBUG_NEW
  31. #undef THIS_FILE
  32. static char THIS_FILE[] = __FILE__;
  33. #endif
  35. /////////////////////////////////////////////////////////////////////////////
  36. // CSysConfig dialog
  39. CSysConfig::CSysConfig(CWnd* pParent /*=NULL*/)
  40. : CDialog(CSysConfig::IDD, pParent)
  41. {
  42. //{{AFX_DATA_INIT(CSysConfig)
  43. mstrChannelName = _T("");
  44. mstrPassword = _T("");
  45. mstrSPAddress = _T("");
  46. miUserID = 0;
  47. // mStrMMSAddr = _T("");
  48. miReconnectTime = 0;
  49. //}}AFX_DATA_INIT
  50. }
  53. void CSysConfig::DoDataExchange(CDataExchange* pDX)
  54. {
  55. CDialog::DoDataExchange(pDX);
  56. //{{AFX_DATA_MAP(CSysConfig)
  57. DDX_Text(pDX, IDC_CHANNELNAME, mstrChannelName);
  58. DDV_MaxChars(pDX, mstrChannelName, 20);
  59. DDX_Text(pDX, IDC_PASSWORD, mstrPassword);
  60. DDV_MaxChars(pDX, mstrPassword, 40);
  61. DDX_Text(pDX, IDC_SPADDRESS, mstrSPAddress);
  62. DDX_Text(pDX, IDC_USERID, miUserID);
  63. DDX_Text(pDX, IDC_RE_TIME, miReconnectTime);
  65. //}}AFX_DATA_MAP
  66. }
  69. BEGIN_MESSAGE_MAP(CSysConfig, CDialog)
  70. //{{AFX_MSG_MAP(CSysConfig)
  71. //}}AFX_MSG_MAP
  72. ON_BN_CLICKED(IDC_TEST_NET, OnBnClickedTestNet)
  73. ON_BN_CLICKED(IDOK, OnBnClickedOk)
  74. END_MESSAGE_MAP()
  76. /////////////////////////////////////////////////////////////////////////////
  77. // CSysConfig message handlers
  79. void CSysConfig::OnOK() 
  80. {
  81. UpdateData();
  83. if (!JudgeChannelName(mstrChannelName.GetBuffer(0)))
  84. {
  85. return ;
  86. }
  88. //该种加密是不可呢的，不能解密。
  89. MD5 loMd5((unsigned char*)mstrPassword.GetBuffer(0),strlen(mstrPassword.GetBuffer(0)));
  90. mstrPassword = loMd5.hex_digest();
  92. if (!JudgePassword(mstrPassword.GetBuffer(0)))
  93. {
  94. return ;
  95. }
  97. //保存为ini文件
  98. char lszEXEPath[MAX_PATH];
  99. GetCurrentCfgFile(lszEXEPath);
  101. SaveSysConfig(lszEXEPath);
  103. CDialog::OnOK();
  104. }
  107. //得到当前配置文件路径
  108. bool CSysConfig::GetCurrentCfgFile(char *aszFileName)
  109. {
  110. GetModuleFileName(NULL, aszFileName, MAX_PATH);
  112. int liLengthPath = strlen(aszFileName);
  114. while (aszFileName[liLengthPath - 1] != '\')
  115. {
  116. --liLengthPath;
  117. }
  119. if (liLengthPath > 0)
  120. {
  121. aszFileName[liLengthPath] = '';
  122. strcat(aszFileName, "CaptureServer.cfg");
  123. return true;
  124. }
  125. return false;
  126. }
  128. bool CSysConfig::SaveSysConfig(char *aszFileName)
  129. {
  130. //将所得的变量转换
  131. char lszChannelName[DEF_CHANNELNAME_LENGTH + 1] = {0};
  132. char lszPassword[DEF_KEY_LENGTH + 1] = {0};
  133. char lszSPAddress[DEF_ADDRESS_LENGTH + 1] = {0};
  134. //char lszmStrMMSAddr[DEF_ADDRESS_LENGTH + 1] = {0};
  136. strncpy(lszChannelName, mstrChannelName, DEF_CHANNELNAME_LENGTH);
  137. strncpy(lszPassword, mstrPassword, DEF_KEY_LENGTH);
  138. strncpy(lszSPAddress, mstrSPAddress, DEF_ADDRESS_LENGTH);
  139. // strncpy(lszmStrMMSAddr ,mStrMMSAddr, DEF_ADDRESS_LENGTH);
  141. int liWriteLen = 0;
  142. char lszTemp[DEF_TEMP_LENGTH + 1] = {0}; //用来定义临时的拷贝buf
  144. FILE* lfpFile = NULL;
  145. lfpFile = fopen(aszFileName, "w");
  147. if(!lfpFile)
  148. {
  149. MessageBox("写文件出错，可能该文件不可读","错误");
  150. return false;
  151. }
  153. //组成字符串文件头和长度
  154. strcpy(lszTemp, "[CaptureServer]nspAddress=");
  155. strcat(lszTemp, lszSPAddress);
  156. strcat(lszTemp,"n");
  157. strcat(lszTemp,"password=");
  158. strcat(lszTemp,lszPassword);
  159. strcat(lszTemp,"n");
  160. strcat(lszTemp,"channelName=");
  161. strcat(lszTemp,lszChannelName);
  162. strcat(lszTemp,"n");
  163. //strcat(lszTemp,"MMSAddress=");
  164. //strcat(lszTemp,lszmStrMMSAddr);
  165. strcat(lszTemp,"n");
  167. liWriteLen = fwrite(lszTemp, sizeof(char), strlen(lszTemp), lfpFile);
  168. if (liWriteLen != (int)strlen(lszTemp))
  169. {
  170. return false;
  171. }
  173. //写用户ID
  174. ZeroMemory(lszTemp, sizeof(DEF_TEMP_LENGTH));
  175. sprintf(lszTemp,"userID=%dn", miUserID);
  177. liWriteLen = fwrite(lszTemp, sizeof(char), strlen(lszTemp), lfpFile);
  178. if (liWriteLen != (int)strlen(lszTemp))
  179. {
  180. return false;
  181. }
  183. //写重新连接时间
  184. ZeroMemory(lszTemp, sizeof(DEF_TEMP_LENGTH));
  185. sprintf(lszTemp,"ReconnectTime=%dn", miReconnectTime);
  187. liWriteLen = (int)fwrite(lszTemp, sizeof(char), strlen(lszTemp), lfpFile);
  188. if (liWriteLen != (int)strlen(lszTemp))
  189. {
  190. return false;
  191. }
  194. if (0 != fclose(lfpFile))
  195. {
  196. return false;
  197. }
  198. return true;
  200. }
  203. BOOL CSysConfig::OnInitDialog() 
  204. {
  205. CDialog::OnInitDialog();
  207. char lszEXEPath[MAX_PATH] = {0};
  209. GetCurrentCfgFile(lszEXEPath);
  211. ConfigFile cfgFile(lszEXEPath);
  212. if(cfgFile.fileNotFound) 
  213. {
  214. MessageBox("无法找到配置文件CaptureServer.cfg.", "错误", MB_OK|MB_ICONERROR);
  215. return FALSE;
  216. }
  218. mstrSPAddress= cfgFile.Value("CaptureServer", "spAddress").c_str();
  220. if(cfgFile.stringNotFound) {
  221. MessageBox("配置文件中必须存在spAddress一项。", "错误", MB_OK|MB_ICONERROR);
  222. return FALSE;
  223. }
  224. mstrChannelName = cfgFile.Value("CaptureServer", "channelName").c_str();
  225. if(cfgFile.stringNotFound) {
  226. MessageBox( "配置文件中必须存在channelName一项。", "错误", MB_OK|MB_ICONERROR);
  227. return FALSE;
  228. }
  230. string reconnect = cfgFile.Value("CaptureServer", "ReconnectTime").c_str();
  231. if(!cfgFile.fileNotFound)
  232. {
  233. miReconnectTime = atoi(reconnect.data());
  234. }
  236. string id = cfgFile.Value("CaptureServer","userID");
  237. if(cfgFile.fileNotFound){
  238. MessageBox( "配置文件中必须存在userID一项。", "错误", MB_OK|MB_ICONERROR);
  239. return FALSE;
  240. }
  241. miUserID = atoi(id.data());
  243. //读入的是32位密码
  244. mstrPassword = cfgFile.Value("CaptureServer", "password").c_str();
  245. if(cfgFile.fileNotFound){
  246. MessageBox("配置文件中必须存在password一项目。", "错误", MB_OK|MB_ICONERROR);
  247. return FALSE;
  248. }
  249. /*
  250. //读入MMSAddress
  251. mStrMMSAddr = cfgFile.Value("CaptureServer", "MMSAddress").c_str();
  252. if(cfgFile.fileNotFound)
  253. {
  254. MessageBox( "配置文件中必须存在MMSAddress一项目。", "错误", MB_OK|MB_ICONERROR);
  255. return FALSE;
  256. }
  257. */
  258. UpdateData(false);
  259. return TRUE;  // return TRUE unless you set the focus to a control
  260.               // EXCEPTION: OCX Property Pages should return FALSE
  261. }
  263. bool CSysConfig::JudgeChannelName(string astrChannelName)
  264. {
  265. string::const_iterator it = astrChannelName.begin();
  266. BYTE tmpChar;
  267. for(; it != astrChannelName.end(); it++) 
  268. {
  269. tmpChar = static_cast<BYTE>(*it);
  270. if(isdigit(tmpChar) || isalpha(tmpChar))
  271. continue;
  272. if(tmpChar == '[' || tmpChar == ']')
  273. continue;
  274. //GB18030-2000编码标准的第一个字节的编码范围在0x81 ~ 0xFE之间；第二个字节的编码范围在0x40 ~ 0x7E和0x80 ~ 0xFE之间。
  275. //Big5编码标准的第一个字节的编码范围在0xA1 ~ 0xF9 之间；第二个字节的编码范围在0x40 ~ 0x7E 和0xA1 ~ 0xFE之间。
  276. //两者的并集是第一个字节[0x81, 0xfe]，第二个字节[0x40, 0x7e]、[0x80, 0xfe];
  277. if(tmpChar>=0x81 && tmpChar <= 0xfe && it+1 != astrChannelName.end()) {
  278. tmpChar = static_cast<BYTE>(*(it+1));
  279. if( (tmpChar>=0x40 && tmpChar <= 0x7e) || (tmpChar>=0x80 && tmpChar <= 0xfe))
  280. {
  281. it++; // 跳过一个汉字字符
  282. continue;
  283. }
  284. }
  285. MessageBox("配置文件中channelName一项只能由字母、数字、汉字和方括弧组成。", "错误", MB_OK|MB_ICONERROR);
  286. return false;
  287. }
  288. return true;
  289. }
  291. //判断ip正确性
  292. bool CSysConfig::JudgeIP(string astrIP)
  293. {
  294. string::const_iterator it = astrIP.begin();
  295. BYTE tmpChar;
  296. for(; it != astrIP.end(); it++) 
  297. {
  298. tmpChar = static_cast<BYTE>(*it);
  299. if(isdigit(tmpChar) ||':' == tmpChar || '.' == tmpChar)
  300. continue;
  301. MessageBox("配置文件中服务器地址一项只能由数字和分隔符“:”组成。", "错误", MB_OK|MB_ICONERROR);
  302. return false;
  303. }
  304. return true;
  305. }
  307. bool CSysConfig::JudgePassword(string astrPassword)
  308. {
  309. if(astrPassword.length() != 32) 
  310. {
  311. MessageBox("加密过程出错,配置文件中Password一项md5加密后其长度必须等于32字符。", "错误", MB_OK|MB_ICONERROR);
  312. return false;
  313. }
  314. return true;
  315. }
  317. void CSysConfig::OnBnClickedTestNet()
  318. {
  319. UpdateData();
  320. if (false == JudgeIP((LPCTSTR)mstrSPAddress))
  321. {
  322. return;
  323. }
  324. CTestNetDlg loTestNet;
  325. loTestNet.GetAddr(mstrSPAddress);
  326. loTestNet.DoModal();
  327. }
  329. void CSysConfig::OnBnClickedOk()
  330. {
  331. // TODO: Add your control notification handler code here
  332. OnOK();
  333. }