src/python/network.c

   1 /*
   2         libloc - A library to determine the location of someone on the Internet
   3
   4         Copyright (C) 2017 IPFire Development Team <info@ipfire.org>
   5
   6         This library is free software; you can redistribute it and/or
   7         modify it under the terms of the GNU Lesser General Public
   8         License as published by the Free Software Foundation; either
   9         version 2.1 of the License, or (at your option) any later version.
  10
  11         This library 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 GNU
  14         Lesser General Public License for more details.
  15 */
  16
  17 #include <Python.h>
  18
  19 #include <errno.h>
  20
  21 #include <loc/libloc.h>
  22 #include <loc/network.h>
  23
  24 #include "locationmodule.h"
  25 #include "network.h"
  26
  27 PyObject* new_network(PyTypeObject* type, struct loc_network* network) {
  28         NetworkObject* self = (NetworkObject*)type->tp_alloc(type, 0);
  29         if (self) {
  30                 self->network = loc_network_ref(network);
  31         }
  32
  33         return (PyObject*)self;
  34 }
  35
  36 static PyObject* Network_new(PyTypeObject* type, PyObject* args, PyObject* kwds) {
  37         NetworkObject* self = (NetworkObject*)type->tp_alloc(type, 0);
  38
  39         return (PyObject*)self;
  40 }
  41
  42 static void Network_dealloc(NetworkObject* self) {
  43         if (self->network)
  44                 loc_network_unref(self->network);
  45
  46         Py_TYPE(self)->tp_free((PyObject* )self);
  47 }
  48
  49 static int Network_init(NetworkObject* self, PyObject* args, PyObject* kwargs) {
  50         const char* network = NULL;
  51
  52         if (!PyArg_ParseTuple(args, "s", &network))
  53                 return -1;
  54
  55         // Load the Network
  56         int r = loc_network_new_from_string(loc_ctx, &self->network, network);
  57         if (r) {
  58                 PyErr_Format(PyExc_ValueError, "Invalid network: %s", network);
  59                 return -1;
  60         }
  61
  62         return 0;
  63 }
  64
  65 static PyObject* Network_repr(NetworkObject* self) {
  66         char* network = loc_network_str(self->network);
  67
  68         PyObject* obj = PyUnicode_FromFormat("<location.Network %s>", network);
  69         free(network);
  70
  71         return obj;
  72 }
  73
  74 static PyObject* Network_get_country_code(NetworkObject* self) {
  75         const char* country_code = loc_network_get_country_code(self->network);
  76
  77         return PyUnicode_FromString(country_code);
  78 }
  79
  80 static int Network_set_country_code(NetworkObject* self, PyObject* value) {
  81         const char* country_code = PyUnicode_AsUTF8(value);
  82
  83         int r = loc_network_set_country_code(self->network, country_code);
  84         if (r) {
  85                 if (r == -EINVAL)
  86                         PyErr_SetString(PyExc_ValueError, "Invalid country code");
  87
  88                 return -1;
  89         }
  90
  91         return 0;
  92 }
  93
  94 static PyObject* Network_get_asn(NetworkObject* self) {
  95         uint32_t asn = loc_network_get_asn(self->network);
  96
  97         if (asn)
  98                 return PyLong_FromLong(asn);
  99
 100         Py_RETURN_NONE;
 101 }
 102
 103 static int Network_set_asn(NetworkObject* self, PyObject* value) {
 104         long int asn = PyLong_AsLong(value);
 105
 106         // Check if the ASN is within the valid range
 107         if (asn <= 0 || asn > UINT32_MAX) {
 108                 PyErr_Format(PyExc_ValueError, "Invalid ASN %ld", asn);
 109                 return -1;
 110         }
 111
 112         int r = loc_network_set_asn(self->network, asn);
 113         if (r)
 114                 return -1;
 115
 116         return 0;
 117 }
 118
 119 static struct PyGetSetDef Network_getsetters[] = {
 120         {
 121                 "asn",
 122                 (getter)Network_get_asn,
 123                 (setter)Network_set_asn,
 124                 NULL,
 125                 NULL,
 126         },
 127         {
 128                 "country_code",
 129                 (getter)Network_get_country_code,
 130                 (setter)Network_set_country_code,
 131                 NULL,
 132                 NULL,
 133         },
 134         { NULL },
 135 };
 136
 137 PyTypeObject NetworkType = {
 138         PyVarObject_HEAD_INIT(NULL, 0)
 139         tp_name:                "location.Network",
 140         tp_basicsize:           sizeof(NetworkObject),
 141         tp_flags:               Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
 142         tp_new:                 Network_new,
 143         tp_dealloc:             (destructor)Network_dealloc,
 144         tp_init:                (initproc)Network_init,
 145         tp_doc:                 "Network object",
 146         tp_getset:              Network_getsetters,
 147         tp_repr:                (reprfunc)Network_repr,
 148 };