X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fhttp%2Fpages%2Flocalidentitiespage.cpp;h=400260f6cf0091f2932d266cfc2d0f0abf56602e;hb=dabd19d7f764b8275c9c8370c7b89675b6a78243;hp=8b6afcb8ff317691ec4e01968db8866010e99753;hpb=8a0a83a78390f22f99d4487cda2569909dfbc28e;p=fms.git

diff --git a/src/http/pages/localidentitiespage.cpp b/src/http/pages/localidentitiespage.cpp
index 8b6afcb..400260f 100644
--- a/src/http/pages/localidentitiespage.cpp
+++ b/src/http/pages/localidentitiespage.cpp
@@ -1,200 +1,30 @@
 #include "../../../include/http/pages/localidentitiespage.h"
 #include "../../../include/stringfunctions.h"
 #include "../../../include/http/identityexportxml.h"
+#include "../../../include/global.h"
 
 #ifdef XMEM
 	#include <xmem.h>
 #endif
 
-const std::string LocalIdentitiesPage::CreateTrueFalseDropDown(const std::string &name, const std::string &selected)
-{
-	std::string rval="";
-
-	rval+="<select name=\""+name+"\">";
-	rval+="<option value=\"true\"";
-	if(selected=="true")
-	{
-		rval+=" SELECTED";
-	}
-	rval+=">true</option>";
-	rval+="<option value=\"false\"";
-	if(selected=="false")
-	{
-		rval+=" SELECTED";
-	}
-	rval+=">false</option>";
-	rval+="</select>";
-
-	return rval;
-}
-
 const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, const std::map<std::string,std::string> &queryvars)
 {
 	int count;
 	std::string countstr;
 	std::string content="";
 
-	if(queryvars.find("formaction")!=queryvars.end())
-	{
-		int id;
-		std::vector<std::string> ids;
-		std::vector<std::string> singleuse;
-		std::vector<std::string> publishtrustlist;
-		std::vector<std::string> publishboardlist;
-		std::vector<std::string> publishfreesite;
-
-		CreateArgArray(queryvars,"chkidentityid",ids);
-		CreateArgArray(queryvars,"singleuse",singleuse);
-		CreateArgArray(queryvars,"publishtrustlist",publishtrustlist);
-		CreateArgArray(queryvars,"publishboardlist",publishboardlist);
-		CreateArgArray(queryvars,"publishfreesite",publishfreesite);
-
-		if((*queryvars.find("formaction")).second=="update")
-		{
-			SQLite3DB::Statement update=m_db->Prepare("UPDATE tblLocalIdentity SET SingleUse=?, PublishTrustList=?, PublishBoardList=?, PublishFreesite=? WHERE LocalIdentityID=?;");
-			for(int i=0; i<ids.size(); i++)
-			{
-				if(ids[i]!="")
-				{
-					StringFunctions::Convert(ids[i],id);
-					update.Bind(0,singleuse[i]);
-					update.Bind(1,publishtrustlist[i]);
-					update.Bind(2,publishboardlist[i]);
-					update.Bind(3,publishfreesite[i]);
-					update.Bind(4,id);
-					update.Step();
-					update.Reset();
-				}
-			}
-		}
-		if((*queryvars.find("formaction")).second=="delete")
-		{
-			SQLite3DB::Statement del=m_db->Prepare("DELETE FROM tblLocalIdentity WHERE LocalIdentityID=?;");
-			for(int i=0; i<ids.size(); i++)
-			{
-				if(ids[i]!="")
-				{
-					StringFunctions::Convert(ids[i],id);
-					del.Bind(0,id);
-					del.Step();
-					del.Reset();
-				}
-			}
-		}
-		if((*queryvars.find("formaction")).second=="export")
-		{
-			IdentityExportXML xml;
-			SQLite3DB::Statement exp=m_db->Prepare("SELECT Name,PublicKey,PrivateKey,SingleUse,PublishTrustList,PublishBoardList,PublishFreesite FROM tblLocalIdentity WHERE PublicKey IS NOT NULL AND PrivateKey IS NOT NULL;");
-			exp.Step();
-			while(exp.RowReturned())
-			{
-				std::string name="";
-				std::string publickey="";
-				std::string privatekey="";
-				std::string tempval="";
-				bool singleuse=false;
-				bool publishtrustlist=false;
-				bool publishboardlist=false;
-				bool publishfreesite=false;
-
-				exp.ResultText(0,name);
-				exp.ResultText(1,publickey);
-				exp.ResultText(2,privatekey);
-				exp.ResultText(3,tempval);
-				if(tempval=="true")
-				{
-					singleuse=true;
-				}
-				exp.ResultText(4,tempval);
-				if(tempval=="true")
-				{
-					publishtrustlist=true;
-				}
-				exp.ResultText(5,tempval);
-				if(tempval=="true")
-				{
-					publishboardlist=true;
-				}
-				exp.ResultText(6,tempval);
-				if(tempval=="true")
-				{
-					publishfreesite=true;
-				}
-
-				xml.AddIdentity(name,publickey,privatekey,singleuse,publishtrustlist,publishboardlist,publishfreesite);
-
-				exp.Step();
-			}
-			return "HTTP/1.1 200 OK\r\nContent-Type: text/xml\r\nContent-Disposition: attachment; filename=identities.xml\r\n\r\n"+xml.GetXML();
-		}
-		if((*queryvars.find("formaction")).second=="import")
-		{
-			if(queryvars.find("file")!=queryvars.end())
-			{
-				IdentityExportXML xml;
-				if(xml.ParseXML((*queryvars.find("file")).second))
-				{
-					SQLite3DB::Statement imp=m_db->Prepare("INSERT INTO tblLocalIdentity(Name,PublicKey,PrivateKey,SingleUse,PublishTrustList,PublishBoardList,PublishFreesite) VALUES(?,?,?,?,?,?,?);");
-					for(int i=0; i<xml.GetCount(); i++)
-					{
-						std::string tempval="false";
-						imp.Bind(0,xml.GetName(i));
-						imp.Bind(1,xml.GetPublicKey(i));
-						imp.Bind(2,xml.GetPrivateKey(i));
-						if(xml.GetSingleUse(i))
-						{
-							tempval="true";
-						}
-						else
-						{
-							tempval="false";
-						}
-						imp.Bind(3,tempval);
-						if(xml.GetPublishTrustList(i))
-						{
-							tempval="true";
-						}
-						else
-						{
-							tempval="false";
-						}
-						imp.Bind(4,tempval);
-						if(xml.GetPublishBoardList(i))
-						{
-							tempval="true";
-						}
-						else
-						{
-							tempval="false";
-						}
-						imp.Bind(5,tempval);
-						if(xml.GetPublishFreesite(i))
-						{
-							tempval="true";
-						}
-						else
-						{
-							tempval="false";
-						}
-						imp.Bind(6,tempval);
-						imp.Step();
-						imp.Reset();
-					}
-				}
-			}
-		}
-	}
-
 	content+="<h2>Local Identities</h2>";
 
 	content+="<table><tr><th>Export Identities</th><th>Import Identities</th></tr>";
 	content+="<tr><td>";
 	content+="<form name=\"frmexport\" method=\"POST\">";
+	content+=CreateFormPassword();
 	content+="<input type=\"hidden\" name=\"formaction\" value=\"export\">";
 	content+="<input type=\"submit\" value=\"Export Identities\">";
 	content+="</form>";
 	content+="</td><td>";
 	content+="<form name=\"frmimport\" method=\"POST\" enctype=\"multipart/form-data\">";
+	content+=CreateFormPassword();
 	content+="<input type=\"hidden\" name=\"formaction\" value=\"import\">";
 	content+="<input type=\"file\" name=\"file\">";
 	content+="<input type=\"submit\" value=\"Import Identities\">";
@@ -203,12 +33,13 @@ const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, c
 
 	content+="<hr>";
 
-	content+="<form name=\"frmlocalidentity\" method=\"POST\">";
-	content+="<input type=\"hidden\" name=\"formaction\" value=\"update\">";
-	content+="<table><tr><td></td><th>Name</th><th>Single Use</th><th>Publish Trust List</th><th>Publish Board List</th><th>Publish Freesite</th><th>Announced? *</th></tr>";
+	content+="<table><tr><th>Name</th><th>Single Use</th><th>Publish Trust List</th><th>Publish Board List</th><th>Publish Freesite</th><th>Min Message Delay</th><th>Max Message Delay</th><th>Announced? *</th></tr>";
 
-	SQLite3DB::Statement st=m_db->Prepare("SELECT LocalIdentityID,tblLocalIdentity.Name,tblLocalIdentity.PublicKey,tbLLocalIdentity.PublishTrustList,tblLocalIdentity.SingleUse,tblLocalIdentity.PublishBoardList,tblIdentity.IdentityID,tblLocalIdentity.PublishFreesite FROM tblLocalIdentity LEFT JOIN tblIdentity ON tblLocalIdentity.PublicKey=tblIdentity.PublicKey ORDER BY tblLocalIdentity.Name;");
+	SQLite3DB::Statement st=m_db->Prepare("SELECT LocalIdentityID,tblLocalIdentity.Name,tblLocalIdentity.PublicKey,tbLLocalIdentity.PublishTrustList,tblLocalIdentity.SingleUse,tblLocalIdentity.PublishBoardList,tblIdentity.IdentityID,tblLocalIdentity.PublishFreesite,tblLocalIdentity.MinMessageDelay,tblLocalIdentity.MaxMessageDelay FROM tblLocalIdentity LEFT JOIN tblIdentity ON tblLocalIdentity.PublicKey=tblIdentity.PublicKey ORDER BY tblLocalIdentity.Name;");
 	st.Step();
+	SQLite3DB::Statement st2=m_db->Prepare("SELECT IdentityID FROM tblIdentity WHERE PublicKey=?;");
+
+	SQLite3DB::Statement trustst=m_db->Prepare("SELECT COUNT(*) FROM tblPeerTrust LEFT JOIN tblIdentity ON tblPeerTrust.TargetIdentityID=tblIdentity.IdentityID WHERE tblIdentity.PublicKey=? GROUP BY tblPeerTrust.TargetIdentityID;");
 
 	count=0;
 	while(st.RowReturned())
@@ -219,9 +50,11 @@ const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, c
 		std::string publickey="";
 		std::string publishtrustlist="";
 		std::string singleuse="";
-		std::string keypart="";
 		std::string publishboardlist="";
 		std::string publishfreesite="";
+		std::string minmessagedelay="0";
+		std::string maxmessagedelay="0";
+		std::string identityidstr="";
 
 		st.ResultText(0,id);
 		st.ResultText(1,name);
@@ -230,38 +63,270 @@ const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, c
 		st.ResultText(4,singleuse);
 		st.ResultText(5,publishboardlist);
 		st.ResultText(7,publishfreesite);
+		st.ResultText(8,minmessagedelay);
+		st.ResultText(9,maxmessagedelay);
 
-		if(publickey.size()>8)
+		st2.Bind(0,publickey);
+		st2.Step();
+		if(st2.RowReturned())
 		{
-			keypart=publickey.substr(3,5);
+			st2.ResultText(0,identityidstr);
 		}
+		st2.Reset();
 
 		content+="<tr>";
-		content+="<td><input type=\"checkbox\" name=\"chkidentityid["+countstr+"]\" value=\""+id+"\"></td>";
-		content+="<td title=\""+publickey+"\">"+SanitizeOutput(name+keypart)+"...</td>";
+		content+="<td title=\""+publickey+"\"><form name=\"frmupdate\""+countstr+"\" method=\"POST\"><input type=\"hidden\" name=\"formaction\" value=\"update\">"+CreateFormPassword()+"<input type=\"hidden\" name=\"chkidentityid["+countstr+"]\" value=\""+id+"\">";
+		if(identityidstr!="")
+		{
+			content+="<a href=\"peerdetails.htm?identityid="+identityidstr+"\">";
+		}
+		content+=SanitizeOutput(CreateShortIdentityName(name,publickey));
+		if(identityidstr!="")
+		{
+			content+="</a>";
+		}
+		content+="</td>";
 		content+="<td>"+CreateTrueFalseDropDown("singleuse["+countstr+"]",singleuse)+"</td>";
 		content+="<td>"+CreateTrueFalseDropDown("publishtrustlist["+countstr+"]",publishtrustlist)+"</td>";
 		content+="<td>"+CreateTrueFalseDropDown("publishboardlist["+countstr+"]",publishboardlist)+"</td>";
 		content+="<td>"+CreateTrueFalseDropDown("publishfreesite["+countstr+"]",publishfreesite)+"</td>";
-		if(st.ResultNull(6))
+		content+="<td><input type=\"text\" size=\"2\" name=\"mindelay["+countstr+"]\" value=\""+minmessagedelay+"\"></td>";
+		content+="<td><input type=\"text\" size=\"2\" name=\"maxdelay["+countstr+"]\" value=\""+maxmessagedelay+"\"></td>";
+		
+		trustst.Bind(0,publickey);
+		trustst.Step();
+		if(trustst.RowReturned())
 		{
-			content+="<td>No</td>";
+			std::string numlists="";
+			trustst.ResultText(0,numlists);
+			content+="<td>Yes ("+numlists+")</td>";
 		}
 		else
 		{
-			content+="<td>Yes</td>";
+			content+="<td>No</td>";
 		}
+		trustst.Reset();
+
+		content+="<td><input type=\"submit\" value=\"Update\"></form></td>";
+		content+="<td><form name=\"frmdel\""+countstr+"\" method=\"POST\" action=\"confirm.htm\">"+CreateFormPassword()+"<input type=\"hidden\" name=\"formaction\" value=\"delete\"><input type=\"hidden\" name=\"chkidentityid["+countstr+"]\" value=\""+id+"\"><input type=\"hidden\" name=\"targetpage\" value=\"localidentities.htm\"><input type=\"hidden\" name=\"confirmdescription\" value=\"Are you sure you want to delete "+SanitizeOutput(CreateShortIdentityName(name,publickey))+"?\"><input type=\"submit\" value=\"Delete\"></form></td>";
 		content+="</tr>";
+		content+="<tr><td></td><td colspan=\"7\" class=\"smaller\">"+publickey+"</td></tr>";
 		st.Step();
 		count++;
 	}
 
