2
0

SqliteNetworkController.cpp 95 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119
  1. /*
  2. * ZeroTier One - Network Virtualization Everywhere
  3. * Copyright (C) 2011-2015 ZeroTier, Inc.
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. *
  18. * --
  19. *
  20. * ZeroTier may be used and distributed under the terms of the GPLv3, which
  21. * are available at: http://www.gnu.org/licenses/gpl-3.0.html
  22. *
  23. * If you would like to embed ZeroTier into a commercial application or
  24. * redistribute it in a modified binary form, please contact ZeroTier Networks
  25. * LLC. Start here: http://www.zerotier.com/
  26. */
  27. #include <stdint.h>
  28. #include <stdio.h>
  29. #include <stdlib.h>
  30. #include <string.h>
  31. #include <time.h>
  32. #include <sys/time.h>
  33. #include <sys/types.h>
  34. #include <algorithm>
  35. #include <utility>
  36. #include <stdexcept>
  37. #include <set>
  38. #include "../include/ZeroTierOne.h"
  39. #include "../node/Constants.hpp"
  40. #ifdef ZT_USE_SYSTEM_JSON_PARSER
  41. #include <json-parser/json.h>
  42. #else
  43. #include "../ext/json-parser/json.h"
  44. #endif
  45. #include "SqliteNetworkController.hpp"
  46. #include "../node/Node.hpp"
  47. #include "../node/Utils.hpp"
  48. #include "../node/CertificateOfMembership.hpp"
  49. #include "../node/NetworkConfig.hpp"
  50. #include "../node/NetworkConfigRequestMetaData.hpp"
  51. #include "../node/InetAddress.hpp"
  52. #include "../node/MAC.hpp"
  53. #include "../node/Address.hpp"
  54. #include "../osdep/OSUtils.hpp"
  55. // Include ZT_NETCONF_SCHEMA_SQL constant to init database
  56. #include "schema.sql.c"
  57. // Stored in database as schemaVersion key in Config.
  58. // If not present, database is assumed to be empty and at the current schema version
  59. // and this key/value is added automatically.
  60. #define ZT_NETCONF_SQLITE_SCHEMA_VERSION 3
  61. #define ZT_NETCONF_SQLITE_SCHEMA_VERSION_STR "3"
  62. // API version reported via JSON control plane
  63. #define ZT_NETCONF_CONTROLLER_API_VERSION 1
  64. // Min duration between requests for an address/nwid combo to prevent floods
  65. #define ZT_NETCONF_MIN_REQUEST_PERIOD 1000
  66. // Delay between backups in milliseconds
  67. #define ZT_NETCONF_BACKUP_PERIOD 300000
  68. // Number of NodeHistory entries to maintain per node and network (can be changed)
  69. #define ZT_NETCONF_NODE_HISTORY_LENGTH 64
  70. // Nodes are considered active if they've queried in less than this long
  71. #define ZT_NETCONF_NODE_ACTIVE_THRESHOLD ((ZT_NETWORK_AUTOCONF_DELAY * 2) + 5000)
  72. namespace ZeroTier {
  73. namespace {
  74. static std::string _jsonEscape(const char *s)
  75. {
  76. if (!s)
  77. return std::string();
  78. std::string buf;
  79. for(const char *p=s;(*p);++p) {
  80. switch(*p) {
  81. case '\t': buf.append("\\t"); break;
  82. case '\b': buf.append("\\b"); break;
  83. case '\r': buf.append("\\r"); break;
  84. case '\n': buf.append("\\n"); break;
  85. case '\f': buf.append("\\f"); break;
  86. case '"': buf.append("\\\""); break;
  87. case '\\': buf.append("\\\\"); break;
  88. case '/': buf.append("\\/"); break;
  89. default: buf.push_back(*p); break;
  90. }
  91. }
  92. return buf;
  93. }
  94. static std::string _jsonEscape(const std::string &s) { return _jsonEscape(s.c_str()); }
  95. // Converts an InetAddress to a blob and an int for storage in database
  96. static void _ipToBlob(const InetAddress &a,char *ipBlob,int &ipVersion) /* blob[16] */
  97. {
  98. switch(a.ss_family) {
  99. case AF_INET:
  100. if ((a.netmaskBits() >= 0)&&(a.netmaskBits() <= 32)) {
  101. memset(ipBlob,0,12);
  102. memcpy(ipBlob + 12,a.rawIpData(),4);
  103. ipVersion = 4;
  104. }
  105. break;
  106. case AF_INET6:
  107. if ((a.netmaskBits() >= 0)&&(a.netmaskBits() <= 128)) {
  108. memcpy(ipBlob,a.rawIpData(),16);
  109. ipVersion = 6;
  110. }
  111. break;
  112. }
  113. }
  114. struct MemberRecord {
  115. int64_t rowid;
  116. char nodeId[16];
  117. bool authorized;
  118. bool activeBridge;
  119. };
  120. struct NetworkRecord {
  121. char id[24];
  122. const char *name;
  123. const char *v4AssignMode;
  124. const char *v6AssignMode;
  125. bool isPrivate;
  126. bool enableBroadcast;
  127. bool allowPassiveBridging;
  128. int multicastLimit;
  129. uint64_t creationTime;
  130. uint64_t revision;
  131. uint64_t memberRevisionCounter;
  132. };
  133. } // anonymous namespace
  134. SqliteNetworkController::SqliteNetworkController(Node *node,const char *dbPath,const char *circuitTestPath) :
  135. _node(node),
  136. _backupThreadRun(true),
  137. _dbPath(dbPath),
  138. _circuitTestPath(circuitTestPath),
  139. _db((sqlite3 *)0)
  140. {
  141. if (sqlite3_open_v2(dbPath,&_db,SQLITE_OPEN_READWRITE|SQLITE_OPEN_CREATE,(const char *)0) != SQLITE_OK)
  142. throw std::runtime_error("SqliteNetworkController cannot open database file");
  143. sqlite3_busy_timeout(_db,10000);
  144. sqlite3_exec(_db,"PRAGMA synchronous = OFF",0,0,0);
  145. sqlite3_exec(_db,"PRAGMA journal_mode = MEMORY",0,0,0);
  146. sqlite3_stmt *s = (sqlite3_stmt *)0;
  147. if ((sqlite3_prepare_v2(_db,"SELECT v FROM Config WHERE k = 'schemaVersion';",-1,&s,(const char **)0) == SQLITE_OK)&&(s)) {
  148. int schemaVersion = -1234;
  149. if (sqlite3_step(s) == SQLITE_ROW) {
  150. schemaVersion = sqlite3_column_int(s,0);
  151. }
  152. sqlite3_finalize(s);
  153. if (schemaVersion == -1234) {
  154. sqlite3_close(_db);
  155. throw std::runtime_error("SqliteNetworkController schemaVersion not found in Config table (init failure?)");
  156. }
  157. if (schemaVersion < 2) {
  158. // Create NodeHistory table to upgrade from version 1 to version 2
  159. if (sqlite3_exec(_db,
  160. "CREATE TABLE NodeHistory (\n"
  161. " nodeId char(10) NOT NULL REFERENCES Node(id) ON DELETE CASCADE,\n"
  162. " networkId char(16) NOT NULL REFERENCES Network(id) ON DELETE CASCADE,\n"
  163. " networkVisitCounter INTEGER NOT NULL DEFAULT(0),\n"
  164. " networkRequestAuthorized INTEGER NOT NULL DEFAULT(0),\n"
  165. " requestTime INTEGER NOT NULL DEFAULT(0),\n"
  166. " clientMajorVersion INTEGER NOT NULL DEFAULT(0),\n"
  167. " clientMinorVersion INTEGER NOT NULL DEFAULT(0),\n"
  168. " clientRevision INTEGER NOT NULL DEFAULT(0),\n"
  169. " networkRequestMetaData VARCHAR(1024),\n"
  170. " fromAddress VARCHAR(128)\n"
  171. ");\n"
  172. "CREATE INDEX NodeHistory_nodeId ON NodeHistory (nodeId);\n"
  173. "CREATE INDEX NodeHistory_networkId ON NodeHistory (networkId);\n"
  174. "CREATE INDEX NodeHistory_requestTime ON NodeHistory (requestTime);\n"
  175. "UPDATE \"Config\" SET \"v\" = 2 WHERE \"k\" = 'schemaVersion';\n"
  176. ,0,0,0) != SQLITE_OK) {
  177. char err[1024];
  178. Utils::snprintf(err,sizeof(err),"SqliteNetworkController cannot upgrade the database to version 2: %s",sqlite3_errmsg(_db));
  179. sqlite3_close(_db);
  180. throw std::runtime_error(err);
  181. } else {
  182. schemaVersion = 2;
  183. }
  184. }
  185. if (schemaVersion < 3) {
  186. // Create Route table to upgrade from version 2 to version 3 and migrate old
  187. // data. Also delete obsolete Gateway table that was never actually used.
  188. if (sqlite3_exec(_db,
  189. "DROP TABLE Gateway;\n"
  190. "CREATE TABLE Route (\n"
  191. " networkId char(16) NOT NULL REFERENCES Network(id) ON DELETE CASCADE,\n"
  192. " target blob(16) NOT NULL,\n"
  193. " via blob(16),\n"
  194. " targetNetmaskBits integer NOT NULL,\n"
  195. " ipVersion integer NOT NULL,\n"
  196. " flags integer NOT NULL,\n"
  197. " metric integer NOT NULL\n"
  198. ");\n"
  199. "CREATE INDEX Route_networkId ON Route (networkId);\n"
  200. "INSERT INTO Route SELECT DISTINCT networkId,\"ip\" AS \"target\",NULL AS \"via\",ipNetmaskBits AS targetNetmaskBits,ipVersion,0 AS \"flags\",0 AS \"metric\" FROM IpAssignment WHERE nodeId IS NULL AND \"type\" = 1;\n"
  201. "DELETE FROM IpAssignment WHERE nodeId IS NULL AND \"type\" = 1;\n"
  202. "UPDATE \"Config\" SET \"v\" = 3 WHERE \"k\" = 'schemaVersion';\n"
  203. ,0,0,0) != SQLITE_OK) {
  204. char err[1024];
  205. Utils::snprintf(err,sizeof(err),"SqliteNetworkController cannot upgrade the database to version 3: %s",sqlite3_errmsg(_db));
  206. sqlite3_close(_db);
  207. throw std::runtime_error(err);
  208. } else {
  209. schemaVersion = 3;
  210. }
  211. }
  212. if (schemaVersion != ZT_NETCONF_SQLITE_SCHEMA_VERSION) {
  213. sqlite3_close(_db);
  214. throw std::runtime_error("SqliteNetworkController database schema version mismatch");
  215. }
  216. } else {
  217. // Prepare statement will fail if Config table doesn't exist, which means our DB
  218. // needs to be initialized.
  219. if (sqlite3_exec(_db,ZT_NETCONF_SCHEMA_SQL"INSERT INTO Config (k,v) VALUES ('schemaVersion',"ZT_NETCONF_SQLITE_SCHEMA_VERSION_STR");",0,0,0) != SQLITE_OK) {
  220. char err[1024];
  221. Utils::snprintf(err,sizeof(err),"SqliteNetworkController cannot initialize database and/or insert schemaVersion into Config table: %s",sqlite3_errmsg(_db));
  222. sqlite3_close(_db);
  223. throw std::runtime_error(err);
  224. }
  225. }
  226. if (
  227. /* Network */
  228. (sqlite3_prepare_v2(_db,"SELECT name,private,enableBroadcast,allowPassiveBridging,v4AssignMode,v6AssignMode,multicastLimit,creationTime,revision,memberRevisionCounter,(SELECT COUNT(1) FROM Member WHERE Member.networkId = Network.id AND Member.authorized > 0) FROM Network WHERE id = ?",-1,&_sGetNetworkById,(const char **)0) != SQLITE_OK)
  229. ||(sqlite3_prepare_v2(_db,"SELECT revision FROM Network WHERE id = ?",-1,&_sGetNetworkRevision,(const char **)0) != SQLITE_OK)
  230. ||(sqlite3_prepare_v2(_db,"UPDATE Network SET revision = ? WHERE id = ?",-1,&_sSetNetworkRevision,(const char **)0) != SQLITE_OK)
  231. ||(sqlite3_prepare_v2(_db,"INSERT INTO Network (id,name,creationTime,revision) VALUES (?,?,?,1)",-1,&_sCreateNetwork,(const char **)0) != SQLITE_OK)
  232. ||(sqlite3_prepare_v2(_db,"DELETE FROM Network WHERE id = ?",-1,&_sDeleteNetwork,(const char **)0) != SQLITE_OK)
  233. ||(sqlite3_prepare_v2(_db,"SELECT id FROM Network ORDER BY id ASC",-1,&_sListNetworks,(const char **)0) != SQLITE_OK)
  234. ||(sqlite3_prepare_v2(_db,"UPDATE Network SET memberRevisionCounter = (memberRevisionCounter + 1) WHERE id = ?",-1,&_sIncrementMemberRevisionCounter,(const char **)0) != SQLITE_OK)
  235. /* Node */
  236. ||(sqlite3_prepare_v2(_db,"SELECT identity FROM Node WHERE id = ?",-1,&_sGetNodeIdentity,(const char **)0) != SQLITE_OK)
  237. ||(sqlite3_prepare_v2(_db,"INSERT OR REPLACE INTO Node (id,identity) VALUES (?,?)",-1,&_sCreateOrReplaceNode,(const char **)0) != SQLITE_OK)
  238. /* NodeHistory */
  239. ||(sqlite3_prepare_v2(_db,"SELECT IFNULL(MAX(networkVisitCounter),0) FROM NodeHistory WHERE networkId = ? AND nodeId = ?",-1,&_sGetMaxNodeHistoryNetworkVisitCounter,(const char **)0) != SQLITE_OK)
  240. ||(sqlite3_prepare_v2(_db,"INSERT INTO NodeHistory (nodeId,networkId,networkVisitCounter,networkRequestAuthorized,requestTime,clientMajorVersion,clientMinorVersion,clientRevision,networkRequestMetaData,fromAddress) VALUES (?,?,?,?,?,?,?,?,?,?)",-1,&_sAddNodeHistoryEntry,(const char **)0) != SQLITE_OK)
  241. ||(sqlite3_prepare_v2(_db,"DELETE FROM NodeHistory WHERE networkId = ? AND nodeId = ? AND networkVisitCounter <= ?",-1,&_sDeleteOldNodeHistoryEntries,(const char **)0) != SQLITE_OK)
  242. ||(sqlite3_prepare_v2(_db,"SELECT nodeId,requestTime,clientMajorVersion,clientMinorVersion,clientRevision,fromAddress,networkRequestAuthorized FROM NodeHistory WHERE networkId = ? AND requestTime IN (SELECT MAX(requestTime) FROM NodeHistory WHERE networkId = ? AND requestTime >= ? GROUP BY nodeId) ORDER BY nodeId ASC,requestTime DESC",-1,&_sGetActiveNodesOnNetwork,(const char **)0) != SQLITE_OK)
  243. ||(sqlite3_prepare_v2(_db,"SELECT networkVisitCounter,networkRequestAuthorized,requestTime,clientMajorVersion,clientMinorVersion,clientRevision,networkRequestMetaData,fromAddress FROM NodeHistory WHERE networkId = ? AND nodeId = ? ORDER BY requestTime DESC",-1,&_sGetNodeHistory,(const char **)0) != SQLITE_OK)
  244. /* Rule */
  245. ||(sqlite3_prepare_v2(_db,"SELECT etherType FROM Rule WHERE networkId = ? AND \"action\" = 'accept'",-1,&_sGetEtherTypesFromRuleTable,(const char **)0) != SQLITE_OK)
  246. ||(sqlite3_prepare_v2(_db,"INSERT INTO Rule (networkId,ruleNo,nodeId,sourcePort,destPort,vlanId,vlanPcP,etherType,macSource,macDest,ipSource,ipDest,ipTos,ipProtocol,ipSourcePort,ipDestPort,flags,invFlags,\"action\") VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)",-1,&_sCreateRule,(const char **)0) != SQLITE_OK)
  247. ||(sqlite3_prepare_v2(_db,"SELECT ruleNo,nodeId,sourcePort,destPort,vlanId,vlanPcp,etherType,macSource,macDest,ipSource,ipDest,ipTos,ipProtocol,ipSourcePort,ipDestPort,\"flags\",invFlags,\"action\" FROM Rule WHERE networkId = ? ORDER BY ruleNo ASC",-1,&_sListRules,(const char **)0) != SQLITE_OK)
  248. ||(sqlite3_prepare_v2(_db,"DELETE FROM Rule WHERE networkId = ?",-1,&_sDeleteRulesForNetwork,(const char **)0) != SQLITE_OK)
  249. /* IpAssignmentPool */
  250. ||(sqlite3_prepare_v2(_db,"SELECT ipRangeStart,ipRangeEnd FROM IpAssignmentPool WHERE networkId = ? AND ipVersion = ?",-1,&_sGetIpAssignmentPools,(const char **)0) != SQLITE_OK)
  251. ||(sqlite3_prepare_v2(_db,"SELECT ipRangeStart,ipRangeEnd,ipVersion FROM IpAssignmentPool WHERE networkId = ? ORDER BY ipRangeStart ASC",-1,&_sGetIpAssignmentPools2,(const char **)0) != SQLITE_OK)
  252. ||(sqlite3_prepare_v2(_db,"INSERT INTO IpAssignmentPool (networkId,ipRangeStart,ipRangeEnd,ipVersion) VALUES (?,?,?,?)",-1,&_sCreateIpAssignmentPool,(const char **)0) != SQLITE_OK)
  253. ||(sqlite3_prepare_v2(_db,"DELETE FROM IpAssignmentPool WHERE networkId = ?",-1,&_sDeleteIpAssignmentPoolsForNetwork,(const char **)0) != SQLITE_OK)
  254. /* IpAssignment */
  255. ||(sqlite3_prepare_v2(_db,"SELECT \"type\",ip,ipNetmaskBits FROM IpAssignment WHERE networkId = ? AND (nodeId = ? OR nodeId IS NULL) AND ipVersion = ?",-1,&_sGetIpAssignmentsForNode,(const char **)0) != SQLITE_OK)
  256. ||(sqlite3_prepare_v2(_db,"SELECT ip,ipNetmaskBits,ipVersion FROM IpAssignment WHERE networkId = ? AND nodeId = ? AND \"type\" = ? ORDER BY ip ASC",-1,&_sGetIpAssignmentsForNode2,(const char **)0) != SQLITE_OK)
  257. ||(sqlite3_prepare_v2(_db,"SELECT 1 FROM IpAssignment WHERE networkId = ? AND ip = ? AND ipVersion = ? AND \"type\" = ?",-1,&_sCheckIfIpIsAllocated,(const char **)0) != SQLITE_OK)
  258. ||(sqlite3_prepare_v2(_db,"INSERT INTO IpAssignment (networkId,nodeId,\"type\",ip,ipNetmaskBits,ipVersion) VALUES (?,?,?,?,?,?)",-1,&_sAllocateIp,(const char **)0) != SQLITE_OK)
  259. ||(sqlite3_prepare_v2(_db,"DELETE FROM IpAssignment WHERE networkId = ? AND nodeId = ? AND \"type\" = ?",-1,&_sDeleteIpAllocations,(const char **)0) != SQLITE_OK)
  260. /* Relay */
  261. ||(sqlite3_prepare_v2(_db,"SELECT \"address\",\"phyAddress\" FROM Relay WHERE \"networkId\" = ? ORDER BY \"address\" ASC",-1,&_sGetRelays,(const char **)0) != SQLITE_OK)
  262. ||(sqlite3_prepare_v2(_db,"DELETE FROM Relay WHERE networkId = ?",-1,&_sDeleteRelaysForNetwork,(const char **)0) != SQLITE_OK)
  263. ||(sqlite3_prepare_v2(_db,"INSERT INTO Relay (\"networkId\",\"address\",\"phyAddress\") VALUES (?,?,?)",-1,&_sCreateRelay,(const char **)0) != SQLITE_OK)
  264. /* Member */
  265. ||(sqlite3_prepare_v2(_db,"SELECT rowid,authorized,activeBridge FROM Member WHERE networkId = ? AND nodeId = ?",-1,&_sGetMember,(const char **)0) != SQLITE_OK)
  266. ||(sqlite3_prepare_v2(_db,"SELECT m.authorized,m.activeBridge,m.memberRevision,n.identity FROM Member AS m LEFT OUTER JOIN Node AS n ON n.id = m.nodeId WHERE m.networkId = ? AND m.nodeId = ?",-1,&_sGetMember2,(const char **)0) != SQLITE_OK)
  267. ||(sqlite3_prepare_v2(_db,"INSERT INTO Member (networkId,nodeId,authorized,activeBridge,memberRevision) VALUES (?,?,?,0,(SELECT memberRevisionCounter FROM Network WHERE id = ?))",-1,&_sCreateMember,(const char **)0) != SQLITE_OK)
  268. ||(sqlite3_prepare_v2(_db,"SELECT nodeId FROM Member WHERE networkId = ? AND activeBridge > 0 AND authorized > 0",-1,&_sGetActiveBridges,(const char **)0) != SQLITE_OK)
  269. ||(sqlite3_prepare_v2(_db,"SELECT m.nodeId,m.memberRevision FROM Member AS m WHERE m.networkId = ? ORDER BY m.nodeId ASC",-1,&_sListNetworkMembers,(const char **)0) != SQLITE_OK)
  270. ||(sqlite3_prepare_v2(_db,"UPDATE Member SET authorized = ?,memberRevision = (SELECT memberRevisionCounter FROM Network WHERE id = ?) WHERE rowid = ?",-1,&_sUpdateMemberAuthorized,(const char **)0) != SQLITE_OK)
  271. ||(sqlite3_prepare_v2(_db,"UPDATE Member SET activeBridge = ?,memberRevision = (SELECT memberRevisionCounter FROM Network WHERE id = ?) WHERE rowid = ?",-1,&_sUpdateMemberActiveBridge,(const char **)0) != SQLITE_OK)
  272. ||(sqlite3_prepare_v2(_db,"DELETE FROM Member WHERE networkId = ? AND nodeId = ?",-1,&_sDeleteMember,(const char **)0) != SQLITE_OK)
  273. ||(sqlite3_prepare_v2(_db,"DELETE FROM Member WHERE networkId = ?",-1,&_sDeleteAllNetworkMembers,(const char **)0) != SQLITE_OK)
  274. /* Route */
  275. ||(sqlite3_prepare_v2(_db,"INSERT INTO Route (networkId,target,via,targetNetmaskBits,ipVersion,flags,metric) VALUES (?,?,?,?,?,?,?)",-1,&_sCreateRoute,(const char **)0) != SQLITE_OK)
  276. ||(sqlite3_prepare_v2(_db,"SELECT DISTINCT target,via,targetNetmaskBits,ipVersion,flags,metric FROM \"Route\" WHERE networkId = ? ORDER BY ipVersion,target,via",-1,&_sGetRoutes,(const char **)0) != SQLITE_OK)
  277. ||(sqlite3_prepare_v2(_db,"DELETE FROM \"Route\" WHERE networkId = ?",-1,&_sDeleteRoutes,(const char **)0) != SQLITE_OK)
  278. /* Config */
  279. ||(sqlite3_prepare_v2(_db,"SELECT \"v\" FROM \"Config\" WHERE \"k\" = ?",-1,&_sGetConfig,(const char **)0) != SQLITE_OK)
  280. ||(sqlite3_prepare_v2(_db,"INSERT OR REPLACE INTO \"Config\" (\"k\",\"v\") VALUES (?,?)",-1,&_sSetConfig,(const char **)0) != SQLITE_OK)
  281. ) {
  282. std::string err(std::string("SqliteNetworkController unable to initialize one or more prepared statements: ") + sqlite3_errmsg(_db));
  283. sqlite3_close(_db);
  284. throw std::runtime_error(err);
  285. }
  286. /* Generate a 128-bit / 32-character "instance ID" if one isn't already
  287. * defined. Clients can use this to determine if this is the same controller
  288. * database they know and love. */
  289. sqlite3_reset(_sGetConfig);
  290. sqlite3_bind_text(_sGetConfig,1,"instanceId",10,SQLITE_STATIC);
  291. if (sqlite3_step(_sGetConfig) != SQLITE_ROW) {
  292. unsigned char sr[32];
  293. Utils::getSecureRandom(sr,32);
  294. for(unsigned int i=0;i<32;++i)
  295. _instanceId.push_back("0123456789abcdef"[(unsigned int)sr[i] & 0xf]);
  296. sqlite3_reset(_sSetConfig);
  297. sqlite3_bind_text(_sSetConfig,1,"instanceId",10,SQLITE_STATIC);
  298. sqlite3_bind_text(_sSetConfig,2,_instanceId.c_str(),-1,SQLITE_STATIC);
  299. if (sqlite3_step(_sSetConfig) != SQLITE_DONE)
  300. throw std::runtime_error("SqliteNetworkController unable to read or initialize instanceId");
  301. } else {
  302. const char *iid = reinterpret_cast<const char *>(sqlite3_column_text(_sGetConfig,0));
  303. if (!iid)
  304. throw std::runtime_error("SqliteNetworkController unable to read instanceId (it's NULL)");
  305. _instanceId = iid;
  306. }
  307. _backupThread = Thread::start(this);
  308. }
  309. SqliteNetworkController::~SqliteNetworkController()
  310. {
  311. _backupThreadRun = false;
  312. Thread::join(_backupThread);
  313. Mutex::Lock _l(_lock);
  314. if (_db) {
  315. sqlite3_finalize(_sGetNetworkById);
  316. sqlite3_finalize(_sGetMember);
  317. sqlite3_finalize(_sCreateMember);
  318. sqlite3_finalize(_sGetNodeIdentity);
  319. sqlite3_finalize(_sCreateOrReplaceNode);
  320. sqlite3_finalize(_sGetMaxNodeHistoryNetworkVisitCounter);
  321. sqlite3_finalize(_sAddNodeHistoryEntry);
  322. sqlite3_finalize(_sDeleteOldNodeHistoryEntries);
  323. sqlite3_finalize(_sGetActiveNodesOnNetwork);
  324. sqlite3_finalize(_sGetNodeHistory);
  325. sqlite3_finalize(_sGetEtherTypesFromRuleTable);
  326. sqlite3_finalize(_sGetActiveBridges);
  327. sqlite3_finalize(_sGetIpAssignmentsForNode);
  328. sqlite3_finalize(_sGetIpAssignmentPools);
  329. sqlite3_finalize(_sCheckIfIpIsAllocated);
  330. sqlite3_finalize(_sAllocateIp);
  331. sqlite3_finalize(_sDeleteIpAllocations);
  332. sqlite3_finalize(_sGetRelays);
  333. sqlite3_finalize(_sListNetworks);
  334. sqlite3_finalize(_sListNetworkMembers);
  335. sqlite3_finalize(_sGetMember2);
  336. sqlite3_finalize(_sGetIpAssignmentPools2);
  337. sqlite3_finalize(_sListRules);
  338. sqlite3_finalize(_sCreateRule);
  339. sqlite3_finalize(_sCreateNetwork);
  340. sqlite3_finalize(_sGetNetworkRevision);
  341. sqlite3_finalize(_sSetNetworkRevision);
  342. sqlite3_finalize(_sGetIpAssignmentsForNode2);
  343. sqlite3_finalize(_sDeleteRelaysForNetwork);
  344. sqlite3_finalize(_sCreateRelay);
  345. sqlite3_finalize(_sDeleteIpAssignmentPoolsForNetwork);
  346. sqlite3_finalize(_sDeleteRulesForNetwork);
  347. sqlite3_finalize(_sCreateIpAssignmentPool);
  348. sqlite3_finalize(_sUpdateMemberAuthorized);
  349. sqlite3_finalize(_sUpdateMemberActiveBridge);
  350. sqlite3_finalize(_sDeleteMember);
  351. sqlite3_finalize(_sDeleteAllNetworkMembers);
  352. sqlite3_finalize(_sDeleteNetwork);
  353. sqlite3_finalize(_sCreateRoute);
  354. sqlite3_finalize(_sGetRoutes);
  355. sqlite3_finalize(_sDeleteRoutes);
  356. sqlite3_finalize(_sIncrementMemberRevisionCounter);
  357. sqlite3_finalize(_sGetConfig);
  358. sqlite3_finalize(_sSetConfig);
  359. sqlite3_close(_db);
  360. }
  361. }
  362. NetworkController::ResultCode SqliteNetworkController::doNetworkConfigRequest(const InetAddress &fromAddr,const Identity &signingId,const Identity &identity,uint64_t nwid,const NetworkConfigRequestMetaData &metaData,Buffer<8194> &netconf)
  363. {
  364. Mutex::Lock _l(_lock);
  365. return _doNetworkConfigRequest(fromAddr,signingId,identity,nwid,metaData,netconf);
  366. }
  367. unsigned int SqliteNetworkController::handleControlPlaneHttpGET(
  368. const std::vector<std::string> &path,
  369. const std::map<std::string,std::string> &urlArgs,
  370. const std::map<std::string,std::string> &headers,
  371. const std::string &body,
  372. std::string &responseBody,
  373. std::string &responseContentType)
  374. {
  375. Mutex::Lock _l(_lock);
  376. return _doCPGet(path,urlArgs,headers,body,responseBody,responseContentType);
  377. }
  378. unsigned int SqliteNetworkController::handleControlPlaneHttpPOST(
  379. const std::vector<std::string> &path,
  380. const std::map<std::string,std::string> &urlArgs,
  381. const std::map<std::string,std::string> &headers,
  382. const std::string &body,
  383. std::string &responseBody,
  384. std::string &responseContentType)
  385. {
  386. if (path.empty())
  387. return 404;
  388. Mutex::Lock _l(_lock);
  389. if (path[0] == "network") {
  390. if ((path.size() >= 2)&&(path[1].length() == 16)) {
  391. uint64_t nwid = Utils::hexStrToU64(path[1].c_str());
  392. char nwids[24];
  393. Utils::snprintf(nwids,sizeof(nwids),"%.16llx",(unsigned long long)nwid);
  394. int64_t revision = 0;
  395. sqlite3_reset(_sGetNetworkRevision);
  396. sqlite3_bind_text(_sGetNetworkRevision,1,nwids,16,SQLITE_STATIC);
  397. bool networkExists = false;
  398. if (sqlite3_step(_sGetNetworkRevision) == SQLITE_ROW) {
  399. networkExists = true;
  400. revision = sqlite3_column_int64(_sGetNetworkRevision,0);
  401. }
  402. if (path.size() >= 3) {
  403. if (!networkExists)
  404. return 404;
  405. if ((path.size() == 4)&&(path[2] == "member")&&(path[3].length() == 10)) {
  406. uint64_t address = Utils::hexStrToU64(path[3].c_str());
  407. char addrs[24];
  408. Utils::snprintf(addrs,sizeof(addrs),"%.10llx",address);
  409. int64_t addToNetworkRevision = 0;
  410. int64_t memberRowId = 0;
  411. sqlite3_reset(_sGetMember);
  412. sqlite3_bind_text(_sGetMember,1,nwids,16,SQLITE_STATIC);
  413. sqlite3_bind_text(_sGetMember,2,addrs,10,SQLITE_STATIC);
  414. bool memberExists = false;
  415. if (sqlite3_step(_sGetMember) == SQLITE_ROW) {
  416. memberExists = true;
  417. memberRowId = sqlite3_column_int64(_sGetMember,0);
  418. }
  419. if (!memberExists) {
  420. sqlite3_reset(_sCreateMember);
  421. sqlite3_bind_text(_sCreateMember,1,nwids,16,SQLITE_STATIC);
  422. sqlite3_bind_text(_sCreateMember,2,addrs,10,SQLITE_STATIC);
  423. sqlite3_bind_int(_sCreateMember,3,0);
  424. sqlite3_bind_text(_sCreateMember,4,nwids,16,SQLITE_STATIC);
  425. if (sqlite3_step(_sCreateMember) != SQLITE_DONE)
  426. return 500;
  427. memberRowId = (int64_t)sqlite3_last_insert_rowid(_db);
  428. sqlite3_reset(_sIncrementMemberRevisionCounter);
  429. sqlite3_bind_text(_sIncrementMemberRevisionCounter,1,nwids,16,SQLITE_STATIC);
  430. sqlite3_step(_sIncrementMemberRevisionCounter);
  431. addToNetworkRevision = 1;
  432. }
  433. json_value *j = json_parse(body.c_str(),body.length());
  434. if (j) {
  435. if (j->type == json_object) {
  436. for(unsigned int k=0;k<j->u.object.length;++k) {
  437. if (!strcmp(j->u.object.values[k].name,"authorized")) {
  438. if (j->u.object.values[k].value->type == json_boolean) {
  439. sqlite3_reset(_sUpdateMemberAuthorized);
  440. sqlite3_bind_int(_sUpdateMemberAuthorized,1,(j->u.object.values[k].value->u.boolean == 0) ? 0 : 1);
  441. sqlite3_bind_text(_sUpdateMemberAuthorized,2,nwids,16,SQLITE_STATIC);
  442. sqlite3_bind_int64(_sUpdateMemberAuthorized,3,memberRowId);
  443. if (sqlite3_step(_sUpdateMemberAuthorized) != SQLITE_DONE)
  444. return 500;
  445. sqlite3_reset(_sIncrementMemberRevisionCounter);
  446. sqlite3_bind_text(_sIncrementMemberRevisionCounter,1,nwids,16,SQLITE_STATIC);
  447. sqlite3_step(_sIncrementMemberRevisionCounter);
  448. addToNetworkRevision = 1;
  449. }
  450. } else if (!strcmp(j->u.object.values[k].name,"activeBridge")) {
  451. if (j->u.object.values[k].value->type == json_boolean) {
  452. sqlite3_reset(_sUpdateMemberActiveBridge);
  453. sqlite3_bind_int(_sUpdateMemberActiveBridge,1,(j->u.object.values[k].value->u.boolean == 0) ? 0 : 1);
  454. sqlite3_bind_text(_sUpdateMemberActiveBridge,2,nwids,16,SQLITE_STATIC);
  455. sqlite3_bind_int64(_sUpdateMemberActiveBridge,3,memberRowId);
  456. if (sqlite3_step(_sUpdateMemberActiveBridge) != SQLITE_DONE)
  457. return 500;
  458. sqlite3_reset(_sIncrementMemberRevisionCounter);
  459. sqlite3_bind_text(_sIncrementMemberRevisionCounter,1,nwids,16,SQLITE_STATIC);
  460. sqlite3_step(_sIncrementMemberRevisionCounter);
  461. addToNetworkRevision = 1;
  462. }
  463. } else if (!strcmp(j->u.object.values[k].name,"ipAssignments")) {
  464. if (j->u.object.values[k].value->type == json_array) {
  465. sqlite3_reset(_sDeleteIpAllocations);
  466. sqlite3_bind_text(_sDeleteIpAllocations,1,nwids,16,SQLITE_STATIC);
  467. sqlite3_bind_text(_sDeleteIpAllocations,2,addrs,10,SQLITE_STATIC);
  468. sqlite3_bind_int(_sDeleteIpAllocations,3,(int)0 /*ZT_IP_ASSIGNMENT_TYPE_ADDRESS*/);
  469. if (sqlite3_step(_sDeleteIpAllocations) != SQLITE_DONE)
  470. return 500;
  471. for(unsigned int kk=0;kk<j->u.object.values[k].value->u.array.length;++kk) {
  472. json_value *ipalloc = j->u.object.values[k].value->u.array.values[kk];
  473. if (ipalloc->type == json_string) {
  474. InetAddress a(ipalloc->u.string.ptr);
  475. char ipBlob[16];
  476. int ipVersion = 0;
  477. _ipToBlob(a,ipBlob,ipVersion);
  478. if (ipVersion > 0) {
  479. sqlite3_reset(_sAllocateIp);
  480. sqlite3_bind_text(_sAllocateIp,1,nwids,16,SQLITE_STATIC);
  481. sqlite3_bind_text(_sAllocateIp,2,addrs,10,SQLITE_STATIC);
  482. sqlite3_bind_int(_sAllocateIp,3,(int)0 /*ZT_IP_ASSIGNMENT_TYPE_ADDRESS*/);
  483. sqlite3_bind_blob(_sAllocateIp,4,(const void *)ipBlob,16,SQLITE_STATIC);
  484. sqlite3_bind_int(_sAllocateIp,5,(int)a.netmaskBits());
  485. sqlite3_bind_int(_sAllocateIp,6,ipVersion);
  486. if (sqlite3_step(_sAllocateIp) != SQLITE_DONE)
  487. return 500;
  488. }
  489. }
  490. }
  491. addToNetworkRevision = 1;
  492. }
  493. } else if (!strcmp(j->u.object.values[k].name,"identity")) {
  494. // Identity is technically an immutable field, but if the member's Node has
  495. // no identity we allow it to be populated. This is primarily for migrating
  496. // node data from another controller.
  497. json_value *idstr = j->u.object.values[k].value;
  498. if (idstr->type == json_string) {
  499. bool alreadyHaveIdentity = false;
  500. sqlite3_reset(_sGetNodeIdentity);
  501. sqlite3_bind_text(_sGetNodeIdentity,1,addrs,10,SQLITE_STATIC);
  502. if (sqlite3_step(_sGetNodeIdentity) == SQLITE_ROW) {
  503. const char *tmp2 = (const char *)sqlite3_column_text(_sGetNodeIdentity,0);
  504. if ((tmp2)&&(tmp2[0]))
  505. alreadyHaveIdentity = true;
  506. }
  507. if (!alreadyHaveIdentity) {
  508. try {
  509. Identity id2(idstr->u.string.ptr);
  510. if (id2) {
  511. std::string idstr2(id2.toString(false)); // object must persist until after sqlite3_step() for SQLITE_STATIC
  512. sqlite3_reset(_sCreateOrReplaceNode);
  513. sqlite3_bind_text(_sCreateOrReplaceNode,1,addrs,10,SQLITE_STATIC);
  514. sqlite3_bind_text(_sCreateOrReplaceNode,2,idstr2.c_str(),-1,SQLITE_STATIC);
  515. sqlite3_step(_sCreateOrReplaceNode);
  516. }
  517. } catch ( ... ) {} // ignore invalid identities
  518. }
  519. }
  520. }
  521. }
  522. }
  523. json_value_free(j);
  524. }
  525. if ((addToNetworkRevision > 0)&&(revision > 0)) {
  526. sqlite3_reset(_sSetNetworkRevision);
  527. sqlite3_bind_int64(_sSetNetworkRevision,1,revision + addToNetworkRevision);
  528. sqlite3_bind_text(_sSetNetworkRevision,2,nwids,16,SQLITE_STATIC);
  529. sqlite3_step(_sSetNetworkRevision);
  530. }
  531. return _doCPGet(path,urlArgs,headers,body,responseBody,responseContentType);
  532. } else if ((path.size() == 3)&&(path[2] == "test")) {
  533. ZT_CircuitTest *test = (ZT_CircuitTest *)malloc(sizeof(ZT_CircuitTest));
  534. memset(test,0,sizeof(ZT_CircuitTest));
  535. Utils::getSecureRandom(&(test->testId),sizeof(test->testId));
  536. test->credentialNetworkId = nwid;
  537. test->ptr = (void *)this;
  538. json_value *j = json_parse(body.c_str(),body.length());
  539. if (j) {
  540. if (j->type == json_object) {
  541. for(unsigned int k=0;k<j->u.object.length;++k) {
  542. if (!strcmp(j->u.object.values[k].name,"hops")) {
  543. if (j->u.object.values[k].value->type == json_array) {
  544. for(unsigned int kk=0;kk<j->u.object.values[k].value->u.array.length;++kk) {
  545. json_value *hop = j->u.object.values[k].value->u.array.values[kk];
  546. if (hop->type == json_array) {
  547. for(unsigned int kkk=0;kkk<hop->u.array.length;++kkk) {
  548. if (hop->u.array.values[kkk]->type == json_string) {
  549. test->hops[test->hopCount].addresses[test->hops[test->hopCount].breadth++] = Utils::hexStrToU64(hop->u.array.values[kkk]->u.string.ptr) & 0xffffffffffULL;
  550. }
  551. }
  552. ++test->hopCount;
  553. }
  554. }
  555. }
  556. } else if (!strcmp(j->u.object.values[k].name,"reportAtEveryHop")) {
  557. if (j->u.object.values[k].value->type == json_boolean)
  558. test->reportAtEveryHop = (j->u.object.values[k].value->u.boolean == 0) ? 0 : 1;
  559. }
  560. }
  561. }
  562. json_value_free(j);
  563. }
  564. if (!test->hopCount) {
  565. ::free((void *)test);
  566. return 500;
  567. }
  568. test->timestamp = OSUtils::now();
  569. _CircuitTestEntry &te = _circuitTests[test->testId];
  570. te.test = test;
  571. te.jsonResults = "";
  572. _node->circuitTestBegin(test,&(SqliteNetworkController::_circuitTestCallback));
  573. char json[1024];
  574. Utils::snprintf(json,sizeof(json),"{\"testId\":\"%.16llx\"}",test->testId);
  575. responseBody = json;
  576. responseContentType = "application/json";
  577. return 200;
  578. } // else 404
  579. } else {
  580. std::vector<std::string> path_copy(path);
  581. if (!networkExists) {
  582. if (path[1].substr(10) == "______") {
  583. // A special POST /network/##########______ feature lets users create a network
  584. // with an arbitrary unused network number at this controller.
  585. nwid = 0;
  586. uint64_t nwidPrefix = (Utils::hexStrToU64(path[1].substr(0,10).c_str()) << 24) & 0xffffffffff000000ULL;
  587. uint64_t nwidPostfix = 0;
  588. Utils::getSecureRandom(&nwidPostfix,sizeof(nwidPostfix));
  589. uint64_t nwidOriginalPostfix = nwidPostfix;
  590. do {
  591. uint64_t tryNwid = nwidPrefix | (nwidPostfix & 0xffffffULL);
  592. if (!nwidPostfix)
  593. tryNwid |= 1;
  594. Utils::snprintf(nwids,sizeof(nwids),"%.16llx",(unsigned long long)tryNwid);
  595. sqlite3_reset(_sGetNetworkRevision);
  596. sqlite3_bind_text(_sGetNetworkRevision,1,nwids,16,SQLITE_STATIC);
  597. if (sqlite3_step(_sGetNetworkRevision) != SQLITE_ROW) {
  598. nwid = tryNwid;
  599. break;
  600. }
  601. ++nwidPostfix;
  602. } while (nwidPostfix != nwidOriginalPostfix);
  603. // 503 means we have no more free IDs for this prefix. You shouldn't host anywhere
  604. // near 16 million networks on the same controller, so shouldn't happen.
  605. if (!nwid)
  606. return 503;
  607. }
  608. sqlite3_reset(_sCreateNetwork);
  609. sqlite3_bind_text(_sCreateNetwork,1,nwids,16,SQLITE_STATIC);
  610. sqlite3_bind_text(_sCreateNetwork,2,"",0,SQLITE_STATIC);
  611. sqlite3_bind_int64(_sCreateNetwork,3,(long long)OSUtils::now());
  612. if (sqlite3_step(_sCreateNetwork) != SQLITE_DONE)
  613. return 500;
  614. path_copy[1].assign(nwids);
  615. }
  616. json_value *j = json_parse(body.c_str(),body.length());
  617. if (j) {
  618. if (j->type == json_object) {
  619. for(unsigned int k=0;k<j->u.object.length;++k) {
  620. sqlite3_stmt *stmt = (sqlite3_stmt *)0;
  621. if (!strcmp(j->u.object.values[k].name,"name")) {
  622. if ((j->u.object.values[k].value->type == json_string)&&(j->u.object.values[k].value->u.string.ptr[0])) {
  623. if (sqlite3_prepare_v2(_db,"UPDATE Network SET \"name\" = ? WHERE id = ?",-1,&stmt,(const char **)0) == SQLITE_OK)
  624. sqlite3_bind_text(stmt,1,j->u.object.values[k].value->u.string.ptr,-1,SQLITE_STATIC);
  625. }
  626. } else if (!strcmp(j->u.object.values[k].name,"private")) {
  627. if (j->u.object.values[k].value->type == json_boolean) {
  628. if (sqlite3_prepare_v2(_db,"UPDATE Network SET \"private\" = ? WHERE id = ?",-1,&stmt,(const char **)0) == SQLITE_OK)
  629. sqlite3_bind_int(stmt,1,(j->u.object.values[k].value->u.boolean == 0) ? 0 : 1);
  630. }
  631. } else if (!strcmp(j->u.object.values[k].name,"enableBroadcast")) {
  632. if (j->u.object.values[k].value->type == json_boolean) {
  633. if (sqlite3_prepare_v2(_db,"UPDATE Network SET enableBroadcast = ? WHERE id = ?",-1,&stmt,(const char **)0) == SQLITE_OK)
  634. sqlite3_bind_int(stmt,1,(j->u.object.values[k].value->u.boolean == 0) ? 0 : 1);
  635. }
  636. } else if (!strcmp(j->u.object.values[k].name,"allowPassiveBridging")) {
  637. if (j->u.object.values[k].value->type == json_boolean) {
  638. if (sqlite3_prepare_v2(_db,"UPDATE Network SET allowPassiveBridging = ? WHERE id = ?",-1,&stmt,(const char **)0) == SQLITE_OK)
  639. sqlite3_bind_int(stmt,1,(j->u.object.values[k].value->u.boolean == 0) ? 0 : 1);
  640. }
  641. } else if (!strcmp(j->u.object.values[k].name,"v4AssignMode")) {
  642. if (j->u.object.values[k].value->type == json_string) {
  643. if (sqlite3_prepare_v2(_db,"UPDATE Network SET v4AssignMode = ? WHERE id = ?",-1,&stmt,(const char **)0) == SQLITE_OK)
  644. sqlite3_bind_text(stmt,1,j->u.object.values[k].value->u.string.ptr,-1,SQLITE_STATIC);
  645. }
  646. } else if (!strcmp(j->u.object.values[k].name,"v6AssignMode")) {
  647. if (j->u.object.values[k].value->type == json_string) {
  648. if (sqlite3_prepare_v2(_db,"UPDATE Network SET v6AssignMode = ? WHERE id = ?",-1,&stmt,(const char **)0) == SQLITE_OK)
  649. sqlite3_bind_text(stmt,1,j->u.object.values[k].value->u.string.ptr,-1,SQLITE_STATIC);
  650. }
  651. } else if (!strcmp(j->u.object.values[k].name,"multicastLimit")) {
  652. if (j->u.object.values[k].value->type == json_integer) {
  653. if (sqlite3_prepare_v2(_db,"UPDATE Network SET multicastLimit = ? WHERE id = ?",-1,&stmt,(const char **)0) == SQLITE_OK)
  654. sqlite3_bind_int(stmt,1,(int)j->u.object.values[k].value->u.integer);
  655. }
  656. } else if (!strcmp(j->u.object.values[k].name,"relays")) {
  657. if (j->u.object.values[k].value->type == json_array) {
  658. std::map<Address,InetAddress> nodeIdToPhyAddress;
  659. for(unsigned int kk=0;kk<j->u.object.values[k].value->u.array.length;++kk) {
  660. json_value *relay = j->u.object.values[k].value->u.array.values[kk];
  661. const char *address = (const char *)0;
  662. const char *phyAddress = (const char *)0;
  663. if ((relay)&&(relay->type == json_object)) {
  664. for(unsigned int rk=0;rk<relay->u.object.length;++rk) {
  665. if ((!strcmp(relay->u.object.values[rk].name,"address"))&&(relay->u.object.values[rk].value->type == json_string))
  666. address = relay->u.object.values[rk].value->u.string.ptr;
  667. else if ((!strcmp(relay->u.object.values[rk].name,"phyAddress"))&&(relay->u.object.values[rk].value->type == json_string))
  668. phyAddress = relay->u.object.values[rk].value->u.string.ptr;
  669. }
  670. }
  671. if ((address)&&(phyAddress))
  672. nodeIdToPhyAddress[Address(address)] = InetAddress(phyAddress);
  673. }
  674. sqlite3_reset(_sDeleteRelaysForNetwork);
  675. sqlite3_bind_text(_sDeleteRelaysForNetwork,1,nwids,16,SQLITE_STATIC);
  676. sqlite3_step(_sDeleteRelaysForNetwork);
  677. for(std::map<Address,InetAddress>::iterator rl(nodeIdToPhyAddress.begin());rl!=nodeIdToPhyAddress.end();++rl) {
  678. sqlite3_reset(_sCreateRelay);
  679. sqlite3_bind_text(_sCreateRelay,1,nwids,16,SQLITE_STATIC);
  680. std::string a(rl->first.toString()),b(rl->second.toString()); // don't destroy strings until sqlite3_step()
  681. sqlite3_bind_text(_sCreateRelay,2,a.c_str(),-1,SQLITE_STATIC);
  682. sqlite3_bind_text(_sCreateRelay,3,b.c_str(),-1,SQLITE_STATIC);
  683. sqlite3_step(_sCreateRelay);
  684. }
  685. }
  686. } else if (!strcmp(j->u.object.values[k].name,"routes")) {
  687. sqlite3_reset(_sDeleteRoutes);
  688. sqlite3_bind_text(_sDeleteRoutes,1,nwids,16,SQLITE_STATIC);
  689. sqlite3_step(_sDeleteRoutes);
  690. if (j->u.object.values[k].value->type == json_array) {
  691. for(unsigned int kk=0;kk<j->u.object.values[k].value->u.array.length;++kk) {
  692. json_value *r = j->u.object.values[k].value->u.array.values[kk];
  693. if ((r)&&(r->type == json_object)) {
  694. InetAddress r_target,r_via;
  695. int r_flags = 0;
  696. int r_metric = 0;
  697. for(unsigned int rk=0;rk<r->u.object.length;++rk) {
  698. if ((!strcmp(r->u.object.values[rk].name,"target"))&&(r->u.object.values[rk].value->type == json_string))
  699. r_target = InetAddress(std::string(r->u.object.values[rk].value->u.string.ptr));
  700. else if ((!strcmp(r->u.object.values[rk].name,"via"))&&(r->u.object.values[rk].value->type == json_string))
  701. r_via = InetAddress(std::string(r->u.object.values[rk].value->u.string.ptr),0);
  702. else if ((!strcmp(r->u.object.values[rk].name,"flags"))&&(r->u.object.values[rk].value->type == json_integer))
  703. r_flags = (int)(r->u.object.values[rk].value->u.integer & 0xffff);
  704. else if ((!strcmp(r->u.object.values[rk].name,"metric"))&&(r->u.object.values[rk].value->type == json_integer))
  705. r_metric = (int)(r->u.object.values[rk].value->u.integer & 0xffff);
  706. }
  707. if ((r_target)&&((!r_via)||(r_via.ss_family == r_target.ss_family))) {
  708. int r_ipVersion = 0;
  709. char r_targetBlob[16];
  710. char r_viaBlob[16];
  711. _ipToBlob(r_target,r_targetBlob,r_ipVersion);
  712. if (r_ipVersion) {
  713. int r_targetNetmaskBits = r_target.netmaskBits();
  714. if ((r_ipVersion == 4)&&(r_targetNetmaskBits > 32)) r_targetNetmaskBits = 32;
  715. else if ((r_ipVersion == 6)&&(r_targetNetmaskBits > 128)) r_targetNetmaskBits = 128;
  716. sqlite3_reset(_sCreateRoute);
  717. sqlite3_bind_text(_sCreateRoute,1,nwids,16,SQLITE_STATIC);
  718. sqlite3_bind_blob(_sCreateRoute,2,(const void *)r_targetBlob,16,SQLITE_STATIC);
  719. if (r_via) {
  720. _ipToBlob(r_via,r_viaBlob,r_ipVersion);
  721. sqlite3_bind_blob(_sCreateRoute,3,(const void *)r_viaBlob,16,SQLITE_STATIC);
  722. } else {
  723. sqlite3_bind_null(_sCreateRoute,3);
  724. }
  725. sqlite3_bind_int(_sCreateRoute,4,r_targetNetmaskBits);
  726. sqlite3_bind_int(_sCreateRoute,5,r_ipVersion);
  727. sqlite3_bind_int(_sCreateRoute,6,r_flags);
  728. sqlite3_bind_int(_sCreateRoute,7,r_metric);
  729. sqlite3_step(_sCreateRoute);
  730. }
  731. }
  732. }
  733. }
  734. }
  735. } else if (!strcmp(j->u.object.values[k].name,"ipAssignmentPools")) {
  736. if (j->u.object.values[k].value->type == json_array) {
  737. std::vector< std::pair<InetAddress,InetAddress> > pools;
  738. for(unsigned int kk=0;kk<j->u.object.values[k].value->u.array.length;++kk) {
  739. json_value *pool = j->u.object.values[k].value->u.array.values[kk];
  740. const char *iprs = (const char *)0;
  741. const char *ipre = (const char *)0;
  742. if ((pool)&&(pool->type == json_object)) {
  743. for(unsigned int rk=0;rk<pool->u.object.length;++rk) {
  744. if ((!strcmp(pool->u.object.values[rk].name,"ipRangeStart"))&&(pool->u.object.values[rk].value->type == json_string))
  745. iprs = pool->u.object.values[rk].value->u.string.ptr;
  746. else if ((!strcmp(pool->u.object.values[rk].name,"ipRangeEnd"))&&(pool->u.object.values[rk].value->type == json_string))
  747. ipre = pool->u.object.values[rk].value->u.string.ptr;
  748. }
  749. }
  750. if ((iprs)&&(ipre)) {
  751. InetAddress iprs2(iprs);
  752. InetAddress ipre2(ipre);
  753. if (iprs2.ss_family == ipre2.ss_family) {
  754. iprs2.setPort(0);
  755. ipre2.setPort(0);
  756. pools.push_back(std::pair<InetAddress,InetAddress>(iprs2,ipre2));
  757. }
  758. }
  759. }
  760. std::sort(pools.begin(),pools.end());
  761. pools.erase(std::unique(pools.begin(),pools.end()),pools.end());
  762. sqlite3_reset(_sDeleteIpAssignmentPoolsForNetwork);
  763. sqlite3_bind_text(_sDeleteIpAssignmentPoolsForNetwork,1,nwids,16,SQLITE_STATIC);
  764. sqlite3_step(_sDeleteIpAssignmentPoolsForNetwork);
  765. for(std::vector< std::pair<InetAddress,InetAddress> >::const_iterator p(pools.begin());p!=pools.end();++p) {
  766. char ipBlob1[16],ipBlob2[16];
  767. sqlite3_reset(_sCreateIpAssignmentPool);
  768. sqlite3_bind_text(_sCreateIpAssignmentPool,1,nwids,16,SQLITE_STATIC);
  769. if (p->first.ss_family == AF_INET) {
  770. memset(ipBlob1,0,12);
  771. memcpy(ipBlob1 + 12,p->first.rawIpData(),4);
  772. memset(ipBlob2,0,12);
  773. memcpy(ipBlob2 + 12,p->second.rawIpData(),4);
  774. sqlite3_bind_blob(_sCreateIpAssignmentPool,2,(const void *)ipBlob1,16,SQLITE_STATIC);
  775. sqlite3_bind_blob(_sCreateIpAssignmentPool,3,(const void *)ipBlob2,16,SQLITE_STATIC);
  776. sqlite3_bind_int(_sCreateIpAssignmentPool,4,4);
  777. } else if (p->first.ss_family == AF_INET6) {
  778. sqlite3_bind_blob(_sCreateIpAssignmentPool,2,p->first.rawIpData(),16,SQLITE_STATIC);
  779. sqlite3_bind_blob(_sCreateIpAssignmentPool,3,p->second.rawIpData(),16,SQLITE_STATIC);
  780. sqlite3_bind_int(_sCreateIpAssignmentPool,4,6);
  781. } else continue;
  782. sqlite3_step(_sCreateIpAssignmentPool);
  783. }
  784. }
  785. } else if (!strcmp(j->u.object.values[k].name,"rules")) {
  786. if (j->u.object.values[k].value->type == json_array) {
  787. sqlite3_reset(_sDeleteRulesForNetwork);
  788. sqlite3_bind_text(_sDeleteRulesForNetwork,1,nwids,16,SQLITE_STATIC);
  789. sqlite3_step(_sDeleteRulesForNetwork);
  790. for(unsigned int kk=0;kk<j->u.object.values[k].value->u.array.length;++kk) {
  791. json_value *rj = j->u.object.values[k].value->u.array.values[kk];
  792. if ((rj)&&(rj->type == json_object)) {
  793. struct { // NULL pointers indicate missing or NULL -- wildcards
  794. const json_int_t *ruleNo;
  795. const char *nodeId;
  796. const char *sourcePort;
  797. const char *destPort;
  798. const json_int_t *vlanId;
  799. const json_int_t *vlanPcp;
  800. const json_int_t *etherType;
  801. const char *macSource;
  802. const char *macDest;
  803. const char *ipSource;
  804. const char *ipDest;
  805. const json_int_t *ipTos;
  806. const json_int_t *ipProtocol;
  807. const json_int_t *ipSourcePort;
  808. const json_int_t *ipDestPort;
  809. const json_int_t *flags;
  810. const json_int_t *invFlags;
  811. const char *action;
  812. } rule;
  813. memset(&rule,0,sizeof(rule));
  814. for(unsigned int rk=0;rk<rj->u.object.length;++rk) {
  815. if ((!strcmp(rj->u.object.values[rk].name,"ruleNo"))&&(rj->u.object.values[rk].value->type == json_integer))
  816. rule.ruleNo = &(rj->u.object.values[rk].value->u.integer);
  817. else if ((!strcmp(rj->u.object.values[rk].name,"nodeId"))&&(rj->u.object.values[rk].value->type == json_string))
  818. rule.nodeId = rj->u.object.values[rk].value->u.string.ptr;
  819. else if ((!strcmp(rj->u.object.values[rk].name,"sourcePort"))&&(rj->u.object.values[rk].value->type == json_string))
  820. rule.sourcePort = rj->u.object.values[rk].value->u.string.ptr;
  821. else if ((!strcmp(rj->u.object.values[rk].name,"destPort"))&&(rj->u.object.values[rk].value->type == json_string))
  822. rule.destPort = rj->u.object.values[rk].value->u.string.ptr;
  823. else if ((!strcmp(rj->u.object.values[rk].name,"vlanId"))&&(rj->u.object.values[rk].value->type == json_integer))
  824. rule.vlanId = &(rj->u.object.values[rk].value->u.integer);
  825. else if ((!strcmp(rj->u.object.values[rk].name,"vlanPcp"))&&(rj->u.object.values[rk].value->type == json_integer))
  826. rule.vlanPcp = &(rj->u.object.values[rk].value->u.integer);
  827. else if ((!strcmp(rj->u.object.values[rk].name,"etherType"))&&(rj->u.object.values[rk].value->type == json_integer))
  828. rule.etherType = &(rj->u.object.values[rk].value->u.integer);
  829. else if ((!strcmp(rj->u.object.values[rk].name,"macSource"))&&(rj->u.object.values[rk].value->type == json_string))
  830. rule.macSource = rj->u.object.values[rk].value->u.string.ptr;
  831. else if ((!strcmp(rj->u.object.values[rk].name,"macDest"))&&(rj->u.object.values[rk].value->type == json_string))
  832. rule.macDest = rj->u.object.values[rk].value->u.string.ptr;
  833. else if ((!strcmp(rj->u.object.values[rk].name,"ipSource"))&&(rj->u.object.values[rk].value->type == json_string))
  834. rule.ipSource = rj->u.object.values[rk].value->u.string.ptr;
  835. else if ((!strcmp(rj->u.object.values[rk].name,"ipDest"))&&(rj->u.object.values[rk].value->type == json_string))
  836. rule.ipDest = rj->u.object.values[rk].value->u.string.ptr;
  837. else if ((!strcmp(rj->u.object.values[rk].name,"ipTos"))&&(rj->u.object.values[rk].value->type == json_integer))
  838. rule.ipTos = &(rj->u.object.values[rk].value->u.integer);
  839. else if ((!strcmp(rj->u.object.values[rk].name,"ipProtocol"))&&(rj->u.object.values[rk].value->type == json_integer))
  840. rule.ipProtocol = &(rj->u.object.values[rk].value->u.integer);
  841. else if ((!strcmp(rj->u.object.values[rk].name,"ipSourcePort"))&&(rj->u.object.values[rk].value->type == json_integer))
  842. rule.ipSourcePort = &(rj->u.object.values[rk].value->u.integer);
  843. else if ((!strcmp(rj->u.object.values[rk].name,"ipDestPort"))&&(rj->u.object.values[rk].value->type == json_integer))
  844. rule.ipDestPort = &(rj->u.object.values[rk].value->u.integer);
  845. else if ((!strcmp(rj->u.object.values[rk].name,"flags"))&&(rj->u.object.values[rk].value->type == json_integer))
  846. rule.flags = &(rj->u.object.values[rk].value->u.integer);
  847. else if ((!strcmp(rj->u.object.values[rk].name,"invFlags"))&&(rj->u.object.values[rk].value->type == json_integer))
  848. rule.invFlags = &(rj->u.object.values[rk].value->u.integer);
  849. else if ((!strcmp(rj->u.object.values[rk].name,"action"))&&(rj->u.object.values[rk].value->type == json_string))
  850. rule.action = rj->u.object.values[rk].value->u.string.ptr;
  851. }
  852. if ((rule.ruleNo)&&(rule.action)&&(rule.action[0])) {
  853. char mactmp1[16],mactmp2[16];
  854. sqlite3_reset(_sCreateRule);
  855. sqlite3_bind_text(_sCreateRule,1,nwids,16,SQLITE_STATIC);
  856. sqlite3_bind_int64(_sCreateRule,2,*rule.ruleNo);
  857. // Optional values: null by default
  858. for(int i=3;i<=18;++i)
  859. sqlite3_bind_null(_sCreateRule,i);
  860. if ((rule.nodeId)&&(strlen(rule.nodeId) == 10)) sqlite3_bind_text(_sCreateRule,3,rule.nodeId,10,SQLITE_STATIC);
  861. if ((rule.sourcePort)&&(strlen(rule.sourcePort) == 10)) sqlite3_bind_text(_sCreateRule,4,rule.sourcePort,10,SQLITE_STATIC);
  862. if ((rule.destPort)&&(strlen(rule.destPort) == 10)) sqlite3_bind_text(_sCreateRule,5,rule.destPort,10,SQLITE_STATIC);
  863. if (rule.vlanId) sqlite3_bind_int(_sCreateRule,6,(int)*rule.vlanId);
  864. if (rule.vlanPcp) sqlite3_bind_int(_sCreateRule,7,(int)*rule.vlanPcp);
  865. if (rule.etherType) sqlite3_bind_int(_sCreateRule,8,(int)*rule.etherType & (int)0xffff);
  866. if (rule.macSource) {
  867. MAC m(rule.macSource);
  868. Utils::snprintf(mactmp1,sizeof(mactmp1),"%.12llx",(unsigned long long)m.toInt());
  869. sqlite3_bind_text(_sCreateRule,9,mactmp1,-1,SQLITE_STATIC);
  870. }
  871. if (rule.macDest) {
  872. MAC m(rule.macDest);
  873. Utils::snprintf(mactmp2,sizeof(mactmp2),"%.12llx",(unsigned long long)m.toInt());
  874. sqlite3_bind_text(_sCreateRule,10,mactmp2,-1,SQLITE_STATIC);
  875. }
  876. if (rule.ipSource) sqlite3_bind_text(_sCreateRule,11,rule.ipSource,-1,SQLITE_STATIC);
  877. if (rule.ipDest) sqlite3_bind_text(_sCreateRule,12,rule.ipDest,-1,SQLITE_STATIC);
  878. if (rule.ipTos) sqlite3_bind_int(_sCreateRule,13,(int)*rule.ipTos);
  879. if (rule.ipProtocol) sqlite3_bind_int(_sCreateRule,14,(int)*rule.ipProtocol);
  880. if (rule.ipSourcePort) sqlite3_bind_int(_sCreateRule,15,(int)*rule.ipSourcePort & (int)0xffff);
  881. if (rule.ipDestPort) sqlite3_bind_int(_sCreateRule,16,(int)*rule.ipDestPort & (int)0xffff);
  882. if (rule.flags) sqlite3_bind_int64(_sCreateRule,17,(int64_t)*rule.flags);
  883. if (rule.invFlags) sqlite3_bind_int64(_sCreateRule,18,(int64_t)*rule.invFlags);
  884. sqlite3_bind_text(_sCreateRule,19,rule.action,-1,SQLITE_STATIC);
  885. sqlite3_step(_sCreateRule);
  886. }
  887. }
  888. }
  889. }
  890. }
  891. if (stmt) {
  892. sqlite3_bind_text(stmt,2,nwids,16,SQLITE_STATIC);
  893. sqlite3_step(stmt);
  894. sqlite3_finalize(stmt);
  895. }
  896. }
  897. }
  898. json_value_free(j);
  899. }
  900. sqlite3_reset(_sSetNetworkRevision);
  901. sqlite3_bind_int64(_sSetNetworkRevision,1,revision += 1);
  902. sqlite3_bind_text(_sSetNetworkRevision,2,nwids,16,SQLITE_STATIC);
  903. sqlite3_step(_sSetNetworkRevision);
  904. return _doCPGet(path_copy,urlArgs,headers,body,responseBody,responseContentType);
  905. }
  906. } // else 404
  907. } // else 404
  908. return 404;
  909. }
  910. unsigned int SqliteNetworkController::handleControlPlaneHttpDELETE(
  911. const std::vector<std::string> &path,
  912. const std::map<std::string,std::string> &urlArgs,
  913. const std::map<std::string,std::string> &headers,
  914. const std::string &body,
  915. std::string &responseBody,
  916. std::string &responseContentType)
  917. {
  918. if (path.empty())
  919. return 404;
  920. Mutex::Lock _l(_lock);
  921. if (path[0] == "network") {
  922. if ((path.size() >= 2)&&(path[1].length() == 16)) {
  923. uint64_t nwid = Utils::hexStrToU64(path[1].c_str());
  924. char nwids[24];
  925. Utils::snprintf(nwids,sizeof(nwids),"%.16llx",(unsigned long long)nwid);
  926. sqlite3_reset(_sGetNetworkById);
  927. sqlite3_bind_text(_sGetNetworkById,1,nwids,16,SQLITE_STATIC);
  928. if (sqlite3_step(_sGetNetworkById) != SQLITE_ROW)
  929. return 404;
  930. if (path.size() >= 3) {
  931. if ((path.size() == 4)&&(path[2] == "member")&&(path[3].length() == 10)) {
  932. uint64_t address = Utils::hexStrToU64(path[3].c_str());
  933. char addrs[24];
  934. Utils::snprintf(addrs,sizeof(addrs),"%.10llx",address);
  935. sqlite3_reset(_sGetMember);
  936. sqlite3_bind_text(_sGetMember,1,nwids,16,SQLITE_STATIC);
  937. sqlite3_bind_text(_sGetMember,2,addrs,10,SQLITE_STATIC);
  938. if (sqlite3_step(_sGetMember) != SQLITE_ROW)
  939. return 404;
  940. sqlite3_reset(_sDeleteIpAllocations);
  941. sqlite3_bind_text(_sDeleteIpAllocations,1,nwids,16,SQLITE_STATIC);
  942. sqlite3_bind_text(_sDeleteIpAllocations,2,addrs,10,SQLITE_STATIC);
  943. sqlite3_bind_int(_sDeleteIpAllocations,3,(int)0 /*ZT_IP_ASSIGNMENT_TYPE_ADDRESS*/);
  944. if (sqlite3_step(_sDeleteIpAllocations) == SQLITE_DONE) {
  945. sqlite3_reset(_sDeleteMember);
  946. sqlite3_bind_text(_sDeleteMember,1,nwids,16,SQLITE_STATIC);
  947. sqlite3_bind_text(_sDeleteMember,2,addrs,10,SQLITE_STATIC);
  948. if (sqlite3_step(_sDeleteMember) != SQLITE_DONE)
  949. return 500;
  950. } else return 500;
  951. return 200;
  952. }
  953. } else {
  954. sqlite3_reset(_sDeleteNetwork);
  955. sqlite3_bind_text(_sDeleteNetwork,1,nwids,16,SQLITE_STATIC);
  956. if (sqlite3_step(_sDeleteNetwork) == SQLITE_DONE) {
  957. sqlite3_reset(_sDeleteAllNetworkMembers);
  958. sqlite3_bind_text(_sDeleteAllNetworkMembers,1,nwids,16,SQLITE_STATIC);
  959. sqlite3_step(_sDeleteAllNetworkMembers);
  960. return 200;
  961. } else return 500;
  962. }
  963. } // else 404
  964. } // else 404
  965. return 404;
  966. }
  967. void SqliteNetworkController::threadMain()
  968. throw()
  969. {
  970. uint64_t lastBackupTime = OSUtils::now();
  971. uint64_t lastCleanupTime = OSUtils::now();
  972. while (_backupThreadRun) {
  973. if ((OSUtils::now() - lastCleanupTime) >= 5000) {
  974. const uint64_t now = OSUtils::now();
  975. lastCleanupTime = now;
  976. Mutex::Lock _l(_lock);
  977. // Clean out really old circuit tests to prevent memory build-up
  978. for(std::map< uint64_t,_CircuitTestEntry >::iterator ct(_circuitTests.begin());ct!=_circuitTests.end();) {
  979. if (!ct->second.test) {
  980. _circuitTests.erase(ct++);
  981. } else if ((now - ct->second.test->timestamp) >= ZT_SQLITENETWORKCONTROLLER_CIRCUIT_TEST_TIMEOUT) {
  982. _node->circuitTestEnd(ct->second.test);
  983. ::free((void *)ct->second.test);
  984. _circuitTests.erase(ct++);
  985. } else ++ct;
  986. }
  987. }
  988. if ((OSUtils::now() - lastBackupTime) >= ZT_NETCONF_BACKUP_PERIOD) {
  989. lastBackupTime = OSUtils::now();
  990. char backupPath[4096],backupPath2[4096];
  991. Utils::snprintf(backupPath,sizeof(backupPath),"%s.backupInProgress",_dbPath.c_str());
  992. Utils::snprintf(backupPath2,sizeof(backupPath),"%s.backup",_dbPath.c_str());
  993. OSUtils::rm(backupPath); // delete any unfinished backups
  994. sqlite3 *bakdb = (sqlite3 *)0;
  995. sqlite3_backup *bak = (sqlite3_backup *)0;
  996. if (sqlite3_open_v2(backupPath,&bakdb,SQLITE_OPEN_READWRITE|SQLITE_OPEN_CREATE,(const char *)0) != SQLITE_OK) {
  997. fprintf(stderr,"SqliteNetworkController: CRITICAL: backup failed on sqlite3_open_v2()"ZT_EOL_S);
  998. continue;
  999. }
  1000. bak = sqlite3_backup_init(bakdb,"main",_db,"main");
  1001. if (!bak) {
  1002. sqlite3_close(bakdb);
  1003. OSUtils::rm(backupPath); // delete any unfinished backups
  1004. fprintf(stderr,"SqliteNetworkController: CRITICAL: backup failed on sqlite3_backup_init()"ZT_EOL_S);
  1005. continue;
  1006. }
  1007. int rc = SQLITE_OK;
  1008. for(;;) {
  1009. if (!_backupThreadRun) {
  1010. sqlite3_backup_finish(bak);
  1011. sqlite3_close(bakdb);
  1012. OSUtils::rm(backupPath);
  1013. return;
  1014. }
  1015. _lock.lock();
  1016. rc = sqlite3_backup_step(bak,64);
  1017. _lock.unlock();
  1018. if ((rc == SQLITE_OK)||(rc == SQLITE_LOCKED)||(rc == SQLITE_BUSY))
  1019. Thread::sleep(50);
  1020. else break;
  1021. }
  1022. sqlite3_backup_finish(bak);
  1023. sqlite3_close(bakdb);
  1024. OSUtils::rm(backupPath2);
  1025. ::rename(backupPath,backupPath2);
  1026. }
  1027. Thread::sleep(250);
  1028. }
  1029. }
  1030. unsigned int SqliteNetworkController::_doCPGet(
  1031. const std::vector<std::string> &path,
  1032. const std::map<std::string,std::string> &urlArgs,
  1033. const std::map<std::string,std::string> &headers,
  1034. const std::string &body,
  1035. std::string &responseBody,
  1036. std::string &responseContentType)
  1037. {
  1038. // Assumes _lock is locked
  1039. char json[65536];
  1040. if ((path.size() > 0)&&(path[0] == "network")) {
  1041. if ((path.size() >= 2)&&(path[1].length() == 16)) {
  1042. uint64_t nwid = Utils::hexStrToU64(path[1].c_str());
  1043. char nwids[24];
  1044. Utils::snprintf(nwids,sizeof(nwids),"%.16llx",(unsigned long long)nwid);
  1045. if (path.size() >= 3) {
  1046. // /network/<nwid>/...
  1047. if (path[2] == "member") {
  1048. if (path.size() >= 4) {
  1049. // Get specific member info
  1050. uint64_t address = Utils::hexStrToU64(path[3].c_str());
  1051. char addrs[24];
  1052. Utils::snprintf(addrs,sizeof(addrs),"%.10llx",address);
  1053. sqlite3_reset(_sGetMember2);
  1054. sqlite3_bind_text(_sGetMember2,1,nwids,16,SQLITE_STATIC);
  1055. sqlite3_bind_text(_sGetMember2,2,addrs,10,SQLITE_STATIC);
  1056. if (sqlite3_step(_sGetMember2) == SQLITE_ROW) {
  1057. const char *memberIdStr = (const char *)sqlite3_column_text(_sGetMember2,3);
  1058. Utils::snprintf(json,sizeof(json),
  1059. "{\n"
  1060. "\t\"nwid\": \"%s\",\n"
  1061. "\t\"address\": \"%s\",\n"
  1062. "\t\"controllerInstanceId\": \"%s\",\n"
  1063. "\t\"authorized\": %s,\n"
  1064. "\t\"activeBridge\": %s,\n"
  1065. "\t\"memberRevision\": %llu,\n"
  1066. "\t\"clock\": %llu,\n"
  1067. "\t\"identity\": \"%s\",\n"
  1068. "\t\"ipAssignments\": [",
  1069. nwids,
  1070. addrs,
  1071. _instanceId.c_str(),
  1072. (sqlite3_column_int(_sGetMember2,0) > 0) ? "true" : "false",
  1073. (sqlite3_column_int(_sGetMember2,1) > 0) ? "true" : "false",
  1074. (unsigned long long)sqlite3_column_int64(_sGetMember2,2),
  1075. (unsigned long long)OSUtils::now(),
  1076. _jsonEscape(memberIdStr).c_str());
  1077. responseBody = json;
  1078. sqlite3_reset(_sGetIpAssignmentsForNode2);
  1079. sqlite3_bind_text(_sGetIpAssignmentsForNode2,1,nwids,16,SQLITE_STATIC);
  1080. sqlite3_bind_text(_sGetIpAssignmentsForNode2,2,addrs,10,SQLITE_STATIC);
  1081. sqlite3_bind_int(_sGetIpAssignmentsForNode2,3,(int)0 /*ZT_IP_ASSIGNMENT_TYPE_ADDRESS*/);
  1082. bool firstIp = true;
  1083. while (sqlite3_step(_sGetIpAssignmentsForNode2) == SQLITE_ROW) {
  1084. int ipversion = sqlite3_column_int(_sGetIpAssignmentsForNode2,2);
  1085. char ipBlob[16];
  1086. memcpy(ipBlob,(const void *)sqlite3_column_blob(_sGetIpAssignmentsForNode2,0),16);
  1087. InetAddress ip(
  1088. (const void *)(ipversion == 6 ? ipBlob : &ipBlob[12]),
  1089. (ipversion == 6 ? 16 : 4),
  1090. (unsigned int)sqlite3_column_int(_sGetIpAssignmentsForNode2,1)
  1091. );
  1092. responseBody.append(firstIp ? "\"" : ",\"");
  1093. responseBody.append(_jsonEscape(ip.toString()));
  1094. responseBody.push_back('"');
  1095. firstIp = false;
  1096. }
  1097. responseBody.append("],\n\t\"recentLog\": [");
  1098. sqlite3_reset(_sGetNodeHistory);
  1099. sqlite3_bind_text(_sGetNodeHistory,1,nwids,16,SQLITE_STATIC);
  1100. sqlite3_bind_text(_sGetNodeHistory,2,addrs,10,SQLITE_STATIC);
  1101. bool firstHistory = true;
  1102. while (sqlite3_step(_sGetNodeHistory) == SQLITE_ROW) {
  1103. responseBody.append(firstHistory ? "{" : ",{");
  1104. responseBody.append("\"ts\":");
  1105. responseBody.append((const char *)sqlite3_column_text(_sGetNodeHistory,2));
  1106. responseBody.append((sqlite3_column_int(_sGetNodeHistory,1) == 0) ? ",\"authorized\":false,\"clientMajorVersion\":" : ",\"authorized\":true,\"clientMajorVersion\":");
  1107. responseBody.append((const char *)sqlite3_column_text(_sGetNodeHistory,3));
  1108. responseBody.append(",\"clientMinorVersion\":");
  1109. responseBody.append((const char *)sqlite3_column_text(_sGetNodeHistory,4));
  1110. responseBody.append(",\"clientRevision\":");
  1111. responseBody.append((const char *)sqlite3_column_text(_sGetNodeHistory,5));
  1112. responseBody.append(",\"fromAddr\":");
  1113. const char *fa = (const char *)sqlite3_column_text(_sGetNodeHistory,7);
  1114. if (fa) {
  1115. responseBody.push_back('"');
  1116. responseBody.append(_jsonEscape(fa));
  1117. responseBody.append("\"}");
  1118. } else responseBody.append("null}");
  1119. firstHistory = false;
  1120. }
  1121. responseBody.append("]\n}\n");
  1122. responseContentType = "application/json";
  1123. return 200;
  1124. } // else 404
  1125. } else {
  1126. // List members
  1127. sqlite3_reset(_sListNetworkMembers);
  1128. sqlite3_bind_text(_sListNetworkMembers,1,nwids,16,SQLITE_STATIC);
  1129. responseBody.push_back('{');
  1130. bool firstMember = true;
  1131. while (sqlite3_step(_sListNetworkMembers) == SQLITE_ROW) {
  1132. responseBody.append(firstMember ? "\"" : ",\"");
  1133. firstMember = false;
  1134. responseBody.append((const char *)sqlite3_column_text(_sListNetworkMembers,0));
  1135. responseBody.append("\":");
  1136. responseBody.append((const char *)sqlite3_column_text(_sListNetworkMembers,1));
  1137. }
  1138. responseBody.push_back('}');
  1139. responseContentType = "application/json";
  1140. return 200;
  1141. }
  1142. } else if ((path[2] == "active")&&(path.size() == 3)) {
  1143. sqlite3_reset(_sGetActiveNodesOnNetwork);
  1144. sqlite3_bind_text(_sGetActiveNodesOnNetwork,1,nwids,16,SQLITE_STATIC);
  1145. sqlite3_bind_text(_sGetActiveNodesOnNetwork,2,nwids,16,SQLITE_STATIC);
  1146. sqlite3_bind_int64(_sGetActiveNodesOnNetwork,3,(int64_t)(OSUtils::now() - ZT_NETCONF_NODE_ACTIVE_THRESHOLD));
  1147. responseBody.push_back('{');
  1148. bool firstMember = true;
  1149. uint64_t lastNodeId = 0;
  1150. while (sqlite3_step(_sGetActiveNodesOnNetwork) == SQLITE_ROW) {
  1151. const char *nodeId = (const char *)sqlite3_column_text(_sGetActiveNodesOnNetwork,0);
  1152. if (nodeId) {
  1153. const uint64_t nodeIdInt = Utils::hexStrToU64(nodeId);
  1154. if (nodeIdInt == lastNodeId) // technically that SQL query could (rarely) generate a duplicate for a given nodeId, in which case we want the first
  1155. continue;
  1156. lastNodeId = nodeIdInt;
  1157. responseBody.append(firstMember ? "\"" : ",\"");
  1158. firstMember = false;
  1159. responseBody.append(nodeId);
  1160. responseBody.append("\":{");
  1161. responseBody.append("\"ts\":");
  1162. responseBody.append((const char *)sqlite3_column_text(_sGetActiveNodesOnNetwork,1));
  1163. responseBody.append((sqlite3_column_int(_sGetActiveNodesOnNetwork,6) > 0) ? ",\"authorized\":true" : ",\"authorized\":false");
  1164. responseBody.append(",\"clientMajorVersion\":");
  1165. responseBody.append((const char *)sqlite3_column_text(_sGetActiveNodesOnNetwork,2));
  1166. responseBody.append(",\"clientMinorVersion\":");
  1167. responseBody.append((const char *)sqlite3_column_text(_sGetActiveNodesOnNetwork,3));
  1168. responseBody.append(",\"clientRevision\":");
  1169. responseBody.append((const char *)sqlite3_column_text(_sGetActiveNodesOnNetwork,4));
  1170. const char *fromAddr = (const char *)sqlite3_column_text(_sGetActiveNodesOnNetwork,5);
  1171. if ((fromAddr)&&(fromAddr[0])) {
  1172. responseBody.append(",\"fromAddr\":\"");
  1173. responseBody.append(_jsonEscape(fromAddr));
  1174. responseBody.append("\"}");
  1175. } else {
  1176. responseBody.append(",\"fromAddr\":null}");
  1177. }
  1178. }
  1179. }
  1180. responseBody.push_back('}');
  1181. responseContentType = "application/json";
  1182. return 200;
  1183. } else if ((path[2] == "test")&&(path.size() >= 4)) {
  1184. std::map< uint64_t,_CircuitTestEntry >::iterator cte(_circuitTests.find(Utils::hexStrToU64(path[3].c_str())));
  1185. if ((cte != _circuitTests.end())&&(cte->second.test)) {
  1186. responseBody = "[";
  1187. responseBody.append(cte->second.jsonResults);
  1188. responseBody.push_back(']');
  1189. responseContentType = "application/json";
  1190. return 200;
  1191. } // else 404
  1192. } // else 404
  1193. } else {
  1194. sqlite3_reset(_sGetNetworkById);
  1195. sqlite3_bind_text(_sGetNetworkById,1,nwids,16,SQLITE_STATIC);
  1196. if (sqlite3_step(_sGetNetworkById) == SQLITE_ROW) {
  1197. Utils::snprintf(json,sizeof(json),
  1198. "{\n"
  1199. "\t\"nwid\": \"%s\",\n"
  1200. "\t\"controllerInstanceId\": \"%s\",\n"
  1201. "\t\"clock\": %llu,\n"
  1202. "\t\"name\": \"%s\",\n"
  1203. "\t\"private\": %s,\n"
  1204. "\t\"enableBroadcast\": %s,\n"
  1205. "\t\"allowPassiveBridging\": %s,\n"
  1206. "\t\"v4AssignMode\": \"%s\",\n"
  1207. "\t\"v6AssignMode\": \"%s\",\n"
  1208. "\t\"multicastLimit\": %d,\n"
  1209. "\t\"creationTime\": %llu,\n"
  1210. "\t\"revision\": %llu,\n"
  1211. "\t\"memberRevisionCounter\": %llu,\n"
  1212. "\t\"authorizedMemberCount\": %llu,\n"
  1213. "\t\"relays\": [",
  1214. nwids,
  1215. _instanceId.c_str(),
  1216. (unsigned long long)OSUtils::now(),
  1217. _jsonEscape((const char *)sqlite3_column_text(_sGetNetworkById,0)).c_str(),
  1218. (sqlite3_column_int(_sGetNetworkById,1) > 0) ? "true" : "false",
  1219. (sqlite3_column_int(_sGetNetworkById,2) > 0) ? "true" : "false",
  1220. (sqlite3_column_int(_sGetNetworkById,3) > 0) ? "true" : "false",
  1221. _jsonEscape((const char *)sqlite3_column_text(_sGetNetworkById,4)).c_str(),
  1222. _jsonEscape((const char *)sqlite3_column_text(_sGetNetworkById,5)).c_str(),
  1223. sqlite3_column_int(_sGetNetworkById,6),
  1224. (unsigned long long)sqlite3_column_int64(_sGetNetworkById,7),
  1225. (unsigned long long)sqlite3_column_int64(_sGetNetworkById,8),
  1226. (unsigned long long)sqlite3_column_int64(_sGetNetworkById,9),
  1227. (unsigned long long)sqlite3_column_int64(_sGetNetworkById,10));
  1228. responseBody = json;
  1229. sqlite3_reset(_sGetRelays);
  1230. sqlite3_bind_text(_sGetRelays,1,nwids,16,SQLITE_STATIC);
  1231. bool firstRelay = true;
  1232. while (sqlite3_step(_sGetRelays) == SQLITE_ROW) {
  1233. responseBody.append(firstRelay ? "\n\t\t" : ",\n\t\t");
  1234. firstRelay = false;
  1235. responseBody.append("{\"address\":\"");
  1236. responseBody.append((const char *)sqlite3_column_text(_sGetRelays,0));
  1237. responseBody.append("\",\"phyAddress\":\"");
  1238. responseBody.append(_jsonEscape((const char *)sqlite3_column_text(_sGetRelays,1)));
  1239. responseBody.append("\"}");
  1240. }
  1241. responseBody.append("],\n\t\"routes\": [");
  1242. sqlite3_reset(_sGetRoutes);
  1243. sqlite3_bind_text(_sGetRoutes,1,nwids,16,SQLITE_STATIC);
  1244. bool firstRoute = true;
  1245. while (sqlite3_step(_sGetRoutes) == SQLITE_ROW) {
  1246. responseBody.append(firstRoute ? "\n\t\t" : ",\n\t\t");
  1247. firstRoute = false;
  1248. responseBody.append("{\"target\":");
  1249. char tmp[128];
  1250. const unsigned char *ip = (const unsigned char *)sqlite3_column_blob(_sGetRoutes,0);
  1251. switch(sqlite3_column_int(_sGetRoutes,3)) { // ipVersion
  1252. case 4:
  1253. Utils::snprintf(tmp,sizeof(tmp),"\"%d.%d.%d.%d/%d\"",(int)ip[12],(int)ip[13],(int)ip[14],(int)ip[15],sqlite3_column_int(_sGetRoutes,2));
  1254. break;
  1255. case 6:
  1256. Utils::snprintf(tmp,sizeof(tmp),"\"%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x/%d\"",(int)ip[0],(int)ip[1],(int)ip[2],(int)ip[3],(int)ip[4],(int)ip[5],(int)ip[6],(int)ip[7],(int)ip[8],(int)ip[9],(int)ip[10],(int)ip[11],(int)ip[12],(int)ip[13],(int)ip[14],(int)ip[15],sqlite3_column_int(_sGetRoutes,2));
  1257. break;
  1258. }
  1259. responseBody.append(tmp);
  1260. if (sqlite3_column_type(_sGetRoutes,1) == SQLITE_NULL) {
  1261. responseBody.append(",\"via\":null");
  1262. } else {
  1263. responseBody.append(",\"via\":");
  1264. ip = (const unsigned char *)sqlite3_column_blob(_sGetRoutes,1);
  1265. switch(sqlite3_column_int(_sGetRoutes,3)) { // ipVersion
  1266. case 4:
  1267. Utils::snprintf(tmp,sizeof(tmp),"\"%d.%d.%d.%d\"",(int)ip[12],(int)ip[13],(int)ip[14],(int)ip[15]);
  1268. break;
  1269. case 6:
  1270. Utils::snprintf(tmp,sizeof(tmp),"\"%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x:%.2x%.2x\"",(int)ip[0],(int)ip[1],(int)ip[2],(int)ip[3],(int)ip[4],(int)ip[5],(int)ip[6],(int)ip[7],(int)ip[8],(int)ip[9],(int)ip[10],(int)ip[11],(int)ip[12],(int)ip[13],(int)ip[14],(int)ip[15]);
  1271. break;
  1272. }
  1273. responseBody.append(tmp);
  1274. }
  1275. responseBody.append(",\"flags\":");
  1276. responseBody.append((const char *)sqlite3_column_text(_sGetRoutes,4));
  1277. responseBody.append(",\"metric\":");
  1278. responseBody.append((const char *)sqlite3_column_text(_sGetRoutes,5));
  1279. responseBody.push_back('}');
  1280. }
  1281. responseBody.append("],\n\t\"ipAssignmentPools\": [");
  1282. sqlite3_reset(_sGetIpAssignmentPools2);
  1283. sqlite3_bind_text(_sGetIpAssignmentPools2,1,nwids,16,SQLITE_STATIC);
  1284. bool firstIpAssignmentPool = true;
  1285. while (sqlite3_step(_sGetIpAssignmentPools2) == SQLITE_ROW) {
  1286. const char *ipRangeStartB = reinterpret_cast<const char *>(sqlite3_column_blob(_sGetIpAssignmentPools2,0));
  1287. const char *ipRangeEndB = reinterpret_cast<const char *>(sqlite3_column_blob(_sGetIpAssignmentPools2,1));
  1288. if ((ipRangeStartB)&&(ipRangeEndB)) {
  1289. InetAddress ipps,ippe;
  1290. int ipVersion = sqlite3_column_int(_sGetIpAssignmentPools2,2);
  1291. if (ipVersion == 4) {
  1292. ipps.set((const void *)(ipRangeStartB + 12),4,0);
  1293. ippe.set((const void *)(ipRangeEndB + 12),4,0);
  1294. } else if (ipVersion == 6) {
  1295. ipps.set((const void *)ipRangeStartB,16,0);
  1296. ippe.set((const void *)ipRangeEndB,16,0);
  1297. }
  1298. if (ipps) {
  1299. responseBody.append(firstIpAssignmentPool ? "\n\t\t" : ",\n\t\t");
  1300. firstIpAssignmentPool = false;
  1301. Utils::snprintf(json,sizeof(json),"{\"ipRangeStart\":\"%s\",\"ipRangeEnd\":\"%s\"}",
  1302. _jsonEscape(ipps.toIpString()).c_str(),
  1303. _jsonEscape(ippe.toIpString()).c_str());
  1304. responseBody.append(json);
  1305. }
  1306. }
  1307. }
  1308. responseBody.append("],\n\t\"rules\": [");
  1309. sqlite3_reset(_sListRules);
  1310. sqlite3_bind_text(_sListRules,1,nwids,16,SQLITE_STATIC);
  1311. bool firstRule = true;
  1312. while (sqlite3_step(_sListRules) == SQLITE_ROW) {
  1313. responseBody.append(firstRule ? "\n\t{\n" : ",{\n");
  1314. firstRule = false;
  1315. Utils::snprintf(json,sizeof(json),"\t\t\"ruleNo\": %lld,\n",sqlite3_column_int64(_sListRules,0));
  1316. responseBody.append(json);
  1317. if (sqlite3_column_type(_sListRules,1) != SQLITE_NULL) {
  1318. Utils::snprintf(json,sizeof(json),"\t\t\"nodeId\": \"%s\",\n",(const char *)sqlite3_column_text(_sListRules,1));
  1319. responseBody.append(json);
  1320. }
  1321. if (sqlite3_column_type(_sListRules,2) != SQLITE_NULL) {
  1322. Utils::snprintf(json,sizeof(json),"\t\t\"sourcePort\": \"%s\",\n",(const char *)sqlite3_column_text(_sListRules,2));
  1323. responseBody.append(json);
  1324. }
  1325. if (sqlite3_column_type(_sListRules,3) != SQLITE_NULL) {
  1326. Utils::snprintf(json,sizeof(json),"\t\t\"destPort\": \"%s\",\n",(const char *)sqlite3_column_text(_sListRules,3));
  1327. responseBody.append(json);
  1328. }
  1329. if (sqlite3_column_type(_sListRules,4) != SQLITE_NULL) {
  1330. Utils::snprintf(json,sizeof(json),"\t\t\"vlanId\": %d,\n",sqlite3_column_int(_sListRules,4));
  1331. responseBody.append(json);
  1332. }
  1333. if (sqlite3_column_type(_sListRules,5) != SQLITE_NULL) {
  1334. Utils::snprintf(json,sizeof(json),"\t\t\"vlanPcp\": %d,\n",sqlite3_column_int(_sListRules,5));
  1335. responseBody.append(json);
  1336. }
  1337. if (sqlite3_column_type(_sListRules,6) != SQLITE_NULL) {
  1338. Utils::snprintf(json,sizeof(json),"\t\t\"etherType\": %d,\n",sqlite3_column_int(_sListRules,6));
  1339. responseBody.append(json);
  1340. }
  1341. if (sqlite3_column_type(_sListRules,7) != SQLITE_NULL) {
  1342. Utils::snprintf(json,sizeof(json),"\t\t\"macSource\": \"%s\",\n",MAC((const char *)sqlite3_column_text(_sListRules,7)).toString().c_str());
  1343. responseBody.append(json);
  1344. }
  1345. if (sqlite3_column_type(_sListRules,8) != SQLITE_NULL) {
  1346. Utils::snprintf(json,sizeof(json),"\t\t\"macDest\": \"%s\",\n",MAC((const char *)sqlite3_column_text(_sListRules,8)).toString().c_str());
  1347. responseBody.append(json);
  1348. }
  1349. if (sqlite3_column_type(_sListRules,9) != SQLITE_NULL) {
  1350. Utils::snprintf(json,sizeof(json),"\t\t\"ipSource\": \"%s\",\n",_jsonEscape((const char *)sqlite3_column_text(_sListRules,9)).c_str());
  1351. responseBody.append(json);
  1352. }
  1353. if (sqlite3_column_type(_sListRules,10) != SQLITE_NULL) {
  1354. Utils::snprintf(json,sizeof(json),"\t\t\"ipDest\": \"%s\",\n",_jsonEscape((const char *)sqlite3_column_text(_sListRules,10)).c_str());
  1355. responseBody.append(json);
  1356. }
  1357. if (sqlite3_column_type(_sListRules,11) != SQLITE_NULL) {
  1358. Utils::snprintf(json,sizeof(json),"\t\t\"ipTos\": %d,\n",sqlite3_column_int(_sListRules,11));
  1359. responseBody.append(json);
  1360. }
  1361. if (sqlite3_column_type(_sListRules,12) != SQLITE_NULL) {
  1362. Utils::snprintf(json,sizeof(json),"\t\t\"ipProtocol\": %d,\n",sqlite3_column_int(_sListRules,12));
  1363. responseBody.append(json);
  1364. }
  1365. if (sqlite3_column_type(_sListRules,13) != SQLITE_NULL) {
  1366. Utils::snprintf(json,sizeof(json),"\t\t\"ipSourcePort\": %d,\n",sqlite3_column_int(_sListRules,13));
  1367. responseBody.append(json);
  1368. }
  1369. if (sqlite3_column_type(_sListRules,14) != SQLITE_NULL) {
  1370. Utils::snprintf(json,sizeof(json),"\t\t\"ipDestPort\": %d,\n",sqlite3_column_int(_sListRules,14));
  1371. responseBody.append(json);
  1372. }
  1373. if (sqlite3_column_type(_sListRules,15) != SQLITE_NULL) {
  1374. Utils::snprintf(json,sizeof(json),"\t\t\"flags\": %lu,\n",(unsigned long)sqlite3_column_int64(_sListRules,15));
  1375. responseBody.append(json);
  1376. }
  1377. if (sqlite3_column_type(_sListRules,16) != SQLITE_NULL) {
  1378. Utils::snprintf(json,sizeof(json),"\t\t\"invFlags\": %lu,\n",(unsigned long)sqlite3_column_int64(_sListRules,16));
  1379. responseBody.append(json);
  1380. }
  1381. responseBody.append("\t\t\"action\": \"");
  1382. responseBody.append(_jsonEscape( (sqlite3_column_type(_sListRules,17) == SQLITE_NULL) ? "drop" : (const char *)sqlite3_column_text(_sListRules,17) ));
  1383. responseBody.append("\"\n\t}");
  1384. }
  1385. responseBody.append("]\n}\n");
  1386. responseContentType = "application/json";
  1387. return 200;
  1388. } // else 404
  1389. }
  1390. } else if (path.size() == 1) {
  1391. // list networks
  1392. sqlite3_reset(_sListNetworks);
  1393. responseContentType = "application/json";
  1394. responseBody = "[";
  1395. bool first = true;
  1396. while (sqlite3_step(_sListNetworks) == SQLITE_ROW) {
  1397. if (first) {
  1398. first = false;
  1399. responseBody.push_back('"');
  1400. } else responseBody.append(",\"");
  1401. responseBody.append((const char *)sqlite3_column_text(_sListNetworks,0));
  1402. responseBody.push_back('"');
  1403. }
  1404. responseBody.push_back(']');
  1405. return 200;
  1406. } // else 404
  1407. } else {
  1408. // GET /controller returns status and API version if controller is supported
  1409. Utils::snprintf(json,sizeof(json),"{\n\t\"controller\": true,\n\t\"apiVersion\": %d,\n\t\"clock\": %llu,\n\t\"instanceId\": \"%s\"\n}\n",ZT_NETCONF_CONTROLLER_API_VERSION,(unsigned long long)OSUtils::now(),_instanceId.c_str());
  1410. responseBody = json;
  1411. responseContentType = "application/json";
  1412. return 200;
  1413. }
  1414. return 404;
  1415. }
  1416. NetworkController::ResultCode SqliteNetworkController::_doNetworkConfigRequest(const InetAddress &fromAddr,const Identity &signingId,const Identity &identity,uint64_t nwid,const NetworkConfigRequestMetaData &metaData,Buffer<8194> &netconf)
  1417. {
  1418. // Assumes _lock is locked
  1419. if (((!signingId)||(!signingId.hasPrivate()))||(signingId.address().toInt() != (nwid >> 24))) {
  1420. return NetworkController::NETCONF_QUERY_INTERNAL_SERVER_ERROR;
  1421. }
  1422. // Note: we can't reuse prepared statements that return const char * pointers without
  1423. // making our own copy in e.g. a std::string first.
  1424. //const bool clientIs104 = (Utils::compareVersion(metaData.majorVersion,metaData.minorVersion,metaData.revision,1,0,4) >= 0);
  1425. const uint64_t now = OSUtils::now();
  1426. // Check rate limit circuit breaker to prevent flooding
  1427. {
  1428. uint64_t &lrt = _lastRequestTime[std::pair<uint64_t,uint64_t>(identity.address().toInt(),nwid)];
  1429. if ((now - lrt) <= ZT_NETCONF_MIN_REQUEST_PERIOD)
  1430. return NetworkController::NETCONF_QUERY_IGNORE;
  1431. lrt = now;
  1432. }
  1433. NetworkRecord network;
  1434. memset(&network,0,sizeof(network));
  1435. Utils::snprintf(network.id,sizeof(network.id),"%.16llx",(unsigned long long)nwid);
  1436. MemberRecord member;
  1437. memset(&member,0,sizeof(member));
  1438. Utils::snprintf(member.nodeId,sizeof(member.nodeId),"%.10llx",(unsigned long long)identity.address().toInt());
  1439. // Create Node record or do full identity check if we already have one
  1440. sqlite3_reset(_sGetNodeIdentity);
  1441. sqlite3_bind_text(_sGetNodeIdentity,1,member.nodeId,10,SQLITE_STATIC);
  1442. if (sqlite3_step(_sGetNodeIdentity) == SQLITE_ROW) {
  1443. try {
  1444. Identity alreadyKnownIdentity((const char *)sqlite3_column_text(_sGetNodeIdentity,0));
  1445. if (alreadyKnownIdentity != identity)
  1446. return NetworkController::NETCONF_QUERY_ACCESS_DENIED;
  1447. } catch ( ... ) { // identity stored in database is not valid or is NULL
  1448. return NetworkController::NETCONF_QUERY_ACCESS_DENIED;
  1449. }
  1450. } else {
  1451. std::string idstr(identity.toString(false));
  1452. sqlite3_reset(_sCreateOrReplaceNode);
  1453. sqlite3_bind_text(_sCreateOrReplaceNode,1,member.nodeId,10,SQLITE_STATIC);
  1454. sqlite3_bind_text(_sCreateOrReplaceNode,2,idstr.c_str(),-1,SQLITE_STATIC);
  1455. if (sqlite3_step(_sCreateOrReplaceNode) != SQLITE_DONE) {
  1456. return NetworkController::NETCONF_QUERY_INTERNAL_SERVER_ERROR;
  1457. }
  1458. }
  1459. // Fetch Network record
  1460. sqlite3_reset(_sGetNetworkById);
  1461. sqlite3_bind_text(_sGetNetworkById,1,network.id,16,SQLITE_STATIC);
  1462. if (sqlite3_step(_sGetNetworkById) == SQLITE_ROW) {
  1463. network.name = (const char *)sqlite3_column_text(_sGetNetworkById,0);
  1464. network.isPrivate = (sqlite3_column_int(_sGetNetworkById,1) > 0);
  1465. network.enableBroadcast = (sqlite3_column_int(_sGetNetworkById,2) > 0);
  1466. network.allowPassiveBridging = (sqlite3_column_int(_sGetNetworkById,3) > 0);
  1467. network.v4AssignMode = (const char *)sqlite3_column_text(_sGetNetworkById,4);
  1468. network.v6AssignMode = (const char *)sqlite3_column_text(_sGetNetworkById,5);
  1469. network.multicastLimit = sqlite3_column_int(_sGetNetworkById,6);
  1470. network.creationTime = (uint64_t)sqlite3_column_int64(_sGetNetworkById,7);
  1471. network.revision = (uint64_t)sqlite3_column_int64(_sGetNetworkById,8);
  1472. network.memberRevisionCounter = (uint64_t)sqlite3_column_int64(_sGetNetworkById,9);
  1473. } else {
  1474. return NetworkController::NETCONF_QUERY_OBJECT_NOT_FOUND;
  1475. }
  1476. // Fetch Member record
  1477. bool foundMember = false;
  1478. sqlite3_reset(_sGetMember);
  1479. sqlite3_bind_text(_sGetMember,1,network.id,16,SQLITE_STATIC);
  1480. sqlite3_bind_text(_sGetMember,2,member.nodeId,10,SQLITE_STATIC);
  1481. if (sqlite3_step(_sGetMember) == SQLITE_ROW) {
  1482. foundMember = true;
  1483. member.rowid = (int64_t)sqlite3_column_int64(_sGetMember,0);
  1484. member.authorized = (sqlite3_column_int(_sGetMember,1) > 0);
  1485. member.activeBridge = (sqlite3_column_int(_sGetMember,2) > 0);
  1486. }
  1487. // Create Member record for unknown nodes, auto-authorizing if network is public
  1488. if (!foundMember) {
  1489. member.authorized = (network.isPrivate ? false : true);
  1490. member.activeBridge = false;
  1491. sqlite3_reset(_sCreateMember);
  1492. sqlite3_bind_text(_sCreateMember,1,network.id,16,SQLITE_STATIC);
  1493. sqlite3_bind_text(_sCreateMember,2,member.nodeId,10,SQLITE_STATIC);
  1494. sqlite3_bind_int(_sCreateMember,3,(member.authorized ? 1 : 0));
  1495. sqlite3_bind_text(_sCreateMember,4,network.id,16,SQLITE_STATIC);
  1496. if (sqlite3_step(_sCreateMember) != SQLITE_DONE) {
  1497. return NetworkController::NETCONF_QUERY_INTERNAL_SERVER_ERROR;
  1498. }
  1499. member.rowid = (int64_t)sqlite3_last_insert_rowid(_db);
  1500. sqlite3_reset(_sIncrementMemberRevisionCounter);
  1501. sqlite3_bind_text(_sIncrementMemberRevisionCounter,1,network.id,16,SQLITE_STATIC);
  1502. sqlite3_step(_sIncrementMemberRevisionCounter);
  1503. }
  1504. // Update NodeHistory with new log entry and delete expired entries
  1505. {
  1506. int64_t nextVC = 1;
  1507. sqlite3_reset(_sGetMaxNodeHistoryNetworkVisitCounter);
  1508. sqlite3_bind_text(_sGetMaxNodeHistoryNetworkVisitCounter,1,network.id,16,SQLITE_STATIC);
  1509. sqlite3_bind_text(_sGetMaxNodeHistoryNetworkVisitCounter,2,member.nodeId,10,SQLITE_STATIC);
  1510. if (sqlite3_step(_sGetMaxNodeHistoryNetworkVisitCounter) == SQLITE_ROW) {
  1511. nextVC = (int64_t)sqlite3_column_int64(_sGetMaxNodeHistoryNetworkVisitCounter,0) + 1;
  1512. }
  1513. std::string fastr;
  1514. if (fromAddr)
  1515. fastr = fromAddr.toString();
  1516. sqlite3_reset(_sAddNodeHistoryEntry);
  1517. sqlite3_bind_text(_sAddNodeHistoryEntry,1,member.nodeId,10,SQLITE_STATIC);
  1518. sqlite3_bind_text(_sAddNodeHistoryEntry,2,network.id,16,SQLITE_STATIC);
  1519. sqlite3_bind_int64(_sAddNodeHistoryEntry,3,nextVC);
  1520. sqlite3_bind_int(_sAddNodeHistoryEntry,4,(member.authorized ? 1 : 0));
  1521. sqlite3_bind_int64(_sAddNodeHistoryEntry,5,(long long)now);
  1522. sqlite3_bind_int(_sAddNodeHistoryEntry,6,(int)metaData.majorVersion);
  1523. sqlite3_bind_int(_sAddNodeHistoryEntry,7,(int)metaData.minorVersion);
  1524. sqlite3_bind_int(_sAddNodeHistoryEntry,8,(int)metaData.revision);
  1525. sqlite3_bind_text(_sAddNodeHistoryEntry,9,"",-1,SQLITE_STATIC);
  1526. if (fastr.length() > 0)
  1527. sqlite3_bind_text(_sAddNodeHistoryEntry,10,fastr.c_str(),-1,SQLITE_STATIC);
  1528. else sqlite3_bind_null(_sAddNodeHistoryEntry,10);
  1529. sqlite3_step(_sAddNodeHistoryEntry);
  1530. nextVC -= ZT_NETCONF_NODE_HISTORY_LENGTH;
  1531. if (nextVC >= 0) {
  1532. sqlite3_reset(_sDeleteOldNodeHistoryEntries);
  1533. sqlite3_bind_text(_sDeleteOldNodeHistoryEntries,1,network.id,16,SQLITE_STATIC);
  1534. sqlite3_bind_text(_sDeleteOldNodeHistoryEntries,2,member.nodeId,10,SQLITE_STATIC);
  1535. sqlite3_bind_int64(_sDeleteOldNodeHistoryEntries,3,nextVC);
  1536. sqlite3_step(_sDeleteOldNodeHistoryEntries);
  1537. }
  1538. }
  1539. // Check member authorization
  1540. if (!member.authorized)
  1541. return NetworkController::NETCONF_QUERY_ACCESS_DENIED;
  1542. // Create network configuration -- we create both legacy and new types and send both for backward compatibility
  1543. Dictionary legacy;
  1544. NetworkConfig nc;
  1545. // New network config structure
  1546. nc.networkId = Utils::hexStrToU64(network.id);
  1547. nc.type = network.isPrivate ? ZT_NETWORK_TYPE_PRIVATE : ZT_NETWORK_TYPE_PUBLIC;
  1548. nc.timestamp = now;
  1549. nc.revision = network.revision;
  1550. nc.issuedTo = member.nodeId;
  1551. if (network.enableBroadcast) nc.flags |= ZT_NETWORKCONFIG_FLAG_ENABLE_BROADCAST;
  1552. if (network.allowPassiveBridging) nc.flags |= ZT_NETWORKCONFIG_FLAG_ALLOW_PASSIVE_BRIDGING;
  1553. memcpy(nc.name,network.name,std::min((unsigned int)ZT_MAX_NETWORK_SHORT_NAME_LENGTH,(unsigned int)strlen(network.name)));
  1554. // Legacy dictionary -- will go away eventually
  1555. char tss[24],rs[24];
  1556. Utils::snprintf(tss,sizeof(tss),"%.16llx",(unsigned long long)now);
  1557. Utils::snprintf(rs,sizeof(rs),"%.16llx",(unsigned long long)network.revision);
  1558. legacy[ZT_NETWORKCONFIG_DICT_KEY_TIMESTAMP] = tss;
  1559. legacy[ZT_NETWORKCONFIG_DICT_KEY_REVISION] = rs;
  1560. legacy[ZT_NETWORKCONFIG_DICT_KEY_NETWORK_ID] = network.id;
  1561. legacy[ZT_NETWORKCONFIG_DICT_KEY_ISSUED_TO] = member.nodeId;
  1562. legacy[ZT_NETWORKCONFIG_DICT_KEY_PRIVATE] = network.isPrivate ? "1" : "0";
  1563. legacy[ZT_NETWORKCONFIG_DICT_KEY_NAME] = (network.name) ? network.name : "";
  1564. legacy[ZT_NETWORKCONFIG_DICT_KEY_ENABLE_BROADCAST] = network.enableBroadcast ? "1" : "0";
  1565. legacy[ZT_NETWORKCONFIG_DICT_KEY_ALLOW_PASSIVE_BRIDGING] = network.allowPassiveBridging ? "1" : "0";
  1566. { // TODO: right now only etherTypes are supported in rules
  1567. std::vector<int> allowedEtherTypes;
  1568. sqlite3_reset(_sGetEtherTypesFromRuleTable);
  1569. sqlite3_bind_text(_sGetEtherTypesFromRuleTable,1,network.id,16,SQLITE_STATIC);
  1570. while (sqlite3_step(_sGetEtherTypesFromRuleTable) == SQLITE_ROW) {
  1571. if (sqlite3_column_type(_sGetEtherTypesFromRuleTable,0) == SQLITE_NULL) {
  1572. allowedEtherTypes.clear();
  1573. allowedEtherTypes.push_back(0); // NULL 'allow' matches ANY
  1574. break;
  1575. } else {
  1576. int et = sqlite3_column_int(_sGetEtherTypesFromRuleTable,0);
  1577. if ((et >= 0)&&(et <= 0xffff))
  1578. allowedEtherTypes.push_back(et);
  1579. }
  1580. }
  1581. std::sort(allowedEtherTypes.begin(),allowedEtherTypes.end());
  1582. allowedEtherTypes.erase(std::unique(allowedEtherTypes.begin(),allowedEtherTypes.end()),allowedEtherTypes.end());
  1583. for(long i=0,k=0;((i<(long)allowedEtherTypes.size())&&(k<ZT_MAX_NETWORK_RULES));++i) {
  1584. if (allowedEtherTypes[i] > 0) {
  1585. nc.rules[k].t = ZT_NETWORK_RULE_MATCH_ETHERTYPE;
  1586. nc.rules[k].v.etherType = (uint16_t)allowedEtherTypes[i];
  1587. ++k;
  1588. }
  1589. nc.rules[k++].t = ZT_NETWORK_RULE_ACTION_ACCEPT;
  1590. }
  1591. std::string allowedEtherTypesCsv;
  1592. for(std::vector<int>::const_iterator i(allowedEtherTypes.begin());i!=allowedEtherTypes.end();++i) {
  1593. if (allowedEtherTypesCsv.length())
  1594. allowedEtherTypesCsv.push_back(',');
  1595. char tmp[16];
  1596. Utils::snprintf(tmp,sizeof(tmp),"%.4x",(unsigned int)*i);
  1597. allowedEtherTypesCsv.append(tmp);
  1598. }
  1599. legacy[ZT_NETWORKCONFIG_DICT_KEY_ALLOWED_ETHERNET_TYPES] = allowedEtherTypesCsv;
  1600. }
  1601. nc.multicastLimit = network.multicastLimit;
  1602. if (network.multicastLimit > 0) {
  1603. char ml[16];
  1604. Utils::snprintf(ml,sizeof(ml),"%lx",(unsigned long)network.multicastLimit);
  1605. legacy[ZT_NETWORKCONFIG_DICT_KEY_MULTICAST_LIMIT] = ml;
  1606. }
  1607. bool amActiveBridge = false;
  1608. {
  1609. std::string activeBridges;
  1610. sqlite3_reset(_sGetActiveBridges);
  1611. sqlite3_bind_text(_sGetActiveBridges,1,network.id,16,SQLITE_STATIC);
  1612. while (sqlite3_step(_sGetActiveBridges) == SQLITE_ROW) {
  1613. const char *ab = (const char *)sqlite3_column_text(_sGetActiveBridges,0);
  1614. if ((ab)&&(strlen(ab) == 10)) {
  1615. const uint64_t ab2 = Utils::hexStrToU64(ab);
  1616. if ((ab2)&&(nc.specialistCount < ZT_MAX_NETWORK_SPECIALISTS)) {
  1617. nc.specialists[nc.specialistCount++] = ab2 | ZT_NETWORKCONFIG_SPECIALIST_TYPE_ACTIVE_BRIDGE;
  1618. }
  1619. if (activeBridges.length())
  1620. activeBridges.push_back(',');
  1621. activeBridges.append(ab);
  1622. if (!strcmp(member.nodeId,ab))
  1623. amActiveBridge = true;
  1624. }
  1625. if (activeBridges.length() > 1024) // sanity check -- you can't have too many active bridges at the moment
  1626. break;
  1627. }
  1628. if (activeBridges.length())
  1629. legacy[ZT_NETWORKCONFIG_DICT_KEY_ACTIVE_BRIDGES] = activeBridges;
  1630. }
  1631. // Do not send relays to 1.1.0 since it had a serious bug in using them
  1632. // 1.1.0 will still work, it'll just fall back to roots instead of using network preferred relays
  1633. if (!((metaData.majorVersion == 1)&&(metaData.minorVersion == 1)&&(metaData.revision == 0))) {
  1634. std::string relays;
  1635. sqlite3_reset(_sGetRelays);
  1636. sqlite3_bind_text(_sGetRelays,1,network.id,16,SQLITE_STATIC);
  1637. while (sqlite3_step(_sGetRelays) == SQLITE_ROW) {
  1638. const char *n = (const char *)sqlite3_column_text(_sGetRelays,0);
  1639. const char *a = (const char *)sqlite3_column_text(_sGetRelays,1);
  1640. if ((n)&&(a)) {
  1641. Address node(n);
  1642. InetAddress addr(a);
  1643. if (node) {
  1644. bool sf = false;
  1645. for(unsigned int k=0;k<nc.specialistCount;++k) {
  1646. if ((nc.specialists[k] & 0xffffffffffULL) == node.toInt()) {
  1647. nc.specialists[k] |= ZT_NETWORKCONFIG_SPECIALIST_TYPE_NETWORK_PREFERRED_RELAY;
  1648. sf = true;
  1649. break;
  1650. }
  1651. }
  1652. if ((!sf)&&(nc.specialistCount < ZT_MAX_NETWORK_SPECIALISTS)) {
  1653. nc.specialists[nc.specialistCount++] = node.toInt() | ZT_NETWORKCONFIG_SPECIALIST_TYPE_NETWORK_PREFERRED_RELAY;
  1654. }
  1655. if (relays.length())
  1656. relays.push_back(',');
  1657. relays.append(node.toString());
  1658. if (addr) {
  1659. relays.push_back(';');
  1660. relays.append(addr.toString());
  1661. }
  1662. }
  1663. }
  1664. }
  1665. if (relays.length())
  1666. legacy[ZT_NETWORKCONFIG_DICT_KEY_RELAYS] = relays;
  1667. }
  1668. sqlite3_reset(_sGetRoutes);
  1669. sqlite3_bind_text(_sGetRoutes,1,network.id,16,SQLITE_STATIC);
  1670. while ((sqlite3_step(_sGetRoutes) == SQLITE_ROW)&&(nc.routeCount < ZT_MAX_NETWORK_ROUTES)) {
  1671. ZT_VirtualNetworkRoute *r = &(nc.routes[nc.routeCount]);
  1672. memset(r,0,sizeof(ZT_VirtualNetworkRoute));
  1673. switch(sqlite3_column_int(_sGetRoutes,3)) { // ipVersion
  1674. case 4:
  1675. *(reinterpret_cast<InetAddress *>(&(r->target))) = InetAddress((const void *)((const char *)sqlite3_column_blob(_sGetRoutes,0) + 12),4,(unsigned int)sqlite3_column_int(_sGetRoutes,2));
  1676. break;
  1677. case 6:
  1678. *(reinterpret_cast<InetAddress *>(&(r->target))) = InetAddress((const void *)sqlite3_column_blob(_sGetRoutes,0),16,(unsigned int)sqlite3_column_int(_sGetRoutes,2));
  1679. break;
  1680. default:
  1681. continue;
  1682. }
  1683. if (sqlite3_column_type(_sGetRoutes,1) != SQLITE_NULL) {
  1684. switch(sqlite3_column_int(_sGetRoutes,3)) { // ipVersion
  1685. case 4:
  1686. *(reinterpret_cast<InetAddress *>(&(r->via))) = InetAddress((const void *)((const char *)sqlite3_column_blob(_sGetRoutes,1) + 12),4,0);
  1687. break;
  1688. case 6:
  1689. *(reinterpret_cast<InetAddress *>(&(r->via))) = InetAddress((const void *)sqlite3_column_blob(_sGetRoutes,1),16,0);
  1690. break;
  1691. default:
  1692. continue;
  1693. }
  1694. }
  1695. r->flags = (uint16_t)sqlite3_column_int(_sGetRoutes,4);
  1696. r->metric = (uint16_t)sqlite3_column_int(_sGetRoutes,5);
  1697. ++nc.routeCount;
  1698. }
  1699. if ((network.v4AssignMode)&&(!strcmp(network.v4AssignMode,"zt"))) {
  1700. std::string v4s;
  1701. bool haveStaticIpAssignment = false;
  1702. sqlite3_reset(_sGetIpAssignmentsForNode);
  1703. sqlite3_bind_text(_sGetIpAssignmentsForNode,1,network.id,16,SQLITE_STATIC);
  1704. sqlite3_bind_text(_sGetIpAssignmentsForNode,2,member.nodeId,10,SQLITE_STATIC);
  1705. sqlite3_bind_int(_sGetIpAssignmentsForNode,3,4); // 4 == IPv4
  1706. while (sqlite3_step(_sGetIpAssignmentsForNode) == SQLITE_ROW) {
  1707. const unsigned char *const ip = (const unsigned char *)sqlite3_column_blob(_sGetIpAssignmentsForNode,1);
  1708. if ((!ip)||(sqlite3_column_bytes(_sGetIpAssignmentsForNode,1) != 16))
  1709. continue;
  1710. int ipNetmaskBits = sqlite3_column_int(_sGetIpAssignmentsForNode,2);
  1711. if ((ipNetmaskBits <= 0)||(ipNetmaskBits > 32))
  1712. continue;
  1713. char ips[32];
  1714. Utils::snprintf(ips,sizeof(ips),"%d.%d.%d.%d/%d",(int)ip[12],(int)ip[13],(int)ip[14],(int)ip[15],ipNetmaskBits);
  1715. if (sqlite3_column_int(_sGetIpAssignmentsForNode,0) == 0 /*ZT_IP_ASSIGNMENT_TYPE_ADDRESS*/) {
  1716. haveStaticIpAssignment = true;
  1717. InetAddress tmp2(ips);
  1718. if (tmp2)
  1719. nc.staticIps[nc.staticIpCount++] = tmp2;
  1720. if (v4s.length())
  1721. v4s.push_back(',');
  1722. v4s.append(ips);
  1723. }
  1724. }
  1725. if ((!haveStaticIpAssignment)&&(!amActiveBridge)) {
  1726. // Attempt to auto-assign an IPv4 address from an available routed pool
  1727. sqlite3_reset(_sGetIpAssignmentPools);
  1728. sqlite3_bind_text(_sGetIpAssignmentPools,1,network.id,16,SQLITE_STATIC);
  1729. sqlite3_bind_int(_sGetIpAssignmentPools,2,4); // 4 == IPv4
  1730. while (sqlite3_step(_sGetIpAssignmentPools) == SQLITE_ROW) {
  1731. const unsigned char *ipRangeStartB = reinterpret_cast<const unsigned char *>(sqlite3_column_blob(_sGetIpAssignmentPools,0));
  1732. const unsigned char *ipRangeEndB = reinterpret_cast<const unsigned char *>(sqlite3_column_blob(_sGetIpAssignmentPools,1));
  1733. if ((!ipRangeStartB)||(!ipRangeEndB)||(sqlite3_column_bytes(_sGetIpAssignmentPools,0) != 16)||(sqlite3_column_bytes(_sGetIpAssignmentPools,1) != 16))
  1734. continue;
  1735. uint32_t ipRangeStart = Utils::ntoh(*(reinterpret_cast<const uint32_t *>(ipRangeStartB + 12)));
  1736. uint32_t ipRangeEnd = Utils::ntoh(*(reinterpret_cast<const uint32_t *>(ipRangeEndB + 12)));
  1737. if ((ipRangeEnd <= ipRangeStart)||(ipRangeStart == 0))
  1738. continue;
  1739. uint32_t ipRangeLen = ipRangeEnd - ipRangeStart;
  1740. // Start with the LSB of the member's address
  1741. uint32_t ipTrialCounter = (uint32_t)(identity.address().toInt() & 0xffffffff);
  1742. for(uint32_t k=ipRangeStart,l=0;(k<=ipRangeEnd)&&(l < 1000000);++k,++l) {
  1743. uint32_t ip = (ipRangeLen > 0) ? (ipRangeStart + (ipTrialCounter % ipRangeLen)) : ipRangeStart;
  1744. ++ipTrialCounter;
  1745. if ((ip & 0x000000ff) == 0x000000ff)
  1746. continue; // don't allow addresses that end in .255
  1747. // Check if this IPv4 IP is within a local-to-Ethernet routed network
  1748. int routedNetmaskBits = 0;
  1749. for(unsigned int rk=0;rk<nc.routeCount;++rk) {
  1750. if ((!nc.routes[rk].via.ss_family)&&(nc.routes[rk].target.ss_family == AF_INET)) {
  1751. uint32_t targetIp = Utils::ntoh((uint32_t)(reinterpret_cast<const struct sockaddr_in *>(&(nc.routes[rk].target))->sin_addr.s_addr));
  1752. int targetBits = Utils::ntoh((uint16_t)(reinterpret_cast<const struct sockaddr_in *>(&(nc.routes[rk].target))->sin_port));
  1753. if ((ip & (0xffffffff << (32 - targetBits))) == targetIp) {
  1754. routedNetmaskBits = targetBits;
  1755. break;
  1756. }
  1757. }
  1758. }
  1759. // If it's routed, then try to claim and assign it and if successful end loop
  1760. if (routedNetmaskBits > 0) {
  1761. uint32_t ipBlob[4];
  1762. ipBlob[0] = 0; ipBlob[1] = 0; ipBlob[2] = 0; ipBlob[3] = Utils::hton(ip);
  1763. sqlite3_reset(_sCheckIfIpIsAllocated);
  1764. sqlite3_bind_text(_sCheckIfIpIsAllocated,1,network.id,16,SQLITE_STATIC);
  1765. sqlite3_bind_blob(_sCheckIfIpIsAllocated,2,(const void *)ipBlob,16,SQLITE_STATIC);
  1766. sqlite3_bind_int(_sCheckIfIpIsAllocated,3,4); // 4 == IPv4
  1767. sqlite3_bind_int(_sCheckIfIpIsAllocated,4,(int)0 /*ZT_IP_ASSIGNMENT_TYPE_ADDRESS*/);
  1768. if (sqlite3_step(_sCheckIfIpIsAllocated) != SQLITE_ROW) {
  1769. // No rows returned, so the IP is available
  1770. sqlite3_reset(_sAllocateIp);
  1771. sqlite3_bind_text(_sAllocateIp,1,network.id,16,SQLITE_STATIC);
  1772. sqlite3_bind_text(_sAllocateIp,2,member.nodeId,10,SQLITE_STATIC);
  1773. sqlite3_bind_int(_sAllocateIp,3,(int)0 /*ZT_IP_ASSIGNMENT_TYPE_ADDRESS*/);
  1774. sqlite3_bind_blob(_sAllocateIp,4,(const void *)ipBlob,16,SQLITE_STATIC);
  1775. sqlite3_bind_int(_sAllocateIp,5,routedNetmaskBits); // IP netmask bits from matching route
  1776. sqlite3_bind_int(_sAllocateIp,6,4); // 4 == IPv4
  1777. if (sqlite3_step(_sAllocateIp) == SQLITE_DONE) {
  1778. char ips[32];
  1779. Utils::snprintf(ips,sizeof(ips),"%d.%d.%d.%d/%d",(int)((ip >> 24) & 0xff),(int)((ip >> 16) & 0xff),(int)((ip >> 8) & 0xff),(int)(ip & 0xff),routedNetmaskBits);
  1780. if (nc.staticIpCount < ZT_MAX_ZT_ASSIGNED_ADDRESSES) {
  1781. InetAddress tmp2(ips);
  1782. if (tmp2)
  1783. nc.staticIps[nc.staticIpCount++] = tmp2;
  1784. }
  1785. if (v4s.length())
  1786. v4s.push_back(',');
  1787. v4s.append(ips);
  1788. haveStaticIpAssignment = true;
  1789. break;
  1790. }
  1791. }
  1792. }
  1793. }
  1794. }
  1795. }
  1796. if (v4s.length())
  1797. legacy[ZT_NETWORKCONFIG_DICT_KEY_IPV4_STATIC] = v4s;
  1798. }
  1799. if ((network.v6AssignMode)&&(!strcmp(network.v6AssignMode,"rfc4193"))) {
  1800. InetAddress rfc4193Addr(InetAddress::makeIpv6rfc4193(nwid,identity.address().toInt()));
  1801. legacy[ZT_NETWORKCONFIG_DICT_KEY_IPV6_STATIC] = rfc4193Addr.toString();
  1802. if (nc.staticIpCount < ZT_MAX_ZT_ASSIGNED_ADDRESSES)
  1803. nc.staticIps[nc.staticIpCount++] = rfc4193Addr;
  1804. }
  1805. if (network.isPrivate) {
  1806. CertificateOfMembership com(now,ZT_NETWORK_COM_DEFAULT_REVISION_MAX_DELTA,nwid,identity.address());
  1807. if (com.sign(signingId)) {
  1808. legacy[ZT_NETWORKCONFIG_DICT_KEY_CERTIFICATE_OF_MEMBERSHIP] = com.toString();
  1809. nc.com = com;
  1810. } else {
  1811. return NETCONF_QUERY_INTERNAL_SERVER_ERROR;
  1812. }
  1813. }
  1814. // Append legacy network config data for older devices
  1815. if (metaData.protocolVersion < 6) {
  1816. if (!legacy.sign(signingId,now))
  1817. return NETCONF_QUERY_INTERNAL_SERVER_ERROR;
  1818. std::string legacyStr(legacy.toString());
  1819. netconf.append((const void *)legacyStr.data(),(unsigned int)legacyStr.length());
  1820. }
  1821. netconf.append((uint8_t)0);
  1822. // Append new format data for newer devices
  1823. if (metaData.protocolVersion >= 6) {
  1824. nc.serialize(netconf);
  1825. }
  1826. return NetworkController::NETCONF_QUERY_OK;
  1827. }
  1828. void SqliteNetworkController::_circuitTestCallback(ZT_Node *node,ZT_CircuitTest *test,const ZT_CircuitTestReport *report)
  1829. {
  1830. char tmp[65535];
  1831. SqliteNetworkController *const self = reinterpret_cast<SqliteNetworkController *>(test->ptr);
  1832. if (!test)
  1833. return;
  1834. if (!report)
  1835. return;
  1836. Mutex::Lock _l(self->_lock);
  1837. std::map< uint64_t,_CircuitTestEntry >::iterator cte(self->_circuitTests.find(test->testId));
  1838. if (cte == self->_circuitTests.end()) { // sanity check: a circuit test we didn't launch?
  1839. self->_node->circuitTestEnd(test);
  1840. ::free((void *)test);
  1841. return;
  1842. }
  1843. Utils::snprintf(tmp,sizeof(tmp),
  1844. "%s{\n"
  1845. "\t\"timestamp\": %llu,"ZT_EOL_S
  1846. "\t\"testId\": \"%.16llx\","ZT_EOL_S
  1847. "\t\"upstream\": \"%.10llx\","ZT_EOL_S
  1848. "\t\"current\": \"%.10llx\","ZT_EOL_S
  1849. "\t\"receivedTimestamp\": %llu,"ZT_EOL_S
  1850. "\t\"remoteTimestamp\": %llu,"ZT_EOL_S
  1851. "\t\"sourcePacketId\": \"%.16llx\","ZT_EOL_S
  1852. "\t\"flags\": %llu,"ZT_EOL_S
  1853. "\t\"sourcePacketHopCount\": %u,"ZT_EOL_S
  1854. "\t\"errorCode\": %u,"ZT_EOL_S
  1855. "\t\"vendor\": %d,"ZT_EOL_S
  1856. "\t\"protocolVersion\": %u,"ZT_EOL_S
  1857. "\t\"majorVersion\": %u,"ZT_EOL_S
  1858. "\t\"minorVersion\": %u,"ZT_EOL_S
  1859. "\t\"revision\": %u,"ZT_EOL_S
  1860. "\t\"platform\": %d,"ZT_EOL_S
  1861. "\t\"architecture\": %d,"ZT_EOL_S
  1862. "\t\"receivedOnLocalAddress\": \"%s\","ZT_EOL_S
  1863. "\t\"receivedFromRemoteAddress\": \"%s\""ZT_EOL_S
  1864. "}",
  1865. ((cte->second.jsonResults.length() > 0) ? ",\n" : ""),
  1866. (unsigned long long)report->timestamp,
  1867. (unsigned long long)test->testId,
  1868. (unsigned long long)report->upstream,
  1869. (unsigned long long)report->current,
  1870. (unsigned long long)OSUtils::now(),
  1871. (unsigned long long)report->remoteTimestamp,
  1872. (unsigned long long)report->sourcePacketId,
  1873. (unsigned long long)report->flags,
  1874. report->sourcePacketHopCount,
  1875. report->errorCode,
  1876. (int)report->vendor,
  1877. report->protocolVersion,
  1878. report->majorVersion,
  1879. report->minorVersion,
  1880. report->revision,
  1881. (int)report->platform,
  1882. (int)report->architecture,
  1883. reinterpret_cast<const InetAddress *>(&(report->receivedOnLocalAddress))->toString().c_str(),
  1884. reinterpret_cast<const InetAddress *>(&(report->receivedFromRemoteAddress))->toString().c_str());
  1885. cte->second.jsonResults.append(tmp);
  1886. }
  1887. } // namespace ZeroTier