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

/*
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;
}