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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
html.cpp
资源名称：P2PSource-PeerCast.rar　[点击查看]
上传用户：chn_coc
上传日期：2007-12-20
资源大小：563k
文件大小：10k
源码类别：

P2P编程

开发平台：

Windows_Unix

html.cpp：源码内容
  1. // ------------------------------------------------ // File : html.cpp // Date: 4-apr-2002 // Author: giles // Desc:  // HTML protocol handling // // (c) 2002 peercast.org // ------------------------------------------------ // This program is free software; you can redistribute it and/or modify // it under the terms of the GNU General Public License as published by // the Free Software Foundation; either version 2 of the License, or // (at your option) any later version. // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the // GNU General Public License for more details. // ------------------------------------------------ #include <stdarg.h> #include <stdlib.h>
  2. #include "html.h" #include "http.h" #include "stream.h" #include "gnutella.h" #include "servmgr.h" #include "channel.h" #include "stats.h"
  3. #include "version2.h"
  4. // -------------------------------------- HTML::HTML(const char *t, Stream &o) { out = &o; out->writeCRLF = false; title.set(t); tagLevel = 0; refresh = 0; } // -------------------------------------- void HTML::writeOK(const char *content) { out->writeLine(HTTP_SC_OK); out->writeLineF("%s %s",HTTP_HS_SERVER,PCX_AGENT); //out->writeLine("%s %s",HTTP_HS_CACHE,"no-cache"); out->writeLineF("%s %s",HTTP_HS_CONNECTION,"close");     out->writeLineF("%s %s",HTTP_HS_CONTENT,content);
  5. out->writeLine(""); } // -------------------------------------- void HTML::writeVariable(Stream &s,const String &varName, int loop) { bool r=false; if (varName.startsWith("servMgr.")) r=servMgr->writeVariable(s,varName+8); else if (varName.startsWith("chanMgr.")) r=chanMgr->writeVariable(s,varName+8,loop); else if (varName.startsWith("stats.")) r=stats.writeVariable(s,varName+6);
  6. else if (varName.startsWith("sys."))
  7. {
  8. if (varName == "sys.log.dumpHTML")
  9. {
  10. sys->logBuf->dumpHTML(s);
  11. r=true;
  12. }
  13. }
  14. else if (varName.startsWith("loop."))
  15. {
  16. if (varName.startsWith("loop.channel."))
  17. {
  18. Channel *ch = chanMgr->findChannelByIndex(loop);
  19. if (ch)
  20. r = ch->writeVariable(s,varName+13,loop);
  21. }else if (varName.startsWith("loop.servent."))
  22. {
  23. Servent *sv = servMgr->findServentByIndex(loop);
  24. if (sv)
  25. r = sv->writeVariable(s,varName+13);
  26. }else if (varName.startsWith("loop.filter."))
  27. {
  28. ServFilter *sf = &servMgr->filters[loop];
  29. r = sf->writeVariable(s,varName+12);
  31. }else if (varName.startsWith("loop.bcid."))
  32. {
  33. BCID *bcid = servMgr->findValidBCID(loop);
  34. if (bcid)
  35. r = bcid->writeVariable(s,varName+10);
  37. }else if (varName == "loop.indexEven")
  38. {
  39. s.writeStringF("%d",(loop&1)==0);
  40. r = true;
  41. }else if (varName == "loop.index")
  42. {
  43. s.writeStringF("%d",loop);
  44. r = true;
  45. }else if (varName.startsWith("loop.hit."))
  46. {
  47. char *idstr = getCGIarg(tmplArgs,"id=");
  48. if (idstr)
  49. {
  50. GnuID id;
  51. id.fromStr(idstr);
  52. ChanHitList *chl = chanMgr->findHitListByID(id);
  53. if (chl)
  54. {
  55. int cnt=0;
  56. ChanHit *ch = chl->hit;
  57. while (ch)
  58. {
  59. if (ch->host.ip && !ch->dead)
  60. {
  61. if (cnt == loop)
  62. {
  63. r = ch->writeVariable(s,varName+9);
  64. break;
  65. }
  66. cnt++;
  67. }
  68. ch=ch->next;
  69. }
  71. }
  72. }
  73. }
  75. }
  76. else if (varName.startsWith("page."))
  77. {
  78. if (varName.startsWith("page.channel."))
  79. {
  80. char *idstr = getCGIarg(tmplArgs,"id=");
  81. if (idstr)
  82. {
  83. GnuID id;
  84. id.fromStr(idstr);
  85. Channel *ch = chanMgr->findChannelByID(id);
  86. if (ch)
  87. r = ch->writeVariable(s,varName+13,loop);
  88. }
  89. }else
  90. {
  92. String v = varName+5;
  93. v.append('=');
  94. char *a = getCGIarg(tmplArgs,v);
  95. if (a)
  96. {
  97. s.writeString(a);
  98. r=true;
  99. }
  100. }
  101. }
  102. if (!r) s.writeString(varName); } // --------------------------------------
  103. int HTML::getIntVariable(const String &varName,int loop)
  104. {
  105. String val;
  106. MemoryStream mem(val.cstr(),String::MAX_LEN);
  108. writeVariable(mem,varName,loop);
  110. return atoi(val.cstr());
  111. }
  112. // --------------------------------------
  113. bool HTML::getBoolVariable(const String &varName,int loop)
  114. {
  115. String val;
  116. MemoryStream mem(val.cstr(),String::MAX_LEN);
  118. writeVariable(mem,varName,loop);
  120. // integer
  121. if ((val[0] >= '0') && (val[0] <= '9'))
  122. return atoi(val.cstr()) != 0;
  124. // string
  125. if (val[0]!=0)
  126. return true;
  128. return false;
  129. }
  131. // --------------------------------------
  132. void HTML::readIf(Stream &in,Stream *outp,int loop)
  133. {
  134. String var;
  135. bool varCond=true;
  137. while (!in.eof())
  138. {
  139. char c = in.readChar();
  141. if (c == '}')
  142. {
  143. if (getBoolVariable(var,loop)==varCond)
  144. {
  145. if (readTemplate(in,outp,loop))
  146. readTemplate(in,NULL,loop);
  147. }else{
  148. if (readTemplate(in,NULL,loop))
  149. readTemplate(in,outp,loop);
  150. }
  151. return;
  152. }else if (c == '!')
  153. {
  154. varCond = !varCond;
  155. }else
  156. {
  157. var.append(c);
  158. }
  159. }
  161. }
  163. // --------------------------------------
  164. void HTML::readLoop(Stream &in,Stream *outp,int loop)
  165. {
  166. String var;
  167. while (!in.eof())
  168. {
  169. char c = in.readChar();
  171. if (c == '}')
  172. {
  173. int cnt = getIntVariable(var,loop);
  175. if (cnt)
  176. {
  177. int spos = in.getPosition();
  178. for(int i=0; i<cnt; i++)
  179. {
  180. in.seekTo(spos);
  181. readTemplate(in,outp,i);
  182. }
  183. }else
  184. {
  185. readTemplate(in,NULL,0);
  186. }
  187. return;
  189. }else
  190. {
  191. var.append(c);
  192. }
  193. }
  195. }
  197. // --------------------------------------
  198. int HTML::readCmd(Stream &in,Stream *outp,int loop)
  199. {
  200. String cmd;
  202. int tmpl = TMPL_UNKNOWN;
  204. while (!in.eof())
  205. {
  206. char c = in.readChar();
  208. if (String::isWhitespace(c) || (c=='}'))
  209. {
  210. if (cmd == "loop")
  211. {
  212. readLoop(in,outp,loop);
  213. tmpl = TMPL_LOOP;
  214. }else if (cmd == "if")
  215. {
  216. readIf(in,outp,loop);
  217. tmpl = TMPL_IF;
  218. }else if (cmd == "end")
  219. {
  220. tmpl = TMPL_END;
  221. }
  222. else if (cmd == "else")
  223. {
  224. tmpl = TMPL_ELSE;
  225. }
  226. break;
  227. }else
  228. {
  229. cmd.append(c);
  230. }
  231. }
  232. return tmpl;
  233. }
  235. // --------------------------------------
  236. void HTML::readVariable(Stream &in,Stream *outp,int loop)
  237. {
  238. String var;
  239. while (!in.eof())
  240. {
  241. char c = in.readChar();
  242. if (c == '}')
  243. {
  244. if (outp)
  245. writeVariable(*outp,var,loop);
  246. return;
  247. }else
  248. {
  249. var.append(c);
  250. }
  251. }
  253. }
  254. // --------------------------------------
  255. bool HTML::readTemplate(Stream &in,Stream *outp,int loop)
  256. {
  257. while (!in.eof())
  258. {
  259. char c = in.readChar();
  261. if (c == '{')
  262. {
  263. c = in.readChar();
  264. if (c == '$')
  265. {
  266. readVariable(in,outp,loop);
  267. }
  268. else if (c == '@')
  269. {
  270. int t = readCmd(in,outp,loop);
  271. if (t == TMPL_END)
  272. return false;
  273. else if (t == TMPL_ELSE)
  274. return true;
  275. }
  276. else
  277. throw StreamException("Unknown template escape");
  278. }else
  279. {
  280. if (outp)
  281. outp->writeChar(c);
  282. }
  283. }
  284. return false;
  285. }
  287. // -------------------------------------- void HTML::writeTemplate(const char *fileName, const char *args) {
  288. FileStream file; try { file.openReadOnly(fileName);
  289. tmplArgs = args;
  290. readTemplate(file,out,0);
  291. }catch(StreamException &e) { out->writeString(e.msg); out->writeString(" : "); out->writeString(fileName);
  292. } file.close(); } // --------------------------------------
  293. void HTML::writeRawFile(const char *fileName)
  294. {
  295. FileStream file;
  296. try
  297. {
  298. file.openReadOnly(fileName);
  300. file.writeTo(*out,file.length());
  302. }catch(StreamException &)
  303. {
  304. }
  306. file.close();
  307. }
  308. // -------------------------------------- void HTML::locateTo(const char *url) { out->writeLine(HTTP_SC_FOUND); out->writeLineF("Location: %s",url); out->writeLine(""); } // -------------------------------------- void HTML::startHTML() { startNode("html"); } // -------------------------------------- void HTML::startBody() { startNode("body"); } // -------------------------------------- void HTML::addHead() { char buf[512]; startNode("head"); startTagEnd("title",title.cstr()); startTagEnd("meta http-equiv="Content-Type" content="text/html; charset=iso-8859-1""); if (!refreshURL.isEmpty()) { sprintf(buf,"meta http-equiv="refresh" content="%d;URL=%s"",refresh,refreshURL.cstr()); startTagEnd(buf); }else if (refresh) { sprintf(buf,"meta http-equiv="refresh" content="%d"",refresh); startTagEnd(buf); } end(); } // -------------------------------------- void HTML::addContent(const char *s) { out->writeString(s); } // -------------------------------------- void HTML::startNode(const char *tag, const char *data) { const char *p = tag; char *o = &currTag[tagLevel][0]; int i; for(i=0; i<MAX_TAGLEN-1; i++) { char c = *p++; if ((c==0) || (c==' ')) break; else *o++ = c; } *o = 0; out->writeString("<"); out->writeString(tag); out->writeString(">"); if (data) out->writeString(data); tagLevel++; if (tagLevel >= MAX_TAGLEVEL) throw StreamException("HTML too deep!"); } // -------------------------------------- void HTML::end() { tagLevel--; if (tagLevel < 0) throw StreamException("HTML premature end!"); out->writeString("</"); out->writeString(&currTag[tagLevel][0]); out->writeString(">"); } // -------------------------------------- void HTML::addLink(const char *url, const char *text, bool toblank) { char buf[1024]; sprintf(buf,"a href="%s" %s",url,toblank?"target="_blank"":""); startNode(buf,text); end(); } // -------------------------------------- void HTML::startTag(const char *tag, const char *fmt,...) { if (fmt) { va_list ap;    va_start(ap, fmt); char tmp[512]; vsprintf(tmp,fmt,ap); startNode(tag,tmp);     va_end(ap); }else{ startNode(tag,NULL); } } // -------------------------------------- void HTML::startTagEnd(const char *tag, const char *fmt,...) { if (fmt) { va_list ap;    va_start(ap, fmt); char tmp[512]; vsprintf(tmp,fmt,ap); startNode(tag,tmp);     va_end(ap); }else{ startNode(tag,NULL); } end(); } // -------------------------------------- void HTML::startSingleTagEnd(const char *fmt,...) { va_list ap; va_start(ap, fmt); char tmp[512]; vsprintf(tmp,fmt,ap); startNode(tmp); va_end(ap); end(); } // -------------------------------------- void HTML::startTableRow(int i) { if (i & 1) startTag("tr bgcolor="#dddddd" align="left""); else startTag("tr bgcolor="#eeeeee" align="left""); }