-	content+="<tr><td colspan=\"5\"><center><input type=\"submit\" value=\"Update Selected\"> <input type=\"submit\" value=\"Delete Selected\"></td></tr>";
 	content+="</table>";
-	content+="<p class=\"paragraph\">* An identity is considered successfully announced when you have downloaded a trust list from someone that contains the identity.</p>";
+	content+="<p class=\"paragraph\">* An identity is considered successfully announced when you have downloaded a trust list from someone that contains the identity.  The number in parenthesis is how many trust lists that identity appears in.</p>";
 	content+="<p class=\"paragraph\">Single Use Identities will automatically be deleted 7 days after creation.</p>";
+	content+="<p class=\"paragraph\">Messages that each identity sends may be delayed by a random number of minutes between min and max.  Set both to 0 to send messages as soon as possible.</p>";
+
+	return StringFunctions::Replace(m_template,"[CONTENT]",content);
+}
+
+void LocalIdentitiesPage::HandleDelete(const std::map<std::string,std::string> &queryvars)
+{
+	int id=0;
+	std::vector<std::string> ids;
+	CreateArgArray(queryvars,"chkidentityid",ids);
+
+	SQLite3DB::Statement del=m_db->Prepare("DELETE FROM tblLocalIdentity WHERE LocalIdentityID=?;");
+	for(int i=0; i<ids.size(); i++)
+	{
+		if(ids[i]!="")
+		{
+			StringFunctions::Convert(ids[i],id);
+			del.Bind(0,id);
+			del.Step();
+			del.Reset();
+		}
+	}
+}
+
+const std::string LocalIdentitiesPage::HandleExport()
+{
+	IdentityExportXML xml;
+	SQLite3DB::Statement exp=m_db->Prepare("SELECT Name,PublicKey,PrivateKey,SingleUse,PublishTrustList,PublishBoardList,PublishFreesite FROM tblLocalIdentity WHERE PublicKey IS NOT NULL AND PrivateKey IS NOT NULL;");
+	exp.Step();
+	while(exp.RowReturned())
+	{
+		std::string name="";
+		std::string publickey="";
+		std::string privatekey="";
+		std::string tempval="";
+		bool singleuse=false;
+		bool publishtrustlist=false;
+		bool publishboardlist=false;
+		bool publishfreesite=false;
+
+		exp.ResultText(0,name);
+		exp.ResultText(1,publickey);
+		exp.ResultText(2,privatekey);
+		exp.ResultText(3,tempval);
+		if(tempval=="true")
+		{
+			singleuse=true;
+		}
+		exp.ResultText(4,tempval);
+		if(tempval=="true")
+		{
+			publishtrustlist=true;
+		}
+		exp.ResultText(5,tempval);
+		if(tempval=="true")
+		{
+			publishboardlist=true;
+		}
+		exp.ResultText(6,tempval);
+		if(tempval=="true")
+		{
+			publishfreesite=true;
+		}
+
+		xml.AddIdentity(name,publickey,privatekey,singleuse,publishtrustlist,publishboardlist,publishfreesite);
+
+		exp.Step();
+	}
+	return xml.GetXML();
+}
+
+void LocalIdentitiesPage::HandleImport(const std::map<std::string,std::string> &queryvars)
+{
+	if(queryvars.find("file")!=queryvars.end())
+	{
+		IdentityExportXML xml;
+		if(xml.ParseXML((*queryvars.find("file")).second))
+		{
+			SQLite3DB::Statement imp=m_db->Prepare("INSERT INTO tblLocalIdentity(Name,PublicKey,PrivateKey,SingleUse,PublishTrustList,PublishBoardList,PublishFreesite) VALUES(?,?,?,?,?,?,?);");
+			for(int i=0; i<xml.GetCount(); i++)
+			{
+				std::string tempval="false";
+				imp.Bind(0,xml.GetName(i));
+				imp.Bind(1,xml.GetPublicKey(i));
+				imp.Bind(2,xml.GetPrivateKey(i));
+				if(xml.GetSingleUse(i))
+				{
+					tempval="true";
+				}
+				else
+				{
+					tempval="false";
+				}
+				imp.Bind(3,tempval);
+				if(xml.GetPublishTrustList(i))
+				{
+					tempval="true";
+				}
+				else
+				{
+					tempval="false";
+				}
+				imp.Bind(4,tempval);
+				if(xml.GetPublishBoardList(i))
+				{
+					tempval="true";
+				}
+				else
+				{
+					tempval="false";
+				}
+				imp.Bind(5,tempval);
+				if(xml.GetPublishFreesite(i))
+				{
+					tempval="true";
+				}
+				else
+				{
+					tempval="false";
+				}
+				imp.Bind(6,tempval);
+				imp.Step();
+				imp.Reset();
+			}
+		}
+	}
+}
 
