+/// Devuelve la parte izquierda de n, si la parte derecha coincide exactamente
+/// con m, si no devuelve un vector vacío.
+/// Elemplo: name_split("mi.domi.nio", "domi.nio") == ["mi"]
+/// name_split("dos.mi.domi.nio", "domi.nio") == ["dos", "mi"]
+/// name_split("domi.nio", "domi.nio") == []
+/// name_split("papeli.nio", "domi.nio") == []
+static NameServer::Name
+name_split(NameServer::Name n, NameServer::Name m)
+{
+ NameServer::Name r;
+ std::reverse(n.begin(), n.end());
+ std::reverse(m.begin(), m.end());
+ // Si m es más grande o igual que n, no hay nada que hacer
+ if (n.size() <= m.size())
+ return r;
+ // Si no coincide la parte derecha, no hay nada que hacer
+ NameServer::Name::size_type i;
+ for (i = 0; i < m.size(); ++i)
+ if (n[i] != m[i])
+ return r;
+ // Si era todo igual y sobran cosas, devolvemos lo que "sobra"
+ while (i < n.size())
+ r.push_back(n[i++]);
+ std::reverse(r.begin(), r.end());
+#ifdef DEBUG_NAME
+ std::cout << "NameServer::name_split(" << n << ", " << m << ") -> "
+ << r << "\n";
+#endif
+ return r;
+}
+
+/// Devuelve -1 si es un nombre "hijo" (la parte derecha de n está contenida
+/// completamente en m, pero m y n no son iguales), 0 si m == n y 1 si es
+/// "padre" (m no coincide con la parte derecha de n).
+/// Elemplo: namecmp("mi.domi.nio", "domi.nio") == -1
+/// namecmp("otra.cosa", "domi.nio") == 1
+/// namecmp("papeli.nio", "domi.nio") == 1
+/// namecmp("domi.nio", "domi.nio") == 0
+/*enum name_cmp_t
+{
+ NC_DIRECT_CHILD, ///> Hijo directo, es decir, está en la zona m
+ NC_CHILD, ///> Hijo indirecto, está en una zona que cuelga de m
+ NC_EQUAL, ///> Es el host de la zona m
+ NC_DIRECT_PARENT, ///> Padre directo, es decir, coincide en parte con m
+ NC_PARENT ///> Padre indirecto, es completamente distinto a m
+};*/
+#if 0
+static int name_cmp(const NameServer::Name& n, const NameServer::Name& m)
+{
+ // Si m es más grande que n, seguro es padre
+ if (n.size() < m.size())
+ return 1;
+ // Si no coincide la parte derecha, seguro es padre
+ NameServer::Name::size_type i;
+ for (i = m.size(); i > 0; --i)
+ if (n[i-1] != m[i-1])
+ return 1;
+ // Si era todo igual y sobran cosas, es hijo
+ if (i)
+ return -1;
+ // Si no, son iguales.
+ return 0;
+}
+#endif
+
+struct search_zone: std::unary_function< NameServer::Zone, bool >
+{
+ bool local;
+ ResolvProtoResponse resp;
+ const NameServer::Name& name;
+ search_zone(const NameServer::Name& n): local(false), name(n) {}
+ bool operator() (const NameServer::Zone& z)
+ {
+ bool found = false;
+ NameServer::Name local_part = name_split(name, z.name);
+ if (!local_part.empty()) // Está en esta zona
+ {
+ local = true;
+ std::string n = local_part.back(); // Obtengo última parte
+ // busco
+ typedef NameServer::Zone::records_t::const_iterator itt;
+ std::pair<itt, itt> p = z.records.equal_range(n);
+ for (; p.first != p.second; ++p.first)
+ {
+ const NameServer::Record& r = (*p.first).second;
+ // Tiene que buscar solo A porque era un nombre
+ if ((local_part.size() == 1) && (r.type != NameServer::Record::A))
+ continue;
+ // Tiene que seguir para abajo, solo busca NS
+ if ((local_part.size() > 1) && r.type != NameServer::Record::NS)
+ continue;
+ found = true;
+ resp.ret = (r.type == NameServer::Record::NS) ? RP_RES_NS
+ : RP_RES_A;
+ resp.ttl = z.ttl;
+ resp.ips.push_back(r.ip);
+ }
+ }
+ return found;
+ }
+};
+
+/// Resuelve un nombre de forma directa (no recursiva)
+ResolvProtoResponse NameServer::resolv_direct(const Name& n)
+{
+#ifdef DEBUG_TRACE
+ std::cout << "NameServer::resolv_direct()\n";
+#endif
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_direct -> tratando de resolver: "
+ << n << "\n";
+#endif
+ search_zone zs(n);
+ bool found;
+ for (zones_t::const_iterator i = zones.begin(); i!= zones.end(); ++i)
+ if ((found = zs(*i)))
+ break;
+ if (found)
+ {
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_direct -> found (local/hijo): "
+ << zs.resp << "\n";
+#endif
+ return zs.resp;
+ }
+ if (zs.local)
+ {
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_direct -> NOT FOUND (es local pero "
+ "no existe)\n";
+#endif
+ return ResolvProtoResponse(RP_RES_NOTFOUND);
+ }
+ cache_t::iterator i = cache.find(n);
+ if (i != cache.end())
+ {
+ const CacheRecord& cr = i->second;
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_direct -> found (en cache): "
+ << cr << "\n";
+#endif
+ if (unsigned(time(NULL)) > cr.expire) // Expiró
+ {
+ cache.erase(i);
+#ifdef DEBUG_RESOLV
+ std::cout << " -> pero no sirve, expiró "
+ "hace unos " << (time(NULL) - cr.expire) << " segundos\n";
+#endif
+ }
+ else
+ {
+ return ResolvProtoResponse(RP_RES_A, cr.ttl, cr.ips);
+ }
+ }
+ if (zones.size())
+ {
+ // Busco una zona con padre para ver si puedo "trepar"
+ for (zones_t::const_iterator i = zones.begin(); i != zones.end(); ++i)
+ {
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_direct -> evaluando padre "
+ << i->parent << "\n";
+#endif
+ if (i->parent != IPAddr(0))
+ {
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_direct -> found (al padre): "
+ << i->parent << "\n";
+#endif
+ ResolvProtoResponse rpr(RP_RES_NS, i->ttl);
+ rpr.ips.push_back(i->parent);
+ return rpr;
+ }
+ }
+ }
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_direct -> NOT FOUND (no hay padre)\n";
+#endif
+ // No hay padre, no puedo hacer nada más
+ return ResolvProtoResponse(RP_RES_NOTFOUND);
+}
+
+/// Resuelve un nombre de forma recursiva
+ResolvProtoResponse NameServer::resolv_recursive(const Name& n)
+{
+#ifdef DEBUG_TRACE
+ std::cout << "NameServer::resolv_recursive()\n";
+#endif
+ ResolvProtoResponse rpr = resolv_direct(n);
+ switch (rpr.ret)
+ {
+ case RP_RES_NS:
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_recursive -> redirect a "
+ << rpr << "\n";
+#endif
+ return resolv_recursive_r(n, rpr); // Sigo "bajando"
+ case RP_RES_A:
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_recursive -> gotcha! "
+ << rpr << "\n";
+#endif
+ break;
+ }
+ return rpr; // Devuelvo el A o NOTFOUND
+}
+
+/// Resuelve un nombre de forma recursiva entrando a otros ns
+ResolvProtoResponse NameServer::resolv_recursive_r(const Name& n,
+ ResolvProtoResponse rpr)
+{
+#ifdef DEBUG_TRACE
+ std::cout << "NameServer::resolv_recursive_r()\n";
+#endif
+ ResolvProtoResponse r;
+ for (ResolvProtoResponse::ipvec_t::const_iterator ip = rpr.ips.begin();
+ ip != rpr.ips.end(); ++ip)
+ {
+ r = query(n, *ip);
+ switch (r.ret)
+ {
+ case RP_RES_NS:
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_recursive_r -> redirect a "
+ << r << "\n";
+#endif
+ return resolv_recursive_r(n, r); // Sigo "bajando"
+ case RP_RES_NOTFOUND:
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_recursive_r -> NOT FOUND en "
+ << *ip << ", sigo probando\n";
+#endif
+ break; // Sigo probando del mismo nivel
+ case RP_RES_A:
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_recursive_r -> gotcha! "
+ << r << "\n";
+#endif
+ return r; // Gotcha!
+ }
+ }
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::resolv_recursive_r -> NOT FOUND, no hay más "
+ "por hacer\n";
+#endif
+ return r; // NOTFOUND
+}
+
+/// Consulta a otro name server sobre un nombre
+ResolvProtoResponse NameServer::query(const Name& n, const IPAddr& ip)
+{
+#ifdef DEBUG_TRACE
+ std::cout << "NameServer::query()\n";
+#endif
+ ResolvProtoRequest r(std::string(n), RP_REQ_DIRECT);
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::query -> pidiendo " << r << " a " << ip << "\n";
+#endif
+ // Envía a través de la cola de envío
+ snd_que.transmit(std::string(r), ip);
+ Dev::mac_type mac = ip;
+ std::string buf = res_que.receive(mac);
+ ResolvProtoResponse resp(buf);
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::query -> recibido " << resp << " de "
+ << ip << "\n";
+#endif
+ // Agrego a cache
+ if (resp.ret == RP_RES_A)
+ cache[n] = CacheRecord(time(NULL) + resp.ttl, resp.ttl, resp.ips);
+ return resp;
+}
+
+void NameServer::recv_loop()
+{
+#ifdef DEBUG_TRACE
+ std::cout << "NameServer::recv_loop()\n";
+#endif
+ while (true)
+ {
+ IPAddr src, dst;
+ uint8_t proto;
+ std::string s = ipin.recv(proto, src, dst);
+#ifdef DEBUG_RESOLV
+ std::cout << "NameServer::recv_loop() -> recibido len=" << s.size()
+ << " de " << src << " para " << dst << " (proto = "
+ << unsigned(proto) << ")\n";
+#endif
+ if (proto == RESOLV_PROTO) // Si es para nosotros
+ {
+ rp_pkt_type_t type;
+ memcpy(&type, s.data(), sizeof(uint8_t));
+ switch (type)
+ {
+ // Request
+ case RP_REQ_DIRECT:
+ case RP_REQ_RECURSIVE:
+#ifdef DEBUG_RESOLV
+ std::cout << "---> " << ResolvProtoRequest(s) << "\n";
+#endif
+ req_que.transmit(s, src); // Encolo
+ break;
+ // Response
+ default:
+#ifdef DEBUG_RESOLV
+ std::cout << "---> " << ResolvProtoResponse(s) << "\n";
+#endif
+ res_que.transmit(s, src); // Encolo
+ }
+ }
+ }
+}
+
+void NameServer::send_loop()
projects / z.facultad / 75.74 / practicos.git / blobdiff