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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 多媒体编程 > 查看源码
WebServer.cpp
资源名称：bfyy.rar　[点击查看]
上传用户：tangyu_668
上传日期：2014-02-27
资源大小：678k
文件大小：18k
源码类别：

多媒体编程

开发平台：

Visual C++

WebServer.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "mplayerc.h"
  3. #include "resource.h"
  4. #include "MainFrm.h"
  5. #include <atlbase.h>
  6. #include <atlisapi.h>
  7. #include "WebServerSocket.h"
  8. #include "WebClientSocket.h"
  9. #include "WebServer.h"
  10. #include "....zlibzlib.h"
  12. CAtlStringMap<CWebServer::RequestHandler> CWebServer::m_internalpages;
  13. CAtlStringMap<UINT> CWebServer::m_downloads;
  14. CAtlStringMap<CStringA, CStringA> CWebServer::m_mimes;
  16. CWebServer::CWebServer(CMainFrame* pMainFrame, int nPort)
  17. : m_pMainFrame(pMainFrame)
  18. , m_nPort(nPort)
  19. {
  20. if(m_internalpages.IsEmpty())
  21. {
  22. m_internalpages[_T("/")] = &CWebClientSocket::OnIndex;
  23. m_internalpages[_T("/index.html")] = &CWebClientSocket::OnIndex;
  24. m_internalpages[_T("/browser.html")] = &CWebClientSocket::OnBrowser;
  25. m_internalpages[_T("/controls.html")] = &CWebClientSocket::OnControls;
  26. m_internalpages[_T("/command.html")] = &CWebClientSocket::OnCommand;
  27. m_internalpages[_T("/status.html")] = &CWebClientSocket::OnStatus;
  28. m_internalpages[_T("/player.html")] = &CWebClientSocket::OnPlayer;
  29. m_internalpages[_T("/snapshot.jpg")] = &CWebClientSocket::OnSnapShotJpeg;
  30. m_internalpages[_T("/404.html")] = &CWebClientSocket::OnError404;
  31. m_internalpages[_T("/convres.html")] = &CWebClientSocket::OnConvRes;
  32. }
  34. if(m_downloads.IsEmpty())
  35. {
  36. m_downloads[_T("/default.css")] = IDF_DEFAULT_CSS;
  37. m_downloads[_T("/vbg.gif")] = IDF_VBR_GIF;
  38. m_downloads[_T("/vbs.gif")] = IDF_VBS_GIF;
  39. m_downloads[_T("/sliderbar.gif")] = IDF_SLIDERBAR_GIF;
  40. m_downloads[_T("/slidergrip.gif")] = IDF_SLIDERGRIP_GIF;
  41. m_downloads[_T("/sliderback.gif")] = IDF_SLIDERBACK_GIF;
  42. m_downloads[_T("/1pix.gif")] = IDF_1PIX_GIF;
  43. m_downloads[_T("/headericon.png")] = IDF_HEADERICON_PNG;
  44. m_downloads[_T("/headerback.png")] = IDF_HEADERBACK_PNG;
  45. m_downloads[_T("/headerclose.png")] = IDF_HEADERCLOSE_PNG;
  46. m_downloads[_T("/leftside.png")] = IDF_LEFTSIDE_PNG;
  47. m_downloads[_T("/rightside.png")] = IDF_RIGHTSIDE_PNG;
  48. m_downloads[_T("/bottomside.png")] = IDF_BOTTOMSIDE_PNG;
  49. m_downloads[_T("/leftbottomside.png")] = IDF_LEFTBOTTOMSIDE_PNG;
  50. m_downloads[_T("/rightbottomside.png")] = IDF_RIGHTBOTTOMSIDE_PNG;
  51. m_downloads[_T("/seekbarleft.png")] = IDF_SEEKBARLEFT_PNG;
  52. m_downloads[_T("/seekbarmid.png")] = IDF_SEEKBARMID_PNG;
  53. m_downloads[_T("/seekbarright.png")] = IDF_SEEKBARRIGHT_PNG;
  54. m_downloads[_T("/seekbargrip.png")] = IDF_SEEKBARGRIP_PNG;
  55. m_downloads[_T("/logo.png")] = IDF_LOGO7;
  56. m_downloads[_T("/controlback.png")] = IDF_CONTROLBACK_PNG;
  57. m_downloads[_T("/controlbuttonplay.png")] = IDF_CONTROLBUTTONPLAY_PNG;
  58. m_downloads[_T("/controlbuttonpause.png")] = IDF_CONTROLBUTTONPAUSE_PNG;
  59. m_downloads[_T("/controlbuttonstop.png")] = IDF_CONTROLBUTTONSTOP_PNG;
  60. m_downloads[_T("/controlbuttonskipback.png")] = IDF_CONTROLBUTTONSKIPBACK_PNG;
  61. m_downloads[_T("/controlbuttondecrate.png")] = IDF_CONTROLBUTTONDECRATE_PNG;
  62. m_downloads[_T("/controlbuttonincrate.png")] = IDF_CONTROLBUTTONINCRATE_PNG;
  63. m_downloads[_T("/controlbuttonskipforward.png")] = IDF_CONTROLBUTTONSKIPFORWARD_PNG;
  64. m_downloads[_T("/controlbuttonstep.png")] = IDF_CONTROLBUTTONSTEP_PNG;
  65. m_downloads[_T("/controlvolumeon.png")] = IDF_CONTROLVOLUMEON_PNG;
  66. m_downloads[_T("/controlvolumeoff.png")] = IDF_CONTROLVOLUMEOFF_PNG;
  67. m_downloads[_T("/controlvolumebar.png")] = IDF_CONTROLVOLUMEBAR_PNG;
  68. m_downloads[_T("/controlvolumegrip.png")] = IDF_CONTROLVOLUMEGRIP_PNG;
  69. }
  71. CRegKey key;
  72. CString str(_T("MIME\Database\Content Type"));
  73. if(ERROR_SUCCESS == key.Open(HKEY_CLASSES_ROOT, str, KEY_READ))
  74. {
  75. TCHAR buff[256];
  76. DWORD len = countof(buff);
  77. for(int i = 0; ERROR_SUCCESS == key.EnumKey(i, buff, &len); i++, len = countof(buff))
  78. {
  79. CRegKey mime;
  80. TCHAR ext[64];
  81. ULONG len = countof(ext);
  82. if(ERROR_SUCCESS == mime.Open(HKEY_CLASSES_ROOT, str + _T("\") + buff, KEY_READ)
  83. && ERROR_SUCCESS == mime.QueryStringValue(_T("Extension"), ext, &len))
  84. m_mimes[CStringA(ext).MakeLower()] = CStringA(buff).MakeLower();
  85. }
  86. }
  88. m_mimes[".html"] = "text/html";
  89. m_mimes[".txt"] = "text/plain";
  90. m_mimes[".css"] = "text/css";
  91. m_mimes[".gif"] = "image/gif";
  92. m_mimes[".jpeg"] = "image/jpeg";
  93. m_mimes[".jpg"] = "image/jpeg";
  94. m_mimes[".png"] = "image/png";
  96. GetModuleFileName(AfxGetInstanceHandle(), str.GetBuffer(MAX_PATH), MAX_PATH);
  97. str.ReleaseBuffer();
  98. m_webroot = CPath(str);
  99. m_webroot.RemoveFileSpec();
  101. CString WebRoot = AfxGetAppSettings().WebRoot;
  102. WebRoot.Replace('/', '\');
  103. WebRoot.Trim();
  104. CPath p(WebRoot);
  105. if(WebRoot.Find(_T(":\")) < 0 && WebRoot.Find(_T("\\")) < 0) m_webroot.Append(WebRoot);
  106. else m_webroot = p;
  107. m_webroot.Canonicalize();
  108. m_webroot.MakePretty();
  109. if(!m_webroot.IsDirectory()) m_webroot = CPath();
  111. CAtlList<CString> sl;
  112. Explode(AfxGetAppSettings().WebServerCGI, sl, ';');
  113. POSITION pos = sl.GetHeadPosition();
  114. while(pos)
  115. {
  116. CAtlList<CString> sl2;
  117. CString ext = Explode(sl.GetNext(pos), sl2, '=', 2);
  118. if(sl2.GetCount() < 2) continue;
  119. m_cgi[ext] = sl2.GetTail();
  120. }
  122. m_ThreadId = 0;
  123.     m_hThread = ::CreateThread(NULL, 0, StaticThreadProc, (LPVOID)this, 0, &m_ThreadId);
  124. }
  126. CWebServer::~CWebServer()
  127. {
  128.     if(m_hThread != NULL)
  129. {
  130. PostThreadMessage(m_ThreadId, WM_QUIT, 0, 0);
  131.         WaitForSingleObject(m_hThread, 10000);
  132.         EXECUTE_ASSERT(CloseHandle(m_hThread));
  133.     }
  134. }
  136. DWORD WINAPI CWebServer::StaticThreadProc(LPVOID lpParam)
  137. {
  138. return ((CWebServer*)lpParam)->ThreadProc();
  139. }
  141. DWORD CWebServer::ThreadProc()
  142. {
  143. if(!AfxSocketInit(NULL))
  144. return -1;
  146. CWebServerSocket s(this, m_nPort);
  148. MSG msg;
  149. while((int)GetMessage(&msg, NULL, 0, 0) > 0)
  150. {
  151. TranslateMessage(&msg);
  152. DispatchMessage(&msg);
  153. }
  155. return 0;
  156. }
  158. static void PutFileContents(LPCTSTR fn, const CStringA& data)
  159. {
  160. if(FILE* f = _tfopen(fn, _T("wb")))
  161. {
  162. fwrite((LPCSTR)data, 1, data.GetLength(), f);
  163. fclose(f);
  164. }
  165. }
  167. void CWebServer::Deploy(CString dir)
  168. {
  169. CStringA data;
  170. if(LoadResource(IDR_HTML_INDEX, data, RT_HTML)) PutFileContents(dir + _T("index.html"), data);
  171. if(LoadResource(IDR_HTML_BROWSER, data, RT_HTML)) PutFileContents(dir + _T("browser.html"), data);
  172. if(LoadResource(IDR_HTML_CONTROLS, data, RT_HTML)) PutFileContents(dir + _T("controls.html"), data);
  173. if(LoadResource(IDR_HTML_404, data, RT_HTML)) PutFileContents(dir + _T("404.html"), data);
  174. if(LoadResource(IDR_HTML_PLAYER, data, RT_HTML)) PutFileContents(dir + _T("player.html"), data);
  176. POSITION pos = m_downloads.GetStartPosition();
  177. while(pos)
  178. {
  179. CString fn;
  180. UINT id;
  181. m_downloads.GetNextAssoc(pos, fn, id);
  182. if(LoadResource(id, data, _T("FILE")))
  183. PutFileContents(dir + fn, data);
  184. }
  185. }
  187. bool CWebServer::ToLocalPath(CString& path, CString& redir)
  188. {
  189. if(!path.IsEmpty() && m_webroot.IsDirectory())
  190. {
  191. CString str = path;
  192. str.Replace('/', '\');
  193. str.TrimLeft('\');
  195. CPath p;
  196. p.Combine(m_webroot, str);
  197. p.Canonicalize();
  199. if(p.IsDirectory())
  200. {
  201. CAtlList<CString> sl;
  202. Explode(AfxGetAppSettings().WebDefIndex, sl, ';');
  203. POSITION pos = sl.GetHeadPosition();
  204. while(pos)
  205. {
  206. str = sl.GetNext(pos);
  207. CPath p2 = p;
  208. p2.Append(str);
  209. if(p2.FileExists())
  210. {
  211. p = p2;
  212. redir = path;
  213. if(redir.GetAt(redir.GetLength()-1) != '/') redir += '/';
  214. redir += str;
  215. break;
  216. }
  217. }
  218. }
  220. if(_tcslen(p) > _tcslen(m_webroot) && p.FileExists())
  221. {
  222. path = (LPCTSTR)p;
  223. return true;
  224. }
  225. }
  227. return false;
  228. }
  230. bool CWebServer::LoadPage(UINT resid, CStringA& str, CString path)
  231. {
  232. CString redir;
  233. if(ToLocalPath(path, redir))
  234. {
  235. if(FILE* f = _tfopen(path, _T("rb")))
  236. {
  237. fseek(f, 0, 2);
  238. char* buff = str.GetBufferSetLength(ftell(f));
  239. fseek(f, 0, 0);
  240. int len = fread(buff, 1, str.GetLength(), f);
  241. fclose(f);
  242. return len == str.GetLength();
  243. }
  244. }
  246. return LoadResource(resid, str, RT_HTML);
  247. }
  249. void CWebServer::OnAccept(CWebServerSocket* pServer)
  250. {
  251. CAutoPtr<CWebClientSocket> p(new CWebClientSocket(this, m_pMainFrame));
  252. if(pServer->Accept(*p))
  253. {
  254. CString name;
  255. UINT port;
  256. if(AfxGetAppSettings().fWebServerLocalhostOnly && p->GetPeerName(name, port) && name != _T("127.0.0.1"))
  257. {
  258. p->Close();
  259. return;
  260. }
  262. m_clients.AddTail(p);
  263. }
  264. }
  266. void CWebServer::OnClose(CWebClientSocket* pClient)
  267. {
  268. POSITION pos = m_clients.GetHeadPosition();
  269. while(pos)
  270. {
  271. POSITION cur = pos;
  272. if(m_clients.GetNext(pos) == pClient)
  273. {
  274. m_clients.RemoveAt(cur);
  275. break;
  276. }
  277. }
  278. }
  280. void CWebServer::OnRequest(CWebClientSocket* pClient, CStringA& hdr, CStringA& body)
  281. {
  282. CPath p(pClient->m_path);
  283. CStringA ext = p.GetExtension().MakeLower();
  284. CStringA mime;
  285. if(ext.IsEmpty()) mime = "text/html";
  286. else m_mimes.Lookup(ext, mime);
  288. hdr = "HTTP/1.0 200 OKrn";
  290. bool fHandled = false, fCGI = false;
  292. if(!fHandled && m_webroot.IsDirectory())
  293. {
  294. CStringA tmphdr;
  295. fHandled = fCGI = CallCGI(pClient, tmphdr, body, mime);
  297. if(fHandled)
  298. {
  299. tmphdr.Replace("rn", "n");
  300. CAtlList<CStringA> hdrlines;
  301. ExplodeMin(tmphdr, hdrlines, 'n');
  302. POSITION pos = hdrlines.GetHeadPosition();
  303. while(pos)
  304. {
  305. POSITION cur = pos;
  306. CAtlList<CStringA> sl;
  307. CStringA key = Explode(hdrlines.GetNext(pos), sl, ':', 2);
  308. if(sl.GetCount() < 2) continue;
  309. key.Trim().MakeLower();
  310. if(key == "content-type") {mime = sl.GetTail().Trim(); hdrlines.RemoveAt(cur);}
  311. else if(key == "content-length") {hdrlines.RemoveAt(cur);}
  312. }
  313. tmphdr = Implode(hdrlines, 'n');
  314. tmphdr.Replace("n", "rn");
  315. hdr += tmphdr + "rn";
  316. }
  317. }
  319. RequestHandler rh = NULL;
  320. if(!fHandled && m_internalpages.Lookup(pClient->m_path, rh) && (pClient->*rh)(hdr, body, mime))
  321. {
  322. if(mime.IsEmpty()) mime = "text/html";
  324. CString redir;
  325. if(pClient->m_get.Lookup(_T("redir"), redir)
  326. || pClient->m_post.Lookup(_T("redir"), redir))
  327. {
  328. if(redir.IsEmpty()) redir = '/';
  330. hdr = 
  331. "HTTP/1.0 302 Foundrn"
  332. "Location: " + CStringA(redir) + "rn";
  333. return;
  334. }
  336. fHandled = true;
  337. }
  339. if(!fHandled && m_webroot.IsDirectory())
  340. {
  341. fHandled = LoadPage(0, body, pClient->m_path);
  342. }
  344. UINT resid;
  345. CStringA res;
  346. if(!fHandled && m_downloads.Lookup(pClient->m_path, resid) && LoadResource(resid, res, _T("FILE")))
  347. {
  348. if(mime.IsEmpty()) mime = "application/octet-stream";
  349. memcpy(body.GetBufferSetLength(res.GetLength()), res.GetBuffer(), res.GetLength());
  350. fHandled = true;
  351. }
  353. if(!fHandled)
  354. {
  355. hdr = mime == "text/html"
  356. ? "HTTP/1.0 301 Moved Permanentlyrn" "Location: /404.htmlrn"
  357. : "HTTP/1.0 404 Not Foundrn";
  358. return;
  359. }
  361. if(mime == "text/html" && !fCGI)
  362. {
  363. hdr += 
  364. "Expires: Thu, 19 Nov 1981 08:52:00 GMTrn"
  365. "Cache-Control: no-store, no-cache, must-revalidate, post-check=0, pre-check=0rn"
  366. "Pragma: no-cachern";
  368. CStringA debug;
  369. if(AfxGetAppSettings().fWebServerPrintDebugInfo)
  370. {
  371. debug += "<hr>rn";
  372. CString key, value;
  373. POSITION pos;
  374. pos = pClient->m_hdrlines.GetStartPosition();
  375. while(pos) {pClient->m_hdrlines.GetNextAssoc(pos, key, value); debug += "HEADER[" + key + "] = " + value + "<br>rn";}
  376. debug += "cmd: " + pClient->m_cmd + "<br>rn";
  377. debug += "path: " + pClient->m_path + "<br>rn";
  378. debug += "ver: " + pClient->m_ver + "<br>rn";
  379. pos = pClient->m_get.GetStartPosition();
  380. while(pos) {pClient->m_get.GetNextAssoc(pos, key, value); debug += "GET[" + key + "] = " + value + "<br>rn";}
  381. pos = pClient->m_post.GetStartPosition();
  382. while(pos) {pClient->m_post.GetNextAssoc(pos, key, value); debug += "POST[" + key + "] = " + value + "<br>rn";}
  383. pos = pClient->m_cookie.GetStartPosition();
  384. while(pos) {pClient->m_cookie.GetNextAssoc(pos, key, value); debug += "COOKIE[" + key + "] = " + value + "<br>rn";}
  385. pos = pClient->m_request.GetStartPosition();
  386. while(pos) {pClient->m_request.GetNextAssoc(pos, key, value); debug += "REQUEST[" + key + "] = " + value + "<br>rn";}
  387. }
  389. body.Replace("[path]", CStringA(pClient->m_path));
  390. body.Replace("[indexpath]", "/index.html");
  391. body.Replace("[commandpath]", "/command.html");
  392. body.Replace("[browserpath]", "/browser.html");
  393. body.Replace("[controlspath]", "/controls.html");
  394. body.Replace("[wmcname]", "wm_command");
  395. body.Replace("[setposcommand]", CMD_SETPOS);
  396. body.Replace("[setvolumecommand]", CMD_SETVOLUME);
  397. body.Replace("[debug]", debug);
  398. // TODO: add more general tags to replace
  399. }
  401. // gzip
  402. if(AfxGetAppSettings().fWebServerUseCompression && hdr.Find("Content-Encoding:") < 0)
  403. do
  404. {
  405. CString accept_encoding;
  406. pClient->m_hdrlines.Lookup(_T("accept-encoding"), accept_encoding);
  407. accept_encoding.MakeLower();
  408. CAtlList<CString> sl;
  409. ExplodeMin(accept_encoding, sl, ',');
  410. if(!sl.Find(_T("gzip"))) break;;
  412. CHAR path[MAX_PATH], fn[MAX_PATH];
  413. if(!GetTempPathA(MAX_PATH, path) || !GetTempFileNameA(path, "mpc_gz", 0, fn))
  414. break;
  416. gzFile gf = gzopen(fn, "wb9");
  417. if(!gf || gzwrite(gf, (LPVOID)(LPCSTR)body, body.GetLength()) != body.GetLength())
  418. {
  419. if(gf) gzclose(gf);
  420. DeleteFileA(fn);
  421. break;
  422. }
  423. gzclose(gf);
  425. FILE* f = fopen(fn, "rb");
  426. if(!f) {DeleteFileA(fn); break;}
  427. fseek(f, 0, 2);
  428. CHAR* s = body.GetBufferSetLength(ftell(f));
  429. fseek(f, 0, 0);
  430. int len = fread(s, 1, body.GetLength(), f);
  431. ASSERT(len == body.GetLength());
  432. fclose(f);
  433. DeleteFileA(fn);
  435. hdr += "Content-Encoding: gziprn";
  436. }
  437. while(0);
  439. CStringA content;
  440. content.Format(
  441. "Content-Type: %srn"
  442. "Content-Length: %drn", 
  443. mime, body.GetLength());
  444. hdr += content;
  445. }
  447. static DWORD WINAPI KillCGI(LPVOID lParam)
  448. {
  449. HANDLE hProcess = (HANDLE)lParam;
  450. if(WaitForSingleObject(hProcess, 30000) == WAIT_TIMEOUT)
  451. TerminateProcess(hProcess, 0);
  452. return 0;
  453. }
  455. bool CWebServer::CallCGI(CWebClientSocket* pClient, CStringA& hdr, CStringA& body, CStringA& mime)
  456. {
  457. CString path = pClient->m_path, redir = path;
  458. if(!ToLocalPath(path, redir)) return false;
  459. CString ext = CPath(path).GetExtension().MakeLower();
  460. CPath dir(path);
  461. dir.RemoveFileSpec();
  463. CString cgi;
  464. if(!m_cgi.Lookup(ext, cgi) || !CPath(cgi).FileExists())
  465. return false;
  467. HANDLE hProcess = GetCurrentProcess();
  468. HANDLE hChildStdinRd, hChildStdinWr, hChildStdinWrDup = NULL;
  469. HANDLE hChildStdoutRd, hChildStdoutWr, hChildStdoutRdDup = NULL;
  471. SECURITY_ATTRIBUTES saAttr;
  472. ZeroMemory(&saAttr, sizeof(saAttr));
  473. saAttr.nLength = sizeof(saAttr);
  474. saAttr.bInheritHandle = TRUE;
  476. if(CreatePipe(&hChildStdoutRd, &hChildStdoutWr, &saAttr, 0)) 
  477. {
  478. BOOL fSuccess = DuplicateHandle(hProcess, hChildStdoutRd, hProcess, &hChildStdoutRdDup, 0, FALSE, DUPLICATE_SAME_ACCESS);
  479. CloseHandle(hChildStdoutRd);
  480. }
  482. if(CreatePipe(&hChildStdinRd, &hChildStdinWr, &saAttr, 0))
  483. {
  484. BOOL fSuccess = DuplicateHandle(hProcess, hChildStdinWr, hProcess, &hChildStdinWrDup, 0, FALSE, DUPLICATE_SAME_ACCESS);
  485. CloseHandle(hChildStdinWr);
  486. }
  488. STARTUPINFO siStartInfo;
  489. ZeroMemory(&siStartInfo, sizeof(siStartInfo));
  490. siStartInfo.cb = sizeof(siStartInfo);
  491. siStartInfo.hStdError = hChildStdoutWr;
  492. siStartInfo.hStdOutput = hChildStdoutWr;
  493. siStartInfo.hStdInput = hChildStdinRd;
  494. siStartInfo.dwFlags |= STARTF_USESTDHANDLES|STARTF_USESHOWWINDOW;
  495. siStartInfo.wShowWindow = SW_HIDE;
  497. PROCESS_INFORMATION piProcInfo;
  498. ZeroMemory(&piProcInfo, sizeof(piProcInfo));
  500. CStringA envstr;
  502. if(LPVOID lpvEnv = GetEnvironmentStrings())
  503. {
  504. CString str; 
  506. CAtlList<CString> env;
  507. for(LPTSTR lpszVariable = (LPTSTR)lpvEnv; *lpszVariable; lpszVariable += _tcslen(lpszVariable)+1)
  508. if(lpszVariable != (LPTSTR)lpvEnv)
  509. env.AddTail(lpszVariable);
  511. env.AddTail(_T("GATEWAY_INTERFACE=CGI/1.1"));
  512. env.AddTail(_T("SERVER_SOFTWARE=Media Player Classic/6.4.x.y"));
  513. env.AddTail(_T("SERVER_PROTOCOL=") + pClient->m_ver);
  514. env.AddTail(_T("REQUEST_METHOD=") + pClient->m_cmd);
  515. env.AddTail(_T("PATH_INFO=") + redir);
  516. env.AddTail(_T("PATH_TRANSLATED=") + path);
  517. env.AddTail(_T("SCRIPT_NAME=") + redir);
  518. env.AddTail(_T("QUERY_STRING=") + pClient->m_query);
  520. if(pClient->m_hdrlines.Lookup(_T("content-type"), str))
  521. env.AddTail(_T("CONTENT_TYPE=") + str);
  522. if(pClient->m_hdrlines.Lookup(_T("content-length"), str))
  523. env.AddTail(_T("CONTENT_LENGTH=") + str);
  525. POSITION pos = pClient->m_hdrlines.GetStartPosition();
  526. while(pos)
  527. {
  528. CString key = pClient->m_hdrlines.GetKeyAt(pos);
  529. CString value = pClient->m_hdrlines.GetNextValue(pos);
  530. key.Replace(_T("-"), _T("_"));
  531. key.MakeUpper();
  532. env.AddTail(_T("HTTP_") + key + _T("=") + value);
  533. }
  535. CString name;
  536. UINT port;
  538. if(pClient->GetPeerName(name, port))
  539. {
  540. str.Format(_T("%d"), port);
  541. env.AddTail(_T("REMOTE_ADDR=")+name);
  542. env.AddTail(_T("REMOTE_HOST=")+name);
  543. env.AddTail(_T("REMOTE_PORT=")+str);
  544. }
  546. if(pClient->GetSockName(name, port))
  547. {
  548. str.Format(_T("%d"), port);
  549. env.AddTail(_T("SERVER_NAME=")+name);
  550. env.AddTail(_T("SERVER_PORT=")+str);
  551. }
  553. env.AddTail(_T(""));
  555. str = Implode(env, '');
  556. envstr = CStringA(str, str.GetLength());
  558. FreeEnvironmentStrings((LPTSTR)lpvEnv);
  559. }
  561. TCHAR* cmdln = new TCHAR[32768];
  562. _sntprintf(cmdln, 32768, _T(""%s" "%s""), cgi, path);
  564. if(hChildStdinRd && hChildStdoutWr)
  565. if(CreateProcess(
  566. NULL, cmdln, NULL, NULL, TRUE, 0, 
  567. envstr.GetLength() ? (LPVOID)(LPCSTR)envstr : NULL, 
  568. dir, &siStartInfo, &piProcInfo))
  569. {
  570. DWORD ThreadId;
  571. CreateThread(NULL, 0, KillCGI, (LPVOID)piProcInfo.hProcess, 0, &ThreadId);
  573. static const int BUFFSIZE = 1024;
  574. DWORD dwRead, dwWritten = 0;
  576. int i = 0, len = pClient->m_data.GetLength();
  577. for(; i < len; i += dwWritten)
  578. if(!WriteFile(hChildStdinWrDup, (LPCSTR)pClient->m_data + i, min(len - i, BUFFSIZE), &dwWritten, NULL)) 
  579. break;
  581. CloseHandle(hChildStdinWrDup);
  582. CloseHandle(hChildStdoutWr);
  584. body.Empty();
  586. CStringA buff;
  587. while(i == len && ReadFile(hChildStdoutRdDup, buff.GetBuffer(BUFFSIZE), BUFFSIZE, &dwRead, NULL) && dwRead)
  588. {
  589. buff.ReleaseBufferSetLength(dwRead);
  590. body += buff;
  591. }
  593. int hdrend = body.Find("rnrn");
  594. if(hdrend >= 0)
  595. {
  596. hdr = body.Left(hdrend+2);
  597. body = body.Mid(hdrend+4);
  598. }
  600. CloseHandle(hChildStdinRd);
  601. CloseHandle(hChildStdoutRdDup);
  603. CloseHandle(piProcInfo.hProcess);
  604. CloseHandle(piProcInfo.hThread);
  605. }
  606. else
  607. {
  608. body = _T("CGI Error");
  609. }
  611. delete [] cmdln;
  613. return true;
  614. }