1. #include "stdafx.h"
2. #include <sstream>
4. #include "desc.h"
5. #include "desc_manager.h"
6. #include "char.h"
7. #include "buffer_manager.h"
8. #include "config.h"
9. #include "profiler.h"
10. #include "p2p.h"
11. #include "log.h"
12. #include "db.h"
13. #include "questmanager.h"
14. #include "login_sim.h"
15. #include "fishing.h"
16. #include "TrafficProfiler.h"
17. #include "priv_manager.h"
18. #include "castle.h"
19. #include "dev_log.h"
21. extern time_t get_global_time();
23. bool IsEmptyAdminPage()
24. {
25. return g_stAdminPageIP.empty();
26. }
28. bool IsAdminPage(const char * ip)
29. {
30. for (size_t n = 0; n < g_stAdminPageIP.size(); ++n)
31. {
32. if (g_stAdminPageIP[n] == ip)
33. return 1;
34. }
35. return 0;
36. }
38. void ClearAdminPages()
39. {
40. for (size_t n = 0; n < g_stAdminPageIP.size(); ++n)
41. g_stAdminPageIP[n].clear();
43. g_stAdminPageIP.clear();
44. }
46. CInputProcessor::CInputProcessor() : m_pPacketInfo(NULL), m_iBufferLeft(0)
47. {
48. if (!m_pPacketInfo)
49. BindPacketInfo(&m_packetInfoCG);
50. }
52. void CInputProcessor::BindPacketInfo(CPacketInfo * pPacketInfo)
53. {
54. m_pPacketInfo = pPacketInfo;
55. }
57. bool CInputProcessor::Process(LPDESC lpDesc, const void * c_pvOrig, int iBytes, int & r_iBytesProceed)
58. {
59. const char * c_pData = (const char *) c_pvOrig;
61. BYTE bLastHeader = 0;
62. int iLastPacketLen = 0;
63. int iPacketLen;
65. if (!m_pPacketInfo)
66. {
67. sys_err("No packet info has been binded to");
68. return true;
69. }
71. for (m_iBufferLeft = iBytes; m_iBufferLeft > 0;)
72. {
73. BYTE bHeader = (BYTE) *(c_pData);
74. const char * c_pszName;
76. if (bHeader == 0) // ¾Ïȣȭ 󸮰¡ ÀÖÀ¸¹Ç·Î 0¹ø Çì´õ´Â ½ºÅµÇÑ´Ù.
77. iPacketLen = 1;
78. else if (!m_pPacketInfo->Get(bHeader, &iPacketLen, &c_pszName))
79. {
80. sys_err("UNKNOWN HEADER: %d, LAST HEADER: %d(%d), REMAIN BYTES: %d, fd: %d",
81. bHeader, bLastHeader, iLastPacketLen, m_iBufferLeft, lpDesc->GetSocket());
82. //printdata((BYTE *) c_pvOrig, m_iBufferLeft);
83. lpDesc->SetPhase(PHASE_CLOSE);
84. return true;
85. }
87. if (m_iBufferLeft < iPacketLen)
88. return true;
90. if (bHeader)
91. {
92. if (test_server && bHeader != HEADER_CG_MOVE)
93. sys_log(0, "Packet Analyze [Header %d][bufferLeft %d] ", bHeader, m_iBufferLeft);
95. m_pPacketInfo->Start();
97. int iExtraPacketSize = Analyze(lpDesc, bHeader, c_pData);
99. if (iExtraPacketSize < 0)
100. return true;
102. iPacketLen += iExtraPacketSize;
103. lpDesc->Log("%s %d", c_pszName, iPacketLen);
104. m_pPacketInfo->End();
105. }
107. // TRAFFIC_PROFILER
108. if (g_bTrafficProfileOn)
109. TrafficProfiler::instance().Report(TrafficProfiler::IODIR_INPUT, bHeader, iPacketLen);
110. // END_OF_TRAFFIC_PROFILER
112. if (bHeader == HEADER_CG_PONG)
113. sys_log(0, "PONG! %u %u", m_pPacketInfo->IsSequence(bHeader), *(BYTE *) (c_pData + iPacketLen - sizeof(BYTE)));
115. c_pData += iPacketLen;
116. m_iBufferLeft -= iPacketLen;
117. r_iBytesProceed += iPacketLen;
119. iLastPacketLen = iPacketLen;
120. bLastHeader = bHeader;
122. if (GetType() != lpDesc->GetInputProcessor()->GetType())
123. return false;
124. }
126. return true;
127. }
129. void CInputProcessor::Pong(LPDESC d)
130. {
131. d->SetPong(true);
132. }
134. void CInputProcessor::Handshake(LPDESC d, const char * c_pData)
135. {
136. TPacketCGHandshake * p = (TPacketCGHandshake *) c_pData;
138. if (d->GetHandshake() != p->dwHandshake)
139. {
140. sys_err("Invalid Handshake on %d", d->GetSocket());
141. d->SetPhase(PHASE_CLOSE);
142. }
143. else
144. {
145. if (d->IsPhase(PHASE_HANDSHAKE))
146. {
147. if (d->HandshakeProcess(p->dwTime, p->lDelta, false))
148. {
149. #ifdef _IMPROVED_PACKET_ENCRYPTION_
150. d->SendKeyAgreement();
151. #else
152. if (g_bAuthServer)
153. d->SetPhase(PHASE_AUTH);
154. else
155. d->SetPhase(PHASE_LOGIN);
156. #endif // #ifdef _IMPROVED_PACKET_ENCRYPTION_
157. }
158. }
159. else
160. d->HandshakeProcess(p->dwTime, p->lDelta, true);
161. }
162. }
164. void CInputProcessor::Version(LPCHARACTER ch, const char* c_pData)
165. {
166. if (!ch)
167. return;
169. TPacketCGClientVersion * p = (TPacketCGClientVersion *) c_pData;
170. sys_log(0, "VERSION: %s %s %s", ch->GetName(), p->timestamp, p->filename);
171. ch->GetDesc()->SetClientVersion(p->timestamp);
172. }
174. void LoginFailure(LPDESC d, const char * c_pszStatus)
175. {
176. if (!d)
177. return;
179. TPacketGCLoginFailure failurePacket;
181. failurePacket.header = HEADER_GC_LOGIN_FAILURE;
182. strlcpy(failurePacket.szStatus, c_pszStatus, sizeof(failurePacket.szStatus));
184. d->Packet(&failurePacket, sizeof(failurePacket));
185. }
187. CInputHandshake::CInputHandshake()
188. {
189. CPacketInfoCG * pkPacketInfo = M2_NEW CPacketInfoCG;
190. pkPacketInfo->SetSequence(HEADER_CG_PONG, false);
192. m_pMainPacketInfo = m_pPacketInfo;
193. BindPacketInfo(pkPacketInfo);
194. }
196. CInputHandshake::~CInputHandshake()
197. {
198. if( NULL != m_pPacketInfo )
199. {
200. M2_DELETE(m_pPacketInfo);
201. m_pPacketInfo = NULL;
202. }
203. }
206. std::map<DWORD, CLoginSim *> g_sim;
207. std::map<DWORD, CLoginSim *> g_simByPID;
208. std::vector<TPlayerTable> g_vec_save;
210. // BLOCK_CHAT
211. ACMD(do_block_chat);
212. // END_OF_BLOCK_CHAT
214. int CInputHandshake::Analyze(LPDESC d, BYTE bHeader, const char * c_pData)
215. {
216. if (bHeader == 10) // ¿£ÅÍ´Â ¹«½Ã
217. return 0;
219. if (bHeader == HEADER_CG_TEXT)
220. {
221. if (IsEmptyAdminPage() || !IsAdminPage(inet_ntoa(d->GetAddr().sin_addr)))
222. {
223. sys_log(0, "SOCKET_CMD: BLOCK FROM(%s)", d->GetHostName());
224. return -1;
225. }
227. ++c_pData;
228. const char * c_pSep;
230. if (!(c_pSep = strchr(c_pData, '\n'))) // \nÀ» ã´Â´Ù.
231. return -1;
233. if (*(c_pSep - 1) == '\r')
234. --c_pSep;
236. std::string stResult;
237. std::string stBuf;
238. stBuf.assign(c_pData, 0, c_pSep - c_pData);
240. sys_log(0, "SOCKET_CMD: FROM(%s) CMD(%s)", d->GetHostName(), stBuf.c_str());
242. if (!stBuf.compare("IS_SERVER_UP"))
243. {
244. if (g_bNoMoreClient)
245. stResult = "NO";
246. else
247. stResult = "YES";
248. }
249. //else if (!stBuf.compare("SHOWMETHEMONEY"))
250. else if (stBuf == g_stAdminPagePassword)
251. {
252. if (!IsEmptyAdminPage())
253. {
254. if (!IsAdminPage(inet_ntoa(d->GetAddr().sin_addr)))
255. {
256. char szTmp[64];
257. snprintf(szTmp, sizeof(szTmp), "WEBADMIN : Wrong Connector : %s", inet_ntoa(d->GetAddr().sin_addr));
258. stResult += szTmp;
259. }
260. else
261. {
262. d->SetAdminMode();
263. stResult = "UNKNOWN";
264. }
265. }
266. else
267. {
268. d->SetAdminMode();
269. stResult = "UNKNOWN";
270. }
271. }
272. else if (!stBuf.compare("USER_COUNT"))
273. {
274. char szTmp[64];
276. if (!IsEmptyAdminPage())
277. {
278. if (!IsAdminPage(inet_ntoa(d->GetAddr().sin_addr)))
279. {
280. snprintf(szTmp, sizeof(szTmp), "WEBADMIN : Wrong Connector : %s", inet_ntoa(d->GetAddr().sin_addr));
281. }
282. else
283. {
284. int iTotal;
285. int * paiEmpireUserCount;
286. int iLocal;
287. DESC_MANAGER::instance().GetUserCount(iTotal, &paiEmpireUserCount, iLocal);
288. snprintf(szTmp, sizeof(szTmp), "%d %d %d %d %d", iTotal, paiEmpireUserCount[1], paiEmpireUserCount[2], paiEmpireUserCount[3], iLocal);
289. }
290. }
291. else
292. {
293. int iTotal;
294. int * paiEmpireUserCount;
295. int iLocal;
296. DESC_MANAGER::instance().GetUserCount(iTotal, &paiEmpireUserCount, iLocal);
297. snprintf(szTmp, sizeof(szTmp), "%d %d %d %d %d", iTotal, paiEmpireUserCount[1], paiEmpireUserCount[2], paiEmpireUserCount[3], iLocal);
298. }
299. stResult += szTmp;
300. }
301. else if (!stBuf.compare("CHECK_P2P_CONNECTIONS"))
302. {
303. std::ostringstream oss(std::ostringstream::out);
305. oss << "P2P CONNECTION NUMBER : " << P2P_MANAGER::instance().GetDescCount() << "\n";
306. std::string hostNames;
307. P2P_MANAGER::Instance().GetP2PHostNames(hostNames);
308. oss << hostNames;
309. stResult = oss.str();
310. TPacketGGCheckAwakeness packet;
311. packet.bHeader = HEADER_GG_CHECK_AWAKENESS;
313. P2P_MANAGER::instance().Send(&packet, sizeof(packet));
314. }
315. else if (!stBuf.compare("PACKET_INFO"))
316. {
317. m_pMainPacketInfo->Log("packet_info.txt");
318. stResult = "OK";
319. }
320. else if (!stBuf.compare("PROFILE"))
321. {
322. CProfiler::instance().Log("profile.txt");
323. stResult = "OK";
324. }
325. //gift notify delete command
326. else if (!stBuf.compare(0,15,"DELETE_AWARDID "))
327. {
328. char szTmp[64];
329. std::string msg = stBuf.substr(15,26); // item_awardÀÇ id¹üÀ§?
331. TPacketDeleteAwardID p;
332. p.dwID = (DWORD)(atoi(msg.c_str()));
333. snprintf(szTmp,sizeof(szTmp),"Sent to DB cache to delete ItemAward, id: %d",p.dwID);
334. //sys_log(0,"%d",p.dwID);
335. // strlcpy(p.login, msg.c_str(), sizeof(p.login));
336. db_clientdesc->DBPacket(HEADER_GD_DELETE_AWARDID, 0, &p, sizeof(p));
337. stResult += szTmp;
338. }
339. else
340. {
341. stResult = "UNKNOWN";
343. if (d->IsAdminMode())
344. {
345. // ¾îµå¹Î ¸í·Éµé
346. if (!stBuf.compare(0, 7, "NOTICE "))
347. {
348. std::string msg = stBuf.substr(7, 50);
349. LogManager::instance().CharLog(0, 0, 0, 1, "NOTICE", msg.c_str(), d->GetHostName());
350. BroadcastNotice(msg.c_str());
351. }
352. else if (!stBuf.compare("SHUTDOWN"))
353. {
354. LogManager::instance().CharLog(0, 0, 0, 2, "SHUTDOWN", "", d->GetHostName());
355. TPacketGGShutdown p;
356. p.bHeader = HEADER_GG_SHUTDOWN;
357. P2P_MANAGER::instance().Send(&p, sizeof(TPacketGGShutdown));
358. sys_err("Accept shutdown command from %s.", d->GetHostName());
359. Shutdown(10);
360. }
361. else if (!stBuf.compare("SHUTDOWN_ONLY"))
362. {
363. LogManager::instance().CharLog(0, 0, 0, 2, "SHUTDOWN", "", d->GetHostName());
364. sys_err("Accept shutdown only command from %s.", d->GetHostName());
365. Shutdown(10);
366. }
367. else if (!stBuf.compare(0, 3, "DC "))
368. {
369. std::string msg = stBuf.substr(3, LOGIN_MAX_LEN);
371. dev_log(LOG_DEB0, "DC : '%s'", msg.c_str());
373. TPacketGGDisconnect pgg;
375. pgg.bHeader = HEADER_GG_DISCONNECT;
376. strlcpy(pgg.szLogin, msg.c_str(), sizeof(pgg.szLogin));
378. P2P_MANAGER::instance().Send(&pgg, sizeof(TPacketGGDisconnect));
380. // delete login key
381. {
382. TPacketDC p;
383. strlcpy(p.login, msg.c_str(), sizeof(p.login));
384. db_clientdesc->DBPacket(HEADER_GD_DC, 0, &p, sizeof(p));
385. }
386. }
387. else if (!stBuf.compare(0, 10, "RELOAD_CRC"))
388. {
389. LoadValidCRCList();
391. BYTE bHeader = HEADER_GG_RELOAD_CRC_LIST;
392. P2P_MANAGER::instance().Send(&bHeader, sizeof(BYTE));
393. stResult = "OK";
394. }
395. else if (!stBuf.compare(0, 20, "CHECK_CLIENT_VERSION"))
396. {
397. CheckClientVersion();
399. BYTE bHeader = HEADER_GG_CHECK_CLIENT_VERSION;
400. P2P_MANAGER::instance().Send(&bHeader, sizeof(BYTE));
401. stResult = "OK";
402. }
403. else if (!stBuf.compare(0, 6, "RELOAD"))
404. {
405. if (stBuf.size() == 6)
406. {
407. LoadStateUserCount();
408. db_clientdesc->DBPacket(HEADER_GD_RELOAD_PROTO, 0, NULL, 0);
409. DBManager::instance().LoadDBString();
410. }
411. else
412. {
413. char c = stBuf[7];
415. switch (LOWER(c))
416. {
417. case 'u':
418. LoadStateUserCount();
419. break;
421. case 'p':
422. db_clientdesc->DBPacket(HEADER_GD_RELOAD_PROTO, 0, NULL, 0);
423. break;
425. case 's':
426. DBManager::instance().LoadDBString();
427. break;
429. case 'q':
430. quest::CQuestManager::instance().Reload();
431. break;
433. case 'f':
434. fishing::Initialize();
435. break;
437. case 'a':
438. db_clientdesc->DBPacket(HEADER_GD_RELOAD_ADMIN, 0, NULL, 0);
439. sys_log(0, "Reloading admin infomation.");
440. break;
441. }
442. }
443. }
444. else if (!stBuf.compare(0, 6, "EVENT "))
445. {
446. std::istringstream is(stBuf);
447. std::string strEvent, strFlagName;
448. long lValue;
449. is >> strEvent >> strFlagName >> lValue;
451. if (!is.fail())
452. {
453. sys_log(0, "EXTERNAL EVENT FLAG name %s value %d", strFlagName.c_str(), lValue);
454. quest::CQuestManager::instance().RequestSetEventFlag(strFlagName, lValue);
455. stResult = "EVENT FLAG CHANGE ";
456. stResult += strFlagName;
457. }
458. else
459. {
460. stResult = "EVENT FLAG FAIL";
461. }
462. }
463. // BLOCK_CHAT
464. else if (!stBuf.compare(0, 11, "BLOCK_CHAT "))
465. {
466. std::istringstream is(stBuf);
467. std::string strBlockChat, strCharName;
468. long lDuration;
469. is >> strBlockChat >> strCharName >> lDuration;
471. if (!is.fail())
472. {
473. sys_log(0, "EXTERNAL BLOCK_CHAT name %s duration %d", strCharName.c_str(), lDuration);
475. do_block_chat(NULL, const_cast<char*>(stBuf.c_str() + 11), 0, 0);
477. stResult = "BLOCK_CHAT ";
478. stResult += strCharName;
479. }
480. else
481. {
482. stResult = "BLOCK_CHAT FAIL";
483. }
484. }
485. // END_OF_BLOCK_CHAT
486. else if (!stBuf.compare(0, 12, "PRIV_EMPIRE "))
487. {
488. int empire, type, value, duration;
489. std::istringstream is(stBuf);
490. std::string strPrivEmpire;
491. is >> strPrivEmpire >> empire >> type >> value >> duration;
493. // ÃÖ´ëÄ¡ 10¹è
494. value = MINMAX(0, value, 1000);
495. stResult = "PRIV_EMPIRE FAIL";
497. if (!is.fail())
498. {
499. // check parameter
500. if (empire < 0 || 3 < empire);
501. else if (type < 1 || 4 < type);
502. else if (value < 0);
503. else if (duration < 0);
504. else
505. {
506. stResult = "PRIV_EMPIRE SUCCEED";
508. // ½Ã°£ ´ÜÀ§·Î º¯°æ
509. duration = duration * (60 * 60);
511. sys_log(0, "_give_empire_privileage(empire=%d, type=%d, value=%d, duration=%d) by web",
512. empire, type, value, duration);
513. CPrivManager::instance().RequestGiveEmpirePriv(empire, type, value, duration);
514. }
515. }
516. }
517. }
518. }
520. sys_log(1, "TEXT %s RESULT %s", stBuf.c_str(), stResult.c_str());
521. stResult += "\n";
522. d->Packet(stResult.c_str(), stResult.length());
523. return (c_pSep - c_pData) + 1;
524. }
525. else if (bHeader == HEADER_CG_MARK_LOGIN)
526. {
527. if (!guild_mark_server)
528. {
529. // ²÷¾î¹ö·Á! - ¸¶Å© ¼­¹ö°¡ ¾Æ´Ñµ¥ ¸¶Å©¸¦ ¿äûÇÏ·Á°í?
530. sys_err("Guild Mark login requested but i'm not a mark server!");
531. d->SetPhase(PHASE_CLOSE);
532. return 0;
533. }
535. // ¹«Á¶°Ç ÀÎÁõ --;
536. sys_log(0, "MARK_SERVER: Login");
537. d->SetPhase(PHASE_LOGIN);
538. return 0;
539. }
540. else if (bHeader == HEADER_CG_STATE_CHECKER)
541. {
542. if (d->isChannelStatusRequested()) {
543. return 0;
544. }
545. d->SetChannelStatusRequested(true);
546. db_clientdesc->DBPacket(HEADER_GD_REQUEST_CHANNELSTATUS, d->GetHandle(), NULL, 0);
547. }
548. else if (bHeader == HEADER_CG_PONG)
549. Pong(d);
550. else if (bHeader == HEADER_CG_HANDSHAKE)
551. Handshake(d, c_pData);
552. #ifdef _IMPROVED_PACKET_ENCRYPTION_
553. else if (bHeader == HEADER_CG_KEY_AGREEMENT)
554. {
555. // Send out the key agreement completion packet first
556. // to help client to enter encryption mode
557. d->SendKeyAgreementCompleted();
558. // Flush socket output before going encrypted
559. d->ProcessOutput();
561. TPacketKeyAgreement* p = (TPacketKeyAgreement*)c_pData;
562. if (!d->IsCipherPrepared())
563. {
564. sys_err ("Cipher isn't prepared. %s maybe a Hacker.", inet_ntoa(d->GetAddr().sin_addr));
565. d->DelayedDisconnect(5);
566. return 0;
567. }
568. if (d->FinishHandshake(p->wAgreedLength, p->data, p->wDataLength)) {
569. // Handshaking succeeded
570. if (g_bAuthServer) {
571. d->SetPhase(PHASE_AUTH);
572. } else {
573. d->SetPhase(PHASE_LOGIN);
574. }
575. } else {
576. sys_log(0, "[CInputHandshake] Key agreement failed: al=%u dl=%u",
577. p->wAgreedLength, p->wDataLength);
578. d->SetPhase(PHASE_CLOSE);
579. }
580. }
581. #endif // _IMPROVED_PACKET_ENCRYPTION_
582. else
583. sys_err("Handshake phase does not handle packet %d (fd %d)", bHeader, d->GetSocket());
585. return 0;
586. }

INPUT.CPP