practicas/pipi/src/nameserver.cpp

   1 #include "nameserver.h"
   2 #include <sstream>
   3 #include <algorithm>
   4 #include <iterator>
   5 #ifdef DEBUG2
   6 #include <iostream>
   7 #endif
   8
   9 static void parsename(const std::string& s, NameServer::name_t& name)
  10 {
  11     std::istringstream iss(s);
  12     std::string tok;
  13     while (std::getline(iss, tok, '.'))
  14     {
  15         name.push_back(tok);
  16     }
  17 }
  18
  19 static std::ostream& operator<< (std::ostream& os, const NameServer::name_t& name)
  20 {
  21     if (name.empty())
  22         return os;
  23     std::copy(name.begin(), name.end() - 1,
  24             std::ostream_iterator< std::string >(os, "."));
  25     return os << name.back();
  26 }
  27
  28 /// Parsea una zona
  29 static std::istream& parsezone(std::istream& is, NameServer::Zone& z)
  30     throw (std::runtime_error)
  31 {
  32     std::string line, sname, ip;
  33     // [dominio] [ttl] [parent ip]
  34     while (std::getline(is, line) && (line == "")); // Salteo líneas en blanco
  35     if (!is)
  36         return is;
  37     std::istringstream iss(line);
  38     if (!(iss >> sname >> z.ttl >> ip))
  39         throw std::runtime_error("Error al parsear");
  40     parsename(sname, z.name);
  41 #ifdef DEBUG2
  42     std::cerr << "parsezone: IP = " << ip << "\n\n";
  43 #endif
  44     z.parent = IPAddr(ip);
  45     // un record por linea, sin líneas vacías
  46     // [name] [type] [ip]
  47     while (std::getline(is, line) && (line != ""))
  48     {
  49         iss.clear();
  50         iss.str(line);
  51         std::string key, type;
  52         if (!(iss >> key >> type >> ip))
  53             throw std::runtime_error("Error al parsear");
  54         typedef NameServer::Record Rec;
  55 #ifdef DEBUG2
  56         std::cerr << "parsezone: IP = " << ip << "\n\n";
  57 #endif
  58         Rec r((type == "NS") ? Rec::NS : Rec::A, IPAddr(ip));
  59         z.records.insert(NameServer::Zone::records_t::value_type(key, r));
  60     }
  61 #ifdef DEBUG2
  62     std::cerr << "parsezone: " << z << "\n\n";
  63 #endif
  64     return is;
  65 }
  66
  67 /// Constructor
  68 NameServer::Zone::Zone(std::string sname, size_t ttl, const IPAddr& parent):
  69     ttl(ttl), parent(parent)
  70 {
  71     parsename(sname, name);
  72 }
  73
  74 /// Constructor
  75 NameServer::Zone::Zone(std::istream& is)
  76     throw (std::runtime_error)
  77 {
  78     // Parsea la zona
  79     if (!parsezone(is, *this))
  80         throw std::runtime_error("Error de parser, no hay zona");
  81 }
  82
  83 /// Limpia una zona
  84 void NameServer::Zone::clear()
  85 {
  86     name.clear();
  87     records.clear();
  88 }
  89
  90 /// Constructor
  91 NameServer::NameServer(std::istream& is)
  92     throw (std::runtime_error)
  93 {
  94     Zone z;
  95     while (parsezone(is, z))
  96     {
  97         zones.push_back(z);
  98         z.clear();
  99 #ifdef DEBUG2
 100         std::cerr << "NameServer: " << z << "\n\n";
 101 #endif
 102     }
 103 }
 104
 105 std::ostream& operator<< (std::ostream& os, const NameServer::Record& r)
 106 {
 107     if (r.type == NameServer::Record::NS)
 108         os << "NS ";
 109     else
 110         os << "A ";
 111     return os << r.ip;
 112 }
 113
 114 std::ostream& operator<< (std::ostream& os,
 115         const NameServer::Zone::records_t::value_type& p)
 116 {
 117     return os << p.first << ": " << p.second;
 118 }
 119
 120 std::ostream& operator<< (std::ostream& os, const NameServer::Zone& z)
 121 {
 122     os << "Zone " << z.name << " " << z.ttl << " " << z.parent << "\n";
 123     std::copy(z.records.begin(), z.records.end(), std::ostream_iterator<
 124             NameServer::Zone::records_t::value_type >(os, "\n"));
 125     return os;
 126 }
 127
 128 std::ostream& operator<< (std::ostream& os, const NameServer& ns)
 129 {
 130     os << "NameServer: zones[" << ns.zones.size() << "] (\n\n";
 131     std::copy(ns.zones.begin(), ns.zones.end(),
 132             std::ostream_iterator< NameServer::Zone >(os, "\n"));
 133     return os << ")";
 134 }
 135
 136 // vim: set et sw=4 sts=4 :