adam/znc
1
0
Fork 0
mirror of https://github.com/znc/znc.git synced 2026-08-11 19:32:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

Added PrintFile() and refactored some things

Browse Source 
git-svn-id: https://znc.svn.sourceforge.net/svnroot/znc/trunk@610 726aef4b-f618-498e-8847-2d620e286838
This commit is contained in:
prozacx
2006-02-08 03:15:29 +00:00
parent feba97bd33
commit 0917335eab
2 changed files with 68 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
HTTPSock.cpp
+63 -15
View File
@@ -93,12 +93,46 @@ void CHTTPSock::GetPage() {
}
if (PrintHeader(sPage.length())) {
DEBUG_ONLY(cout << "- 200 (OK)" << endl);
Write(sPage);
Close(Csock::CLT_AFTERWRITE);
}
}
bool CHTTPSock::PrintFile(const CString& sFileName, const CString& sContentType) {
CString sFilePath = sFileName;
if (!m_sDocRoot.empty()) {
sFilePath = CUtils::ChangeDir(m_sDocRoot, sFileName, m_sDocRoot);
if (sFilePath.Left(m_sDocRoot.size()) != m_sDocRoot) {
PrintErrorPage(403, "Forbidden", "You don't have permission to access to this document on this server.");
DEBUG_ONLY(cout << "THIS FILE: [" << sFilePath << "] does not live in ..." << endl);
DEBUG_ONLY(cout << "DOCUMENT ROOT: [" << m_sDocRoot << "]" << endl);
return false;
}
}
CFile File(sFileName);
if (!File.Open(CFile::F_Read)) {
PrintNotFound();
return false;
}
PrintHeader(File.GetSize(), "text/css");
char szBuf[4096];
int iLen = 0;
while((iLen = File.Read(szBuf, 4096)) > 0) {
Write(szBuf, iLen);
}
Close(Csock::CLT_AFTERWRITE);
return true;
}
void CHTTPSock::ParseURI() {
ParseParams(m_sURI.Token(1, true, "?"));
m_sURI = m_sURI.Token(0, false, "?");
@@ -119,6 +153,15 @@ void CHTTPSock::ParseParams(const CString& sParams) {
}
}
void CHTTPSock::SetDocRoot(const CString& s) {
m_sDocRoot = s;
m_sDocRoot.Replace("//", "/");
}
const CString& CHTTPSock::GetDocRoot() const {
return m_sDocRoot;
}
const CString& CHTTPSock::GetUser() const {
return m_sUser;
}
@@ -184,6 +227,23 @@ bool CHTTPSock::OnPageRequest(const CString& sURI, CString& sPageRet) {
return false;
}
bool CHTTPSock::PrintNotFound() {
return PrintErrorPage(404, "Not Found", "The requested URL was not found on this server.");
}
bool CHTTPSock::PrintErrorPage(unsigned int uStatusId, const CString& sStatusMsg, const CString& sMessage) {
if (SentHeader()) {
return false;
}
CString sPage = GetErrorPage(uStatusId, sStatusMsg, sMessage);
PrintHeader(sPage.length(), "text/html", uStatusId, sStatusMsg);
Write(sPage);
Close(Csock::CLT_AFTERWRITE);
return true;
}
CString CHTTPSock::GetErrorPage(unsigned int uStatusId, const CString& sStatusMsg, const CString& sMessage) {
return "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\">\r\n"
"<html><head>\r\n<title>" + CString::ToString(uStatusId) + " " + sStatusMsg.Escape_n(CString::EHTML) + "</title>\r\n"
@@ -215,29 +275,17 @@ bool CHTTPSock::OnLogin(const CString& sUser, const CString& sPass) {
return false;
}
bool CHTTPSock::PrintNotFound() {
if (SentHeader()) {
return false;
}
DEBUG_ONLY(cout << "- 404 (Not Found)" << endl);
CString sPage = GetErrorPage(404, "Not Found", "The requested URL was not found on this server.");
PrintHeader(sPage.length(), "text/html", 404, "Not Found");
Write(sPage);
Close(Csock::CLT_AFTERWRITE);
return true;
}
bool CHTTPSock::SentHeader() const {
return m_bSentHeader;
}
bool CHTTPSock::PrintHeader(unsigned long uContentLength, const CString& sContentType, unsigned int uStatusId, const CString& sStatusMsg) {
if (SentHeader()) {
DEBUG_ONLY(cout << "- Header already sent!" << endl);
return false;
}
DEBUG_ONLY(cout << "- " << uStatusId << " (" << sStatusMsg << ")" << endl);
if (!sContentType.empty()) {
m_sContentType = sContentType;
}