src/freenet/trustlistxml.cpp

   1 #include "../../include/freenet/trustlistxml.h"\r
   2 #include "../../include/stringfunctions.h"\r
   3 \r
   4 #include <algorithm>\r
   5 \r
   6 #ifdef XMEM\r
   7         #include <xmem.h>\r
   8 #endif\r
   9 \r
  10 TrustListXML::TrustListXML()\r
  11 {\r
  12         Initialize();\r
  13 }\r
  14 \r
  15 void TrustListXML::AddTrust(const std::string &identity, const long messagetrust, const long trustlisttrust)\r
  16 {\r
  17         m_trust.push_back(trust(identity,messagetrust,trustlisttrust));\r
  18 }\r
  19 \r
  20 std::string TrustListXML::GetIdentity(const long index)\r
  21 {\r
  22         if(index>=0 && index<m_trust.size())\r
  23         {\r
  24                 return m_trust[index].m_identity;\r
  25         }\r
  26         else\r
  27         {\r
  28                 return "";\r
  29         }\r
  30 }\r
  31 \r
  32 long TrustListXML::GetMessageTrust(const long index)\r
  33 {\r
  34         if(index>=0 && index<m_trust.size())\r
  35         {\r
  36                 return m_trust[index].m_messagetrust;\r
  37         }\r
  38         else\r
  39         {\r
  40                 return -1;\r
  41         }       \r
  42 }\r
  43 \r
  44 long TrustListXML::GetTrustListTrust(const long index)\r
  45 {\r
  46         if(index>=0 && index<m_trust.size())\r
  47         {\r
  48                 return m_trust[index].m_trustlisttrust;\r
  49         }\r
  50         else\r
  51         {\r
  52                 return -1;\r
  53         }\r
  54 }\r
  55 \r
  56 std::string TrustListXML::GetXML()\r
  57 {\r
  58         TiXmlDocument td;\r
  59         TiXmlDeclaration *tdec=new TiXmlDeclaration("1.0","UTF-8","");\r
  60         TiXmlElement *tid;\r
  61         TiXmlPrinter tp;\r
  62 \r
  63         td.LinkEndChild(tdec);\r
  64         tid=new TiXmlElement("TrustList");\r
  65         td.LinkEndChild(tid);\r
  66 \r
  67         for(std::vector<trust>::iterator i=m_trust.begin(); i!=m_trust.end(); i++)\r
  68         {\r
  69                 std::string messagetrust;\r
  70                 std::string trustlisttrust;\r
  71                 StringFunctions::Convert((*i).m_messagetrust,messagetrust);\r
  72                 StringFunctions::Convert((*i).m_trustlisttrust,trustlisttrust);\r
  73                 TiXmlElement *tr=new TiXmlElement("Trust");\r
  74                 tid->LinkEndChild(tr);\r
  75                 tr->LinkEndChild(XMLCreateCDATAElement("Identity",(*i).m_identity));\r
  76                 tr->LinkEndChild(XMLCreateTextElement("MessageTrustLevel",messagetrust));\r
  77                 tr->LinkEndChild(XMLCreateTextElement("TrustListTrustLevel",trustlisttrust));\r
  78         }\r
  79 \r
  80         td.Accept(&tp);\r
  81         return std::string(tp.CStr());\r
  82 }\r
  83 \r
  84 void TrustListXML::Initialize()\r
  85 {\r
  86         m_trust.clear();\r
  87 }\r
  88 \r
  89 const bool TrustListXML::ParseXML(const std::string &xml)\r
  90 {\r
  91         TiXmlDocument td;\r
  92         td.Parse(xml.c_str());\r
  93 \r
  94         if(!td.Error())\r
  95         {\r
  96                 std::string identity;\r
  97                 std::string messagetruststr;\r
  98                 std::string trustlisttruststr;\r
  99                 long messagetrust;\r
 100                 long trustlisttrust;\r
 101                 TiXmlText *txt;\r
 102                 TiXmlHandle hnd(&td);\r
 103                 TiXmlNode *node;\r
 104                 std::vector<std::string> m_foundkeys;\r
 105 \r
 106                 Initialize();\r
 107 \r
 108                 node=hnd.FirstChild("TrustList").FirstChild("Trust").ToElement();\r
 109                 while(node)\r
 110                 {\r
 111                         identity="";\r
 112                         messagetrust=-1;\r
 113                         trustlisttrust=-1;\r
 114 \r
 115                         TiXmlHandle hnd2(node);\r
 116                         txt=hnd2.FirstChild("Identity").FirstChild().ToText();\r
 117                         if(txt)\r
 118                         {\r
 119                                 identity=txt->ValueStr();\r
 120                         }\r
 121                         txt=hnd2.FirstChild("MessageTrustLevel").FirstChild().ToText();\r
 122                         if(txt)\r
 123                         {\r
 124                                 messagetruststr=txt->ValueStr();\r
 125                                 StringFunctions::Convert(messagetruststr,messagetrust);\r
 126                         }\r
 127                         txt=hnd2.FirstChild("TrustListTrustLevel").FirstChild().ToText();\r
 128                         if(txt)\r
 129                         {\r
 130                                 trustlisttruststr=txt->ValueStr();\r
 131                                 StringFunctions::Convert(trustlisttruststr,trustlisttrust);\r
 132                         }\r
 133 \r
 134                         if(identity!="" && messagetrust>=0 && messagetrust<=100 && trustlisttrust>=0 && trustlisttrust<=100)\r
 135                         {\r
 136                                 // check so we don't add the same identity multiple times from a trust list\r
 137                                 if(std::find(m_foundkeys.begin(),m_foundkeys.end(),identity)==m_foundkeys.end())\r
 138                                 {\r
 139                                         m_foundkeys.push_back(identity);\r
 140                                         m_trust.push_back(trust(identity,messagetrust,trustlisttrust));\r
 141                                 }\r
 142                         }\r
 143                         else\r
 144                         {\r
 145                                 m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"TrustListXML::ParseXML malformed Trust in TrustList.xml");\r
 146                         }\r
 147                         \r
 148                         node=node->NextSibling("Trust");\r
 149                 }\r
 150                 return true;\r
 151 \r
 152         }\r
 153         else\r
 154         {\r
 155                 return false;\r
 156         }\r
 157 }\r