[people/ms/libloc.git] / src / test-country.c

/*
	libloc - A library to determine the location of someone on the Internet

	Copyright (C) 2019 IPFire Development Team <info@ipfire.org>

	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation; either version 2 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.
*/

#include <errno.h>
#include <stdio.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>

#include <loc/libloc.h>
#include <loc/country.h>
#include <loc/database.h>
#include <loc/network.h>
#include <loc/writer.h>

int main(int argc, char** argv) {
	struct loc_country* country;
	int err;

	// Check some valid country codes
	if (!loc_country_code_is_valid("XX")) {
		fprintf(stderr, "Valid country code detected as invalid: %s\n", "XX");
		exit(EXIT_FAILURE);
	}

	// Check some invalid country codes
	if (loc_country_code_is_valid("X1")) {
		fprintf(stderr, "Invalid country code detected as valid: %s\n", "X1");
		exit(EXIT_FAILURE);
	}

	struct loc_ctx* ctx;
	err = loc_new(&ctx);
	if (err < 0)
		exit(EXIT_FAILURE);

	// Create a database
	struct loc_writer* writer;
	err = loc_writer_new(ctx, &writer, NULL, NULL);
	if (err < 0)
		exit(EXIT_FAILURE);

	// Create a country
	err = loc_writer_add_country(writer, &country, "XX");
	if (err) {
		fprintf(stderr, "Could not create country\n");
		exit(EXIT_FAILURE);
	}

	// Set name & continent
	loc_country_set_name(country, "Testistan");
	loc_country_set_continent_code(country, "YY");

	// Free country
	loc_country_unref(country);

	// Add another country
	err = loc_writer_add_country(writer, &country, "YY");
	if (err) {
		fprintf(stderr, "Could not create country: YY\n");
		exit(EXIT_FAILURE);
	}
	loc_country_unref(country);

	FILE* f = tmpfile();
	if (!f) {
		fprintf(stderr, "Could not open file for writing: %s\n", strerror(errno));
		exit(EXIT_FAILURE);
	}

	err = loc_writer_write(writer, f, LOC_DATABASE_VERSION_UNSET);
	if (err) {
		fprintf(stderr, "Could not write database: %s\n", strerror(-err));
		exit(EXIT_FAILURE);
	}
	loc_writer_unref(writer);

	// And open it again from disk
	struct loc_database* db;
	err = loc_database_new(ctx, &db, f);
	if (err) {
		fprintf(stderr, "Could not open database: %s\n", strerror(-err));
		exit(EXIT_FAILURE);
	}

	// Lookup an address in the subnet
	err = loc_database_get_country(db, &country, "YY");
	if (err) {
		fprintf(stderr, "Could not find country: YY\n");
		exit(EXIT_FAILURE);
	}
	loc_country_unref(country);

	loc_database_unref(db);
	loc_unref(ctx);
	fclose(f);

	return EXIT_SUCCESS;
}
Commit	Line	Data
ec684c1a MT	1	/*
	2	libloc - A library to determine the location of someone on the Internet
	3
	4	Copyright (C) 2019 IPFire Development Team <info@ipfire.org>
	5
	6	This program is free software; you can redistribute it and/or modify
	7	it under the terms of the GNU General Public License as published by
	8	the Free Software Foundation; either version 2 of the License, or
	9	(at your option) any later version.
	10
	11	This program is distributed in the hope that it will be useful,
	12	but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	14	GNU General Public License for more details.
	15	*/
	16
	17	#include <errno.h>
	18	#include <stdio.h>
	19	#include <stddef.h>
	20	#include <stdlib.h>
	21	#include <string.h>
	22
	23	#include <loc/libloc.h>
57146963	24	#include <loc/country.h>
ec684c1a MT	25	#include <loc/database.h>
	26	#include <loc/network.h>
	27	#include <loc/writer.h>
	28
	29	int main(int argc, char** argv) {
	30	struct loc_country* country;
	31	int err;
	32
57146963 MT	33	// Check some valid country codes
	34	if (!loc_country_code_is_valid("XX")) {
	35	fprintf(stderr, "Valid country code detected as invalid: %s\n", "XX");
	36	exit(EXIT_FAILURE);
	37	}
	38
	39	// Check some invalid country codes
	40	if (loc_country_code_is_valid("X1")) {
	41	fprintf(stderr, "Invalid country code detected as valid: %s\n", "X1");
	42	exit(EXIT_FAILURE);
	43	}
	44
ec684c1a MT	45	struct loc_ctx* ctx;
	46	err = loc_new(&ctx);
	47	if (err < 0)
	48	exit(EXIT_FAILURE);
	49
	50	// Create a database
	51	struct loc_writer* writer;
5ce881d4	52	err = loc_writer_new(ctx, &writer, NULL, NULL);
ec684c1a MT	53	if (err < 0)
	54	exit(EXIT_FAILURE);
	55
	56	// Create a country
57146963	57	err = loc_writer_add_country(writer, &country, "XX");
ec684c1a MT	58	if (err) {
	59	fprintf(stderr, "Could not create country\n");
	60	exit(EXIT_FAILURE);
	61	}
	62
	63	// Set name & continent
	64	loc_country_set_name(country, "Testistan");
57146963	65	loc_country_set_continent_code(country, "YY");
ec684c1a MT	66
	67	// Free country
	68	loc_country_unref(country);
	69
8bbddd9a MT	70	// Add another country
	71	err = loc_writer_add_country(writer, &country, "YY");
	72	if (err) {
	73	fprintf(stderr, "Could not create country: YY\n");
	74	exit(EXIT_FAILURE);
	75	}
	76	loc_country_unref(country);
	77
6254dca6	78	FILE* f = tmpfile();
ec684c1a MT	79	if (!f) {
	80	fprintf(stderr, "Could not open file for writing: %s\n", strerror(errno));
	81	exit(EXIT_FAILURE);
	82	}
	83
22c7b98b	84	err = loc_writer_write(writer, f, LOC_DATABASE_VERSION_UNSET);
ec684c1a MT	85	if (err) {
	86	fprintf(stderr, "Could not write database: %s\n", strerror(-err));
	87	exit(EXIT_FAILURE);
	88	}
ec684c1a MT	89	loc_writer_unref(writer);
	90
	91	// And open it again from disk
ec684c1a MT	92	struct loc_database* db;
	93	err = loc_database_new(ctx, &db, f);
	94	if (err) {
	95	fprintf(stderr, "Could not open database: %s\n", strerror(-err));
	96	exit(EXIT_FAILURE);
	97	}
	98
	99	// Lookup an address in the subnet
8bbddd9a	100	err = loc_database_get_country(db, &country, "YY");
ec684c1a	101	if (err) {
8bbddd9a	102	fprintf(stderr, "Could not find country: YY\n");
ec684c1a MT	103	exit(EXIT_FAILURE);
	104	}
	105	loc_country_unref(country);
	106
	107	loc_database_unref(db);
	108	loc_unref(ctx);
6254dca6	109	fclose(f);
ec684c1a MT	110
	111	return EXIT_SUCCESS;
	112	}