-	return "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"+StringFunctions::Replace(m_template,"[CONTENT]",content);
+void LocalIdentitiesPage::handleRequest(Poco::Net::HTTPServerRequest &request, Poco::Net::HTTPServerResponse &response)
+{
+	m_log->trace("LocalIdentitiesPages::handleRequest from "+request.clientAddress().toString());
+
+	std::map<std::string,std::string> vars;
+	CreateQueryVarMap(request,vars);
+
+	std::string formaction="";
+	if(vars.find("formaction")!=vars.end() && ValidateFormPassword(vars))
+	{
+		formaction=(*vars.find("formaction")).second;
+		if(formaction=="update")
+		{
+			HandleUpdate(vars);
+		}
+		else if(formaction=="delete")
+		{
+			HandleDelete(vars);
+		}
+		else if(formaction=="export")
+		{
+			response.setChunkedTransferEncoding(true);
+			response.setContentType("application/xml");
+			response.set("Content-Disposition","attachment; filename=identities.xml");
+			std::ostream &out=response.send();
+			out << HandleExport();
+			return;
+		}
+		else if(formaction=="import")
+		{
+			HandleImport(vars);
+		}
+	}
+
+	response.setChunkedTransferEncoding(true);
+	response.setContentType("text/html");
+
+	std::ostream &ostr = response.send();
+	ostr << GeneratePage(request.getMethod(),vars);
+}
+
+void LocalIdentitiesPage::HandleUpdate(const std::map<std::string,std::string> &queryvars)
+{
+	int id;
+	std::vector<std::string> ids;
+	std::vector<std::string> singleuse;
+	std::vector<std::string> publishtrustlist;
+	std::vector<std::string> publishboardlist;
+	std::vector<std::string> publishfreesite;
+	std::vector<std::string> mindelay;
+	std::vector<std::string> maxdelay;
+
+	CreateArgArray(queryvars,"chkidentityid",ids);
+	CreateArgArray(queryvars,"singleuse",singleuse);
+	CreateArgArray(queryvars,"publishtrustlist",publishtrustlist);
+	CreateArgArray(queryvars,"publishboardlist",publishboardlist);
+	CreateArgArray(queryvars,"publishfreesite",publishfreesite);
+	CreateArgArray(queryvars,"mindelay",mindelay);
+	CreateArgArray(queryvars,"maxdelay",maxdelay);
+
+	SQLite3DB::Statement update=m_db->Prepare("UPDATE tblLocalIdentity SET SingleUse=?, PublishTrustList=?, PublishBoardList=?, PublishFreesite=?, MinMessageDelay=?, MaxMessageDelay=? WHERE LocalIdentityID=?;");
+	for(int i=0; i<ids.size(); i++)
+	{
+		if(ids[i]!="")
+		{
+			int minmessagedelay=0;
+			int maxmessagedelay=0;
+			StringFunctions::Convert(ids[i],id);
+			StringFunctions::Convert(mindelay[i],minmessagedelay);
+			StringFunctions::Convert(maxdelay[i],maxmessagedelay);
+			update.Bind(0,singleuse[i]);
+			update.Bind(1,publishtrustlist[i]);
+			update.Bind(2,publishboardlist[i]);
+			update.Bind(3,publishfreesite[i]);
+			update.Bind(4,minmessagedelay);
+			update.Bind(5,maxmessagedelay);
+			update.Bind(6,id);
+			update.Step();
+			update.Reset();
+		}
+	}
 }
 
 const bool LocalIdentitiesPage::WillHandleURI(const std::string &uri)