X-Git-Url: https://git.pterodactylus.net/?p=fms.git;a=blobdiff_plain;f=src%2Fhttp%2Fpages%2Flocalidentitiespage.cpp;h=5fa76b5053c239d1ac667b646238835821a4ac0a;hp=9aca3f3f226f015601f3b792d5b7838d714ab73b;hb=dec33c63afafabf83c3039e916725cac6faef9b3;hpb=9b22dd53fe62e312c1647310b7ec43aa127090af

diff --git a/src/http/pages/localidentitiespage.cpp b/src/http/pages/localidentitiespage.cpp
index 9aca3f3..5fa76b5 100644
--- a/src/http/pages/localidentitiespage.cpp
+++ b/src/http/pages/localidentitiespage.cpp
@@ -13,167 +13,6 @@ const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, c
 	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;
-		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);
-
-		if((*queryvars.find("formaction")).second=="update")
-		{
-			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();
-				}
-			}
-		}
-		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>";
@@ -192,8 +31,6 @@ 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><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,tblLocalIdentity.MinMessageDelay,tblLocalIdentity.MaxMessageDelay FROM tblLocalIdentity LEFT JOIN tblIdentity ON tblLocalIdentity.PublicKey=tblIdentity.PublicKey ORDER BY tblLocalIdentity.Name;");
@@ -226,7 +63,6 @@ const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, c
 		st.ResultText(9,maxmessagedelay);
 
 		content+="<tr>";
-//		content+="<td><input type=\"checkbox\" name=\"chkidentityid["+countstr+"]\" value=\""+id+"\"></td>";
 		content+="<td title=\""+publickey+"\"><form name=\"frmupdate\""+countstr+"\" method=\"POST\"><input type=\"hidden\" name=\"formaction\" value=\"update\"><input type=\"hidden\" name=\"chkidentityid["+countstr+"]\" value=\""+id+"\">"+SanitizeOutput(CreateShortIdentityName(name,publickey))+"</td>";
 		content+="<td>"+CreateTrueFalseDropDown("singleuse["+countstr+"]",singleuse)+"</td>";
 		content+="<td>"+CreateTrueFalseDropDown("publishtrustlist["+countstr+"]",publishtrustlist)+"</td>";
@@ -249,17 +85,6 @@ const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, c
 		}
 		trustst.Reset();
 
-/*
-		if(st.ResultNull(6))
-		{
-			content+="<td>No</td>";
-		}
-		else
-		{
-			content+="<td>Yes</td>";
-		}
-*/
-
 		content+="<td><input type=\"submit\" value=\"Update\"></form></td>";
 		content+="<td><form name=\"frmdel\""+countstr+"\" method=\"POST\" action=\"confirm.htm\"><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>";
@@ -268,13 +93,218 @@ const std::string LocalIdentitiesPage::GeneratePage(const std::string &method, c
 		count++;
 	}
 
-//	content+="<tr><td colspan=\"5\"><center><input type=\"submit\" value=\"Update Selected\"> <input type=\"submit\" value=\"Delete Selected\" onClick=\"if(confirm('Delete Selected Identities?')){frmlocalidentity.formaction.value='delete';}else{return false;}\"></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.  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 "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"+StringFunctions::Replace(m_template,"[CONTENT]",content);
+	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();
+			}
+		}
+	}
+}
+
+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())
+	{
+		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)