mirror of
https://github.com/qemu/qemu.git
synced 2024-12-14 23:13:29 +08:00
c734cd40a1
The requirement to specify the parent class type makes the macro harder to use and easy to misuse (silent bugs can be introduced if the wrong struct type is specified). Simplify the macro by just not declaring any class struct, allowing us to remove the class_size field from the TypeInfo variables for those types. Signed-off-by: Eduardo Habkost <ehabkost@redhat.com> Reviewed-by: Philippe Mathieu-Daudé <philmd@redhat.com> Message-Id: <20200916182519.415636-3-ehabkost@redhat.com> Signed-off-by: Eduardo Habkost <ehabkost@redhat.com>
280 lines
7.9 KiB
C
280 lines
7.9 KiB
C
/*
|
|
* QEMU DNS resolver
|
|
*
|
|
* Copyright (c) 2016 Red Hat, Inc.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include "qemu/osdep.h"
|
|
#include "io/dns-resolver.h"
|
|
#include "qapi/clone-visitor.h"
|
|
#include "qapi/qapi-visit-sockets.h"
|
|
#include "qemu/sockets.h"
|
|
#include "qapi/error.h"
|
|
#include "qemu/cutils.h"
|
|
#include "qemu/module.h"
|
|
|
|
#ifndef AI_NUMERICSERV
|
|
# define AI_NUMERICSERV 0
|
|
#endif
|
|
|
|
static QIODNSResolver *instance;
|
|
static GOnce instance_init = G_ONCE_INIT;
|
|
|
|
static gpointer qio_dns_resolve_init_instance(gpointer unused G_GNUC_UNUSED)
|
|
{
|
|
instance = QIO_DNS_RESOLVER(object_new(TYPE_QIO_DNS_RESOLVER));
|
|
return NULL;
|
|
}
|
|
|
|
QIODNSResolver *qio_dns_resolver_get_instance(void)
|
|
{
|
|
g_once(&instance_init, qio_dns_resolve_init_instance, NULL);
|
|
return instance;
|
|
}
|
|
|
|
static int qio_dns_resolver_lookup_sync_inet(QIODNSResolver *resolver,
|
|
SocketAddress *addr,
|
|
size_t *naddrs,
|
|
SocketAddress ***addrs,
|
|
Error **errp)
|
|
{
|
|
struct addrinfo ai, *res, *e;
|
|
InetSocketAddress *iaddr = &addr->u.inet;
|
|
char port[33];
|
|
char uaddr[INET6_ADDRSTRLEN + 1];
|
|
char uport[33];
|
|
int rc;
|
|
Error *err = NULL;
|
|
size_t i;
|
|
|
|
*naddrs = 0;
|
|
*addrs = NULL;
|
|
|
|
memset(&ai, 0, sizeof(ai));
|
|
ai.ai_flags = AI_PASSIVE;
|
|
if (iaddr->has_numeric && iaddr->numeric) {
|
|
ai.ai_flags |= AI_NUMERICHOST | AI_NUMERICSERV;
|
|
}
|
|
ai.ai_family = inet_ai_family_from_address(iaddr, &err);
|
|
ai.ai_socktype = SOCK_STREAM;
|
|
|
|
if (err) {
|
|
error_propagate(errp, err);
|
|
return -1;
|
|
}
|
|
|
|
if (iaddr->host == NULL) {
|
|
error_setg(errp, "host not specified");
|
|
return -1;
|
|
}
|
|
if (iaddr->port != NULL) {
|
|
pstrcpy(port, sizeof(port), iaddr->port);
|
|
} else {
|
|
port[0] = '\0';
|
|
}
|
|
|
|
rc = getaddrinfo(strlen(iaddr->host) ? iaddr->host : NULL,
|
|
strlen(port) ? port : NULL, &ai, &res);
|
|
if (rc != 0) {
|
|
error_setg(errp, "address resolution failed for %s:%s: %s",
|
|
iaddr->host, port, gai_strerror(rc));
|
|
return -1;
|
|
}
|
|
|
|
for (e = res; e != NULL; e = e->ai_next) {
|
|
(*naddrs)++;
|
|
}
|
|
|
|
*addrs = g_new0(SocketAddress *, *naddrs);
|
|
|
|
/* create socket + bind */
|
|
for (i = 0, e = res; e != NULL; i++, e = e->ai_next) {
|
|
SocketAddress *newaddr = g_new0(SocketAddress, 1);
|
|
|
|
newaddr->type = SOCKET_ADDRESS_TYPE_INET;
|
|
|
|
getnameinfo((struct sockaddr *)e->ai_addr, e->ai_addrlen,
|
|
uaddr, INET6_ADDRSTRLEN, uport, 32,
|
|
NI_NUMERICHOST | NI_NUMERICSERV);
|
|
|
|
newaddr->u.inet = (InetSocketAddress){
|
|
.host = g_strdup(uaddr),
|
|
.port = g_strdup(uport),
|
|
.has_numeric = true,
|
|
.numeric = true,
|
|
.has_to = iaddr->has_to,
|
|
.to = iaddr->to,
|
|
.has_ipv4 = iaddr->has_ipv4,
|
|
.ipv4 = iaddr->ipv4,
|
|
.has_ipv6 = iaddr->has_ipv6,
|
|
.ipv6 = iaddr->ipv6,
|
|
};
|
|
|
|
(*addrs)[i] = newaddr;
|
|
}
|
|
freeaddrinfo(res);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int qio_dns_resolver_lookup_sync_nop(QIODNSResolver *resolver,
|
|
SocketAddress *addr,
|
|
size_t *naddrs,
|
|
SocketAddress ***addrs,
|
|
Error **errp)
|
|
{
|
|
*naddrs = 1;
|
|
*addrs = g_new0(SocketAddress *, 1);
|
|
(*addrs)[0] = QAPI_CLONE(SocketAddress, addr);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
int qio_dns_resolver_lookup_sync(QIODNSResolver *resolver,
|
|
SocketAddress *addr,
|
|
size_t *naddrs,
|
|
SocketAddress ***addrs,
|
|
Error **errp)
|
|
{
|
|
switch (addr->type) {
|
|
case SOCKET_ADDRESS_TYPE_INET:
|
|
return qio_dns_resolver_lookup_sync_inet(resolver,
|
|
addr,
|
|
naddrs,
|
|
addrs,
|
|
errp);
|
|
|
|
case SOCKET_ADDRESS_TYPE_UNIX:
|
|
case SOCKET_ADDRESS_TYPE_VSOCK:
|
|
case SOCKET_ADDRESS_TYPE_FD:
|
|
return qio_dns_resolver_lookup_sync_nop(resolver,
|
|
addr,
|
|
naddrs,
|
|
addrs,
|
|
errp);
|
|
|
|
default:
|
|
abort();
|
|
}
|
|
}
|
|
|
|
|
|
struct QIODNSResolverLookupData {
|
|
SocketAddress *addr;
|
|
SocketAddress **addrs;
|
|
size_t naddrs;
|
|
};
|
|
|
|
|
|
static void qio_dns_resolver_lookup_data_free(gpointer opaque)
|
|
{
|
|
struct QIODNSResolverLookupData *data = opaque;
|
|
size_t i;
|
|
|
|
qapi_free_SocketAddress(data->addr);
|
|
for (i = 0; i < data->naddrs; i++) {
|
|
qapi_free_SocketAddress(data->addrs[i]);
|
|
}
|
|
|
|
g_free(data->addrs);
|
|
g_free(data);
|
|
}
|
|
|
|
|
|
static void qio_dns_resolver_lookup_worker(QIOTask *task,
|
|
gpointer opaque)
|
|
{
|
|
QIODNSResolver *resolver = QIO_DNS_RESOLVER(qio_task_get_source(task));
|
|
struct QIODNSResolverLookupData *data = opaque;
|
|
Error *err = NULL;
|
|
|
|
qio_dns_resolver_lookup_sync(resolver,
|
|
data->addr,
|
|
&data->naddrs,
|
|
&data->addrs,
|
|
&err);
|
|
if (err) {
|
|
qio_task_set_error(task, err);
|
|
} else {
|
|
qio_task_set_result_pointer(task, opaque, NULL);
|
|
}
|
|
|
|
object_unref(OBJECT(resolver));
|
|
}
|
|
|
|
|
|
void qio_dns_resolver_lookup_async(QIODNSResolver *resolver,
|
|
SocketAddress *addr,
|
|
QIOTaskFunc func,
|
|
gpointer opaque,
|
|
GDestroyNotify notify)
|
|
{
|
|
QIOTask *task;
|
|
struct QIODNSResolverLookupData *data =
|
|
g_new0(struct QIODNSResolverLookupData, 1);
|
|
|
|
data->addr = QAPI_CLONE(SocketAddress, addr);
|
|
|
|
task = qio_task_new(OBJECT(resolver), func, opaque, notify);
|
|
|
|
qio_task_run_in_thread(task,
|
|
qio_dns_resolver_lookup_worker,
|
|
data,
|
|
qio_dns_resolver_lookup_data_free,
|
|
NULL);
|
|
}
|
|
|
|
|
|
void qio_dns_resolver_lookup_result(QIODNSResolver *resolver,
|
|
QIOTask *task,
|
|
size_t *naddrs,
|
|
SocketAddress ***addrs)
|
|
{
|
|
struct QIODNSResolverLookupData *data =
|
|
qio_task_get_result_pointer(task);
|
|
size_t i;
|
|
|
|
*naddrs = 0;
|
|
*addrs = NULL;
|
|
if (!data) {
|
|
return;
|
|
}
|
|
|
|
*naddrs = data->naddrs;
|
|
*addrs = g_new0(SocketAddress *, data->naddrs);
|
|
for (i = 0; i < data->naddrs; i++) {
|
|
(*addrs)[i] = QAPI_CLONE(SocketAddress, data->addrs[i]);
|
|
}
|
|
}
|
|
|
|
|
|
static const TypeInfo qio_dns_resolver_info = {
|
|
.parent = TYPE_OBJECT,
|
|
.name = TYPE_QIO_DNS_RESOLVER,
|
|
.instance_size = sizeof(QIODNSResolver),
|
|
};
|
|
|
|
|
|
static void qio_dns_resolver_register_types(void)
|
|
{
|
|
type_register_static(&qio_dns_resolver_info);
|
|
}
|
|
|
|
|
|
type_init(qio_dns_resolver_register_types);
|