mirror of
https://github.com/ZoiteChat/zoitechat.git
synced 2026-06-22 13:59:26 +00:00
Add DCC passive-first + NAT mapping prefs
This commit is contained in:
@@ -87,6 +87,41 @@ static gboolean dcc_read (GIOChannel *, GIOCondition, struct DCC *);
|
||||
static gboolean dcc_read_ack (GIOChannel *source, GIOCondition condition, struct DCC *dcc);
|
||||
static int dcc_check_timeouts (void);
|
||||
|
||||
|
||||
static int
|
||||
nat_map_run (int add, int port, int proto, int lease)
|
||||
{
|
||||
char cmd[512];
|
||||
int status = 0;
|
||||
if (add)
|
||||
g_snprintf (cmd, sizeof (cmd), "upnpc -a 127.0.0.1 %d %d TCP %d >/dev/null 2>&1", port, port, lease);
|
||||
else
|
||||
g_snprintf (cmd, sizeof (cmd), "upnpc -d %d TCP >/dev/null 2>&1", port);
|
||||
if (!g_spawn_command_line_sync (cmd, NULL, NULL, &status, NULL))
|
||||
return FALSE;
|
||||
return status == 0;
|
||||
}
|
||||
|
||||
static void
|
||||
dcc_nat_unmap (struct DCC *dcc)
|
||||
{
|
||||
if (!dcc->map_port)
|
||||
return;
|
||||
nat_map_run (0, dcc->map_port, dcc->type, 0);
|
||||
dcc->map_port = 0;
|
||||
dcc->map_next_refresh = 0;
|
||||
}
|
||||
|
||||
static void
|
||||
dcc_nat_refresh (struct DCC *dcc)
|
||||
{
|
||||
if (!dcc->map_port || prefs.hex_dcc_nat_lease <= 0)
|
||||
return;
|
||||
if (time (0) < dcc->map_next_refresh)
|
||||
return;
|
||||
if (nat_map_run (1, dcc->map_port, dcc->type, prefs.hex_dcc_nat_lease))
|
||||
dcc->map_next_refresh = time (0) + prefs.hex_dcc_nat_lease / 2;
|
||||
}
|
||||
static int new_id(void)
|
||||
{
|
||||
static int id = 0;
|
||||
@@ -254,7 +289,8 @@ dcc_check_timeouts (void)
|
||||
switch (dcc->dccstat)
|
||||
{
|
||||
case STAT_ACTIVE:
|
||||
dcc_calc_cps (dcc);
|
||||
dcc_calc_cps (dcc);
|
||||
dcc_nat_refresh (dcc);
|
||||
fe_dcc_update (dcc);
|
||||
|
||||
if (dcc->type == TYPE_SEND || dcc->type == TYPE_RECV)
|
||||
@@ -390,6 +426,7 @@ dcc_close (struct DCC *dcc, enum dcc_state dccstat, int destroy)
|
||||
}
|
||||
|
||||
dcc_remove_from_sum (dcc);
|
||||
dcc_nat_unmap (dcc);
|
||||
|
||||
if (dcc->fp != -1)
|
||||
{
|
||||
@@ -1698,6 +1735,11 @@ dcc_listen_init (struct DCC *dcc, session *sess)
|
||||
getsockname (dcc->sok, (struct sockaddr *) &SAddr, &len);
|
||||
|
||||
dcc->port = ntohs (SAddr.sin_port);
|
||||
if (prefs.hex_dcc_nat_map && nat_map_run (1, dcc->port, dcc->type, prefs.hex_dcc_nat_lease))
|
||||
{
|
||||
dcc->map_port = dcc->port;
|
||||
dcc->map_next_refresh = time (0) + prefs.hex_dcc_nat_lease / 2;
|
||||
}
|
||||
|
||||
dcc->addr = dcc_get_my_address (sess);
|
||||
|
||||
@@ -1846,8 +1888,19 @@ dcc_send (struct session *sess, char *to, char *filename, gint64 maxcps, int pas
|
||||
return;
|
||||
}
|
||||
|
||||
if (passive || dcc_listen_init (dcc, sess))
|
||||
if ((passive || prefs.hex_dcc_passive_prefer) || dcc_listen_init (dcc, sess))
|
||||
{
|
||||
if (passive || prefs.hex_dcc_passive_prefer)
|
||||
{
|
||||
guint32 offer_addr = dcc_get_my_address (sess);
|
||||
struct sockaddr_in saddr;
|
||||
socklen_t slen = sizeof (saddr);
|
||||
memset (&saddr, 0, sizeof (saddr));
|
||||
getsockname (dcc->serv->sok, (struct sockaddr *) &saddr, &slen);
|
||||
if (offer_addr == 0)
|
||||
offer_addr = prefs.local_ip != 0xffffffff ? prefs.local_ip : saddr.sin_addr.s_addr;
|
||||
dcc->addr = ntohl (offer_addr);
|
||||
}
|
||||
char havespaces = 0;
|
||||
while (*filename)
|
||||
{
|
||||
@@ -1868,13 +1921,13 @@ dcc_send (struct session *sess, char *to, char *filename, gint64 maxcps, int pas
|
||||
} else
|
||||
fe_dcc_add (dcc);
|
||||
|
||||
if (passive)
|
||||
if (passive || prefs.hex_dcc_passive_prefer)
|
||||
{
|
||||
dcc->pasvid = new_id();
|
||||
g_snprintf (outbuf, sizeof (outbuf), (havespaces) ?
|
||||
"DCC SEND \"%s\" 199 0 %" G_GUINT64_FORMAT " %d" :
|
||||
"DCC SEND %s 199 0 %" G_GUINT64_FORMAT " %d",
|
||||
file_part (dcc->file),
|
||||
"DCC SEND \"%s\" %u 0 %" G_GUINT64_FORMAT " %d" :
|
||||
"DCC SEND %s %u 0 %" G_GUINT64_FORMAT " %d",
|
||||
file_part (dcc->file), dcc->addr,
|
||||
dcc->size, dcc->pasvid);
|
||||
}
|
||||
else
|
||||
@@ -2292,8 +2345,19 @@ dcc_chat (struct session *sess, char *nick, int passive)
|
||||
dcc->dccstat = STAT_QUEUED;
|
||||
dcc->type = TYPE_CHATSEND;
|
||||
dcc->nick = g_strdup (nick);
|
||||
if (passive || dcc_listen_init (dcc, sess))
|
||||
if ((passive || prefs.hex_dcc_passive_prefer) || dcc_listen_init (dcc, sess))
|
||||
{
|
||||
if (passive || prefs.hex_dcc_passive_prefer)
|
||||
{
|
||||
guint32 offer_addr = dcc_get_my_address (sess);
|
||||
struct sockaddr_in saddr;
|
||||
socklen_t slen = sizeof (saddr);
|
||||
memset (&saddr, 0, sizeof (saddr));
|
||||
getsockname (dcc->serv->sok, (struct sockaddr *) &saddr, &slen);
|
||||
if (offer_addr == 0)
|
||||
offer_addr = prefs.local_ip != 0xffffffff ? prefs.local_ip : saddr.sin_addr.s_addr;
|
||||
dcc->addr = ntohl (offer_addr);
|
||||
}
|
||||
if (prefs.hex_gui_autoopen_chat)
|
||||
{
|
||||
if (fe_dcc_open_chat_win (TRUE)) /* already open? add only */
|
||||
@@ -2301,11 +2365,11 @@ dcc_chat (struct session *sess, char *nick, int passive)
|
||||
} else
|
||||
fe_dcc_add (dcc);
|
||||
|
||||
if (passive)
|
||||
if (passive || prefs.hex_dcc_passive_prefer)
|
||||
{
|
||||
dcc->pasvid = new_id ();
|
||||
g_snprintf (outbuf, sizeof (outbuf), "DCC CHAT chat 199 %d %d",
|
||||
dcc->port, dcc->pasvid);
|
||||
g_snprintf (outbuf, sizeof (outbuf), "DCC CHAT chat %u 0 %d",
|
||||
dcc->addr, dcc->pasvid);
|
||||
} else
|
||||
{
|
||||
g_snprintf (outbuf, sizeof (outbuf), "DCC CHAT chat %u %d",
|
||||
|
||||
Reference in New Issue
Block a user