Add a dictionary with countries to the database

[people/ms/libloc.git] / src / loc / database.h

diff --git a/src/loc/database.h b/src/loc/database.h
index 3b3208a4f24ec373cb6bcb5ace055f16a1cec29d..b1f0856feb9ff34b4572dcfec5f5cae559c56ed4 100644 (file)
--- a/src/loc/database.h
+++ b/src/loc/database.h
@@ -24,6 +24,7 @@
 #include <loc/libloc.h>
 #include <loc/network.h>
 #include <loc/as.h>
+#include <loc/country.h>
 
 struct loc_database;
 int loc_database_new(struct loc_ctx* ctx, struct loc_database** database, FILE* f);
@@ -43,6 +44,9 @@ int loc_database_lookup(struct loc_database* db,
 int loc_database_lookup_from_string(struct loc_database* db,
                const char* string, struct loc_network** network);
 
+int loc_database_get_country(struct loc_database* db,
+               struct loc_country** country, const char* code);
+
 enum loc_database_enumerator_mode {
        LOC_DB_ENUMERATE_NETWORKS = 1,
        LOC_DB_ENUMERATE_ASES     = 2,
@@ -57,6 +61,7 @@ struct loc_database_enumerator* loc_database_enumerator_unref(struct loc_databas
 int loc_database_enumerator_set_string(struct loc_database_enumerator* enumerator, const char* string);
 int loc_database_enumerator_set_country_code(struct loc_database_enumerator* enumerator, const char* country_code);
 int loc_database_enumerator_set_asn(struct loc_database_enumerator* enumerator, unsigned int asn);
+int loc_database_enumerator_set_flag(struct loc_database_enumerator* enumerator, enum loc_network_flags flag);
 int loc_database_enumerator_next_as(
        struct loc_database_enumerator* enumerator, struct loc_as** as);
 int loc_database_enumerator_next_network(