You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
180 lines
3.7 KiB
180 lines
3.7 KiB
/*
|
|
Copyright Jeroen Vreeken (jeroen@vreeken.net), 2015
|
|
|
|
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 3 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.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
#define _GNU_SOURCE
|
|
|
|
#include <dml/dml_client.h>
|
|
#include <dml/dml_server.h>
|
|
#include <dml_config.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <stdbool.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <netinet/in.h>
|
|
#include <netinet/tcp.h>
|
|
#include <string.h>
|
|
#include <netdb.h>
|
|
#include <signal.h>
|
|
#include <resolv.h>
|
|
#include <errno.h>
|
|
|
|
struct dml_client {
|
|
int fd;
|
|
GIOChannel *io;
|
|
|
|
char *host;
|
|
unsigned short port;
|
|
|
|
void (*connect_cb)(struct dml_client *dc, void *arg);
|
|
void *arg;
|
|
};
|
|
|
|
struct dml_client *dml_client_create(char *host, unsigned short port, void (*cb)(struct dml_client *dc, void *arg), void *arg)
|
|
{
|
|
struct dml_client *dc;
|
|
|
|
signal(SIGPIPE, SIG_IGN);
|
|
|
|
if (!port)
|
|
port = DML_SERVER_PORT;
|
|
|
|
dc = calloc(1, sizeof(struct dml_client));
|
|
if (!dc)
|
|
goto err_calloc;
|
|
|
|
dc->host = strdup(host);
|
|
if (!dc->host)
|
|
goto err_strdup;
|
|
|
|
dc->port = port;
|
|
dc->fd = -1;
|
|
dc->io = NULL;
|
|
dc->connect_cb = cb;
|
|
dc->arg = arg;
|
|
|
|
return dc;
|
|
err_strdup:
|
|
free(dc);
|
|
err_calloc:
|
|
return NULL;
|
|
}
|
|
|
|
int dml_client_destroy(struct dml_client *dc)
|
|
{
|
|
if (dc->fd >= 0)
|
|
close(dc->fd);
|
|
g_io_channel_unref(dc->io);
|
|
|
|
free(dc->host);
|
|
free(dc);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
static gboolean dml_client_connect_success(GIOChannel *source, GIOCondition condition, gpointer arg)
|
|
{
|
|
struct dml_client *dc = arg;
|
|
|
|
g_source_remove_by_user_data(dc);
|
|
|
|
setsockopt (dc->fd, IPPROTO_TCP, TCP_NODELAY, &(int){1}, sizeof (int));
|
|
setsockopt (dc->fd, SOL_SOCKET, SO_KEEPALIVE, &(int){1}, sizeof (int));
|
|
#ifdef TCP_KEEPINTVL
|
|
setsockopt (dc->fd, IPPROTO_TCP, TCP_KEEPIDLE, &(int){60}, sizeof(int));
|
|
#endif
|
|
|
|
dc->connect_cb(dc, dc->arg);
|
|
|
|
//TODO
|
|
return FALSE;
|
|
}
|
|
|
|
int dml_client_connect(struct dml_client *dc)
|
|
{
|
|
struct addrinfo *result;
|
|
struct addrinfo *entry;
|
|
struct addrinfo hints = { 0 };
|
|
int error;
|
|
int sock = -1;
|
|
char *port;
|
|
|
|
if (asprintf(&port, "%d", dc->port) < 0)
|
|
goto err_asprintf;
|
|
|
|
hints.ai_family = AF_UNSPEC;
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
|
|
error = getaddrinfo(dc->host, port, &hints, &result);
|
|
if (error) {
|
|
res_init();
|
|
goto err_getaddrinfo;
|
|
}
|
|
for (entry = result; entry; entry = entry->ai_next) {
|
|
if (entry->ai_family == AF_INET6) {
|
|
bool ipv6 = atoi(dml_config_value("ipv6", NULL, "1"));
|
|
if (!ipv6) {
|
|
continue;
|
|
}
|
|
}
|
|
sock = socket(entry->ai_family, entry->ai_socktype,
|
|
entry->ai_protocol);
|
|
if (sock >= 0) {
|
|
int flags = fcntl(sock, F_GETFL, 0);
|
|
if (flags >= 0)
|
|
fcntl(sock, F_SETFL, flags | O_NONBLOCK);
|
|
|
|
if (connect(sock, entry->ai_addr, entry->ai_addrlen) &&
|
|
errno != EINPROGRESS) {
|
|
fprintf(stderr, "connect failed %d\n", errno);
|
|
close(sock);
|
|
sock = -1;
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
freeaddrinfo(result);
|
|
|
|
if (sock < 0)
|
|
goto err_connect;
|
|
|
|
free(port);
|
|
dc->fd = sock;
|
|
dc->io = g_io_channel_unix_new (sock);
|
|
g_io_channel_set_encoding(dc->io, NULL, NULL);
|
|
|
|
g_io_add_watch(dc->io, G_IO_OUT, dml_client_connect_success, dc);
|
|
|
|
return 0;
|
|
|
|
err_connect:
|
|
err_getaddrinfo:
|
|
free(port);
|
|
err_asprintf:
|
|
return -1;
|
|
}
|
|
|
|
int dml_client_fd_get(struct dml_client *dc)
|
|
{
|
|
return dc->fd;
|
|
}
|
|
|