-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* TRC fetcher --------- Co-authored-by: Tilmann Zäschke <tilmann.zaeschke@inf.ethz.ch>
- Loading branch information
Showing
31 changed files
with
994 additions
and
434 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
187 changes: 187 additions & 0 deletions
187
src/main/java/org/scion/jpan/internal/GlobalTopology.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,187 @@ | ||
| // Copyright 2023 ETH Zurich | ||
| // | ||
| // Licensed under the Apache License, Version 2.0 (the "License"); | ||
| // you may not use this file except in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, software | ||
| // distributed under the License is distributed on an "AS IS" BASIS, | ||
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| package org.scion.jpan.internal; | ||
|
|
||
| import com.google.gson.JsonArray; | ||
| import com.google.gson.JsonElement; | ||
| import com.google.gson.JsonObject; | ||
| import com.google.gson.JsonParser; | ||
| import java.time.Instant; | ||
| import java.util.*; | ||
| import org.scion.jpan.ScionRuntimeException; | ||
| import org.scion.jpan.ScionUtil; | ||
|
|
||
| /** Parse a topology file into a local topology. */ | ||
| public class GlobalTopology { | ||
|
|
||
| private final Map<Integer, Isd> world = new HashMap<>(); | ||
| /** | ||
| * The topology is "empty" if it wasn't initialized with TRC file (or TRC metadata). THis can | ||
| * happen when it is initialized from a local topology file without bootstrap server. | ||
| */ | ||
| private final boolean isEmpty; | ||
|
|
||
| public GlobalTopology(ScionBootstrapper server) { | ||
| if (server == null) { | ||
| this.isEmpty = true; | ||
| } else { | ||
| this.isEmpty = false; | ||
| getTrcFiles(server); | ||
| } | ||
| } | ||
|
|
||
| public static synchronized GlobalTopology create(ScionBootstrapper server) { | ||
| return new GlobalTopology(server); | ||
| } | ||
|
|
||
| public static GlobalTopology createEmpty() { | ||
| return new GlobalTopology(null); | ||
| } | ||
|
|
||
| private static JsonElement safeGet(JsonObject o, String name) { | ||
| JsonElement e = o.get(name); | ||
| if (e == null) { | ||
| throw new ScionRuntimeException("Entry not found in topology file: " + name); | ||
| } | ||
| return e; | ||
| } | ||
|
|
||
| public Optional<Boolean> isCoreAs(long isdAs) { | ||
| if (isEmpty) { | ||
| return Optional.empty(); | ||
| } | ||
| int isdCode = ScionUtil.extractIsd(isdAs); | ||
| Isd isd = world.get(isdCode); | ||
| if (isd == null) { | ||
| throw new ScionRuntimeException("Unknown ISD: " + isdCode); | ||
| } | ||
|
|
||
| for (Long core : isd.coreASes) { | ||
| if (core == isdAs) { | ||
| return Optional.of(true); | ||
| } | ||
| } | ||
| return Optional.of(false); | ||
| } | ||
|
|
||
| private void getTrcFiles(ScionBootstrapper server) { | ||
| String filesString = server.fetchFile("trcs"); | ||
| parseTrcFiles(filesString); | ||
|
|
||
| for (Isd isd : world.values()) { | ||
| String fileName = "isd" + isd.isd + "-b" + isd.baseNumber + "-s" + isd.serialNumber; | ||
| String file = server.fetchFile("trcs/" + fileName); | ||
| parseTrcFile(file, isd); | ||
| } | ||
| } | ||
|
|
||
| private void parseTrcFiles(String trcFile) { | ||
| JsonElement jsonTree = JsonParser.parseString(trcFile); | ||
| JsonArray entries = jsonTree.getAsJsonArray(); | ||
| for (int i = 0; i < entries.size(); i++) { | ||
| JsonObject entry = entries.get(i).getAsJsonObject(); | ||
| for (Map.Entry<String, JsonElement> e : entry.entrySet()) { | ||
| JsonObject cs = e.getValue().getAsJsonObject(); | ||
| int base = cs.get("base_number").getAsInt(); | ||
| int isd = cs.get("isd").getAsInt(); | ||
| int serial = cs.get("serial_number").getAsInt(); | ||
| world.put(isd, new Isd(isd, base, serial)); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private static void parseTrcFile(String trcFile, Isd isd) { | ||
| JsonElement jsonTree = JsonParser.parseString(trcFile); | ||
| if (jsonTree.isJsonObject()) { | ||
| JsonObject o = jsonTree.getAsJsonObject(); | ||
|
|
||
| JsonArray authoritativeAses = safeGet(o, "authoritative_ases").getAsJsonArray(); | ||
| for (int i = 0; i < authoritativeAses.size(); i++) { | ||
| String isdCode = authoritativeAses.get(i).getAsString(); | ||
| isd.authorativeASes.add(ScionUtil.parseIA(isdCode)); | ||
| } | ||
| JsonArray coreAses = safeGet(o, "core_ases").getAsJsonArray(); | ||
| for (int i = 0; i < coreAses.size(); i++) { | ||
| String isdCode = coreAses.get(i).getAsString(); | ||
| isd.coreASes.add(ScionUtil.parseIA(isdCode)); | ||
| } | ||
|
|
||
| isd.description = safeGet(o, "description").getAsString(); | ||
|
|
||
| JsonObject id = safeGet(o, "id").getAsJsonObject(); | ||
| int base = id.get("base_number").getAsInt(); | ||
| int isdCode = id.get("isd").getAsInt(); | ||
| int serial = id.get("serial_number").getAsInt(); | ||
| if (isd.isd != isdCode || isd.baseNumber != base || isd.serialNumber != serial) { | ||
| throw new IllegalStateException("ISD/Base/Serial mismatch in TRC file."); | ||
| } | ||
|
|
||
| JsonObject validity = safeGet(o, "validity").getAsJsonObject(); | ||
| String afterStr = validity.get("not_after").getAsString(); | ||
| String beforeStr = validity.get("not_before").getAsString(); | ||
| isd.notAfter = Instant.parse(afterStr); | ||
| isd.notBefore = Instant.parse(beforeStr); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public String toString() { | ||
| StringBuilder sb = new StringBuilder(); | ||
| for (Isd isd : world.values()) { | ||
| sb.append("ISD: ").append(isd).append('\n'); | ||
| } | ||
| return sb.toString(); | ||
| } | ||
|
|
||
| private static class Isd { | ||
| String description; | ||
| int baseNumber; | ||
| int isd; | ||
| int serialNumber; | ||
| Instant notAfter; | ||
| Instant notBefore; | ||
| final List<Long> authorativeASes = new ArrayList<>(); | ||
| final List<Long> coreASes = new ArrayList<>(); | ||
|
|
||
| Isd(int isd, int base, int serial) { | ||
| this.isd = isd; | ||
| this.baseNumber = base; | ||
| this.serialNumber = serial; | ||
| } | ||
|
|
||
| @Override | ||
| public String toString() { | ||
| return "{" | ||
| + "description='" | ||
| + description | ||
| + '\'' | ||
| + ", baseNumber=" | ||
| + baseNumber | ||
| + ", isd=" | ||
| + isd | ||
| + ", serialNumber=" | ||
| + serialNumber | ||
| + ", notAfter=" | ||
| + notAfter | ||
| + ", notBefore=" | ||
| + notBefore | ||
| + ", authorativeASes=" | ||
| + authorativeASes | ||
| + ", coreASes=" | ||
| + coreASes | ||
| + '}'; | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.