2020-02-21 04:21:16 +00:00
/*
* $ Id $
*
Copyright ( c ) 2016 - 2020 Chung , Hyung - Hwan . All rights reserved .
Redistribution and use in source and binary forms , with or without
modification , are permitted provided that the following conditions
are met :
1. Redistributions of source code must retain the above copyright
notice , this list of conditions and the following disclaimer .
2. Redistributions in binary form must reproduce the above copyright
notice , this list of conditions and the following disclaimer in the
documentation and / or other materials provided with the distribution .
THIS SOFTWARE IS PROVIDED BY THE AUTHOR " AS IS " AND ANY EXPRESS OR
IMPLIED WARRANTIES , INCLUDING , BUT NOT LIMITED TO , THE IMPLIED WAfRRANTIES
OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED .
IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT , INDIRECT ,
INCIDENTAL , SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES ( INCLUDING , BUT
NOT LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ; LOSS OF USE ,
DATA , OR PROFITS ; OR BUSINESS INTERRUPTION ) HOWEVER CAUSED AND ON ANY
THEORY OF LIABILITY , WHETHER IN CONTRACT , STRICT LIABILITY , OR TORT
( INCLUDING NEGLIGENCE OR OTHERWISE ) ARISING IN ANY WAY OUT OF THE USE OF
THIS SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
*/
# include <mio-dns.h>
# include <mio-sck.h>
# include "mio-prv.h"
# include <netinet/in.h>
struct mio_svc_dns_t
{
2020-05-04 17:51:47 +00:00
MIO_SVC_HEADER ;
/*MIO_DNS_SVC_HEADER;*/
2020-02-21 04:21:16 +00:00
} ;
struct mio_svc_dnc_t
{
2020-05-04 17:51:47 +00:00
MIO_SVC_HEADER ;
/*MIO_DNS_SVC_HEADER;*/
2020-02-21 04:21:16 +00:00
mio_dev_sck_t * udp_sck ;
mio_dev_sck_t * tcp_sck ;
mio_skad_t serv_addr ;
mio_ntime_t send_tmout ;
mio_ntime_t reply_tmout ; /* default reply timeout */
2020-03-02 04:58:22 +00:00
/* For a question sent out, it may wait for a corresponding answer.
* if max_tries is greater than 0 , sending and waiting is limited
* to this value over udp and to 1 over tcp . if max_tries is 0 ,
* it sends out the question but never waits for a response .
* For a non - question message sent out , it never waits for a response
* regardless of max_tries . */
mio_oow_t max_tries ;
2020-02-21 04:21:16 +00:00
2020-07-02 14:21:47 +00:00
mio_dns_cookie_t cookie ;
2020-02-21 04:21:16 +00:00
mio_oow_t seq ;
mio_dns_msg_t * pending_req ;
} ;
struct dnc_sck_xtn_t
{
mio_svc_dnc_t * dnc ;
2020-02-29 16:56:36 +00:00
struct
{
mio_uint8_t * ptr ;
mio_oow_t len ;
mio_oow_t capa ;
} rbuf ; /* used by tcp socket */
2020-02-21 04:21:16 +00:00
} ;
typedef struct dnc_sck_xtn_t dnc_sck_xtn_t ;
/* ----------------------------------------------------------------------- */
struct dnc_dns_msg_xtn_t
{
2020-02-28 07:41:25 +00:00
mio_dev_sck_t * dev ;
2020-02-21 04:21:16 +00:00
mio_tmridx_t rtmridx ;
mio_dns_msg_t * prev ;
mio_dns_msg_t * next ;
mio_skad_t servaddr ;
2020-02-27 17:34:27 +00:00
mio_svc_dnc_on_done_t on_done ;
2020-02-21 04:21:16 +00:00
mio_ntime_t wtmout ;
mio_ntime_t rtmout ;
int rmaxtries ; /* maximum number of tries to receive a reply */
int rtries ; /* number of tries made so far */
2020-03-01 17:43:42 +00:00
int pending ;
2020-02-21 04:21:16 +00:00
} ;
typedef struct dnc_dns_msg_xtn_t dnc_dns_msg_xtn_t ;
# if defined(MIO_HAVE_INLINE)
static MIO_INLINE dnc_dns_msg_xtn_t * dnc_dns_msg_getxtn ( mio_dns_msg_t * msg ) { return ( dnc_dns_msg_xtn_t * ) ( ( mio_uint8_t * ) mio_dns_msg_to_pkt ( msg ) + msg - > pktalilen ) ; }
# else
# define dnc_dns_msg_getxtn(msg) ((dnc_dns_msg_xtn_t*)((mio_uint8_t*)mio_dns_msg_to_pkt(msg) + msg->pktalilen))
# endif
static MIO_INLINE void chain_pending_dns_reqmsg ( mio_svc_dnc_t * dnc , mio_dns_msg_t * msg )
{
if ( dnc - > pending_req )
{
dnc_dns_msg_getxtn ( dnc - > pending_req ) - > prev = msg ;
dnc_dns_msg_getxtn ( msg ) - > next = dnc - > pending_req ;
}
dnc - > pending_req = msg ;
2020-03-01 17:43:42 +00:00
dnc_dns_msg_getxtn ( msg ) - > pending = 1 ;
2020-02-21 04:21:16 +00:00
}
static MIO_INLINE void unchain_pending_dns_reqmsg ( mio_svc_dnc_t * dnc , mio_dns_msg_t * msg )
{
dnc_dns_msg_xtn_t * msgxtn = dnc_dns_msg_getxtn ( msg ) ;
if ( msgxtn - > next ) dnc_dns_msg_getxtn ( msgxtn - > next ) - > prev = msgxtn - > prev ;
if ( msgxtn - > prev ) dnc_dns_msg_getxtn ( msgxtn - > prev ) - > next = msgxtn - > next ;
else dnc - > pending_req = msgxtn - > next ;
2020-03-01 17:43:42 +00:00
dnc_dns_msg_getxtn ( msg ) - > pending = 0 ;
}
static mio_dns_msg_t * make_dns_msg ( mio_svc_dnc_t * dnc , mio_dns_bhdr_t * bdns , mio_dns_bqr_t * qr , mio_oow_t qr_count , mio_dns_brr_t * rr , mio_oow_t rr_count , mio_dns_bedns_t * edns , mio_svc_dnc_on_done_t on_done , mio_oow_t xtnsize )
{
mio_dns_msg_t * msg ;
dnc_dns_msg_xtn_t * msgxtn ;
msg = mio_dns_make_msg ( dnc - > mio , bdns , qr , qr_count , rr , rr_count , edns , MIO_SIZEOF ( * msgxtn ) + xtnsize ) ;
2020-05-05 15:12:08 +00:00
if ( MIO_UNLIKELY ( ! msg ) ) return MIO_NULL ;
2020-03-01 17:43:42 +00:00
if ( bdns - > id < 0 )
{
mio_dns_pkt_t * pkt = mio_dns_msg_to_pkt ( msg ) ;
pkt - > id = mio_hton16 ( dnc - > seq ) ;
dnc - > seq + + ;
}
msgxtn = dnc_dns_msg_getxtn ( msg ) ;
msgxtn - > dev = dnc - > udp_sck ;
msgxtn - > rtmridx = MIO_TMRIDX_INVALID ;
msgxtn - > on_done = on_done ;
msgxtn - > wtmout = dnc - > send_tmout ;
msgxtn - > rtmout = dnc - > reply_tmout ;
2020-03-02 04:58:22 +00:00
msgxtn - > rmaxtries = dnc - > max_tries ;
2020-03-01 17:43:42 +00:00
msgxtn - > rtries = 0 ;
msgxtn - > servaddr = dnc - > serv_addr ;
return msg ;
2020-02-21 04:21:16 +00:00
}
static void release_dns_msg ( mio_svc_dnc_t * dnc , mio_dns_msg_t * msg )
{
mio_t * mio = dnc - > mio ;
dnc_dns_msg_xtn_t * msgxtn = dnc_dns_msg_getxtn ( msg ) ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - releasing dns message - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-21 04:21:16 +00:00
if ( msg = = dnc - > pending_req | | msgxtn - > next | | msgxtn - > prev )
{
/* it's chained in the pending request. unchain it */
unchain_pending_dns_reqmsg ( dnc , msg ) ;
}
if ( msgxtn - > rtmridx ! = MIO_TMRIDX_INVALID )
{
mio_deltmrjob ( mio , msgxtn - > rtmridx ) ;
MIO_ASSERT ( mio , msgxtn - > rtmridx = = MIO_TMRIDX_INVALID ) ;
}
2020-02-25 07:09:20 +00:00
/* TODO: add it to the free msg list instead of just freeing it. */
2020-02-21 04:21:16 +00:00
mio_dns_free_msg ( dnc - > mio , msg ) ;
}
2020-02-21 09:42:10 +00:00
/* ----------------------------------------------------------------------- */
2020-02-29 16:56:36 +00:00
static int handle_tcp_packet ( mio_dev_sck_t * dev , mio_dns_pkt_t * pkt , mio_uint16_t pktlen )
{
mio_t * mio = dev - > mio ;
dnc_sck_xtn_t * sckxtn = mio_dev_sck_getxtn ( dev ) ;
mio_svc_dnc_t * dnc = sckxtn - > dnc ;
mio_uint16_t id ;
mio_dns_msg_t * reqmsg ;
if ( ! pkt - > qr )
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG0 ( mio , " DNC - dropping dns request received over tcp ... \n " ) ; /* TODO: add source info */
2020-02-29 16:56:36 +00:00
return 0 ; /* drop request. nothing to do */
}
id = mio_ntoh16 ( pkt - > id ) ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - got dns response over tcp - msgid:%d \n " , id ) ;
2020-02-29 16:56:36 +00:00
reqmsg = dnc - > pending_req ;
while ( reqmsg )
{
mio_dns_pkt_t * reqpkt = mio_dns_msg_to_pkt ( reqmsg ) ;
dnc_dns_msg_xtn_t * reqmsgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
if ( dev = = ( mio_dev_sck_t * ) reqmsgxtn - > dev & & pkt - > id = = reqpkt - > id )
{
if ( MIO_LIKELY ( reqmsgxtn - > on_done ) ) reqmsgxtn - > on_done ( dnc , reqmsg , MIO_ENOERR , pkt , pktlen ) ;
release_dns_msg ( dnc , reqmsg ) ;
return 0 ;
}
reqmsg = reqmsgxtn - > next ;
}
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - unknown dns response over tcp... msgid:%d \n " , id ) ; /* TODO: add source info */
2020-02-29 16:56:36 +00:00
return 0 ;
}
static MIO_INLINE int copy_data_to_sck_rbuf ( mio_dev_sck_t * dev , const void * data , mio_uint16_t dlen )
{
dnc_sck_xtn_t * sckxtn = mio_dev_sck_getxtn ( dev ) ;
if ( sckxtn - > rbuf . capa - sckxtn - > rbuf . len < dlen )
{
mio_uint16_t newcapa ;
mio_uint8_t * tmp ;
newcapa = sckxtn - > rbuf . len + dlen ;
newcapa = MIO_ALIGN_POW2 ( newcapa , 512 ) ;
tmp = mio_reallocmem ( dev - > mio , sckxtn - > rbuf . ptr , newcapa ) ;
if ( ! tmp ) return - 1 ;
sckxtn - > rbuf . capa = newcapa ;
sckxtn - > rbuf . ptr = tmp ;
}
MIO_MEMCPY ( & sckxtn - > rbuf . ptr [ sckxtn - > rbuf . len ] , data , dlen ) ;
sckxtn - > rbuf . len + = dlen ;
return 0 ;
}
2020-02-21 09:42:10 +00:00
static int on_tcp_read ( mio_dev_sck_t * dev , const void * data , mio_iolen_t dlen , const mio_skad_t * srcaddr )
{
2020-02-27 17:34:27 +00:00
mio_t * mio = dev - > mio ;
2020-02-29 16:56:36 +00:00
dnc_sck_xtn_t * sckxtn = mio_dev_sck_getxtn ( dev ) ;
2020-02-28 08:10:29 +00:00
mio_uint16_t pktlen ;
2020-02-29 16:56:36 +00:00
mio_uint8_t * dptr ;
2020-02-28 08:10:29 +00:00
mio_iolen_t rem ;
2020-02-27 17:34:27 +00:00
2020-02-25 07:09:20 +00:00
if ( MIO_UNLIKELY ( dlen < = - 1 ) )
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - dns tcp read error ....%js \n " , mio_geterrmsg ( mio ) ) ; /* TODO: add source packet */
2020-02-28 07:41:25 +00:00
goto oops ;
2020-02-25 07:09:20 +00:00
}
2020-02-28 07:41:25 +00:00
else if ( MIO_UNLIKELY ( dlen = = 0 ) )
2020-02-25 07:09:20 +00:00
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG0 ( mio , " DNC - dns tcp read error ...premature tcp socket end \n " ) ; /* TODO: add source packet */
2020-02-28 07:41:25 +00:00
goto oops ;
2020-02-25 07:09:20 +00:00
}
2020-02-27 17:34:27 +00:00
2020-02-29 16:56:36 +00:00
dptr = ( mio_uint8_t * ) data ;
2020-02-28 08:10:29 +00:00
rem = dlen ;
do
{
2020-02-29 16:56:36 +00:00
if ( MIO_UNLIKELY ( sckxtn - > rbuf . len = = 1 ) )
2020-02-28 08:10:29 +00:00
{
2020-02-29 16:56:36 +00:00
pktlen = ( ( mio_uint16_t ) sckxtn - > rbuf . ptr [ 0 ] < < 8 ) | * ( mio_uint8_t * ) dptr ;
if ( MIO_UNLIKELY ( ( rem - 1 ) < pktlen ) ) goto incomplete_data ;
dptr + = 1 ; rem - = 1 ; sckxtn - > rbuf . len = 0 ;
handle_tcp_packet ( dev , ( mio_dns_pkt_t * ) dptr , pktlen ) ;
dptr + = pktlen ; rem - = pktlen ;
2020-02-28 08:10:29 +00:00
}
2020-02-29 16:56:36 +00:00
else if ( MIO_UNLIKELY ( sckxtn - > rbuf . len > 1 ) )
2020-02-28 08:10:29 +00:00
{
2020-02-29 16:56:36 +00:00
mio_uint16_t cplen ;
pktlen = ( ( mio_uint16_t ) sckxtn - > rbuf . ptr [ 0 ] < < 8 ) | sckxtn - > rbuf . ptr [ 1 ] ;
if ( MIO_UNLIKELY ( sckxtn - > rbuf . len - 2 + rem < pktlen ) ) goto incomplete_data ;
cplen = pktlen - ( sckxtn - > rbuf . len - 2 ) ;
if ( copy_data_to_sck_rbuf ( dev , dptr , cplen ) < = - 1 ) goto oops ;
dptr + = cplen ; rem - = cplen ; sckxtn - > rbuf . len = 0 ;
handle_tcp_packet ( dev , ( mio_dns_pkt_t * ) & sckxtn - > rbuf . ptr [ 2 ] , pktlen ) ;
2020-02-28 08:10:29 +00:00
}
else
{
2020-02-29 16:56:36 +00:00
if ( MIO_LIKELY ( rem > = 2 ) )
2020-02-28 08:10:29 +00:00
{
2020-02-29 16:56:36 +00:00
pktlen = ( ( mio_uint16_t ) * ( mio_uint8_t * ) dptr < < 8 ) | * ( ( mio_uint8_t * ) dptr + 1 ) ;
dptr + = 2 ; rem - = 2 ;
if ( MIO_UNLIKELY ( rem < pktlen ) ) goto incomplete_data ;
handle_tcp_packet ( dev , ( mio_dns_pkt_t * ) dptr , pktlen ) ;
dptr + = pktlen ; rem - = pktlen ;
2020-02-28 08:10:29 +00:00
}
else
{
incomplete_data :
2020-02-29 16:56:36 +00:00
if ( copy_data_to_sck_rbuf ( dev , dptr , rem ) < = - 1 ) goto oops ;
2020-02-28 08:10:29 +00:00
rem = 0 ;
}
}
}
while ( rem > 0 ) ;
2020-02-28 07:41:25 +00:00
2020-02-27 17:34:27 +00:00
return 0 ;
2020-02-28 07:41:25 +00:00
oops :
2020-05-07 04:32:32 +00:00
mio_dev_sck_halt ( dev ) ;
2020-02-21 09:42:10 +00:00
return 0 ;
}
2020-02-27 08:40:39 +00:00
static void on_tcp_reply_timeout ( mio_t * mio , const mio_ntime_t * now , mio_tmrjob_t * job )
{
mio_dns_msg_t * reqmsg = ( mio_dns_msg_t * ) job - > ctx ;
2020-02-28 07:41:25 +00:00
dnc_dns_msg_xtn_t * reqmsgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
mio_dev_sck_t * dev = reqmsgxtn - > dev ;
2020-02-27 08:40:39 +00:00
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
2020-02-28 07:41:25 +00:00
MIO_ASSERT ( mio , reqmsgxtn - > rtmridx = = MIO_TMRIDX_INVALID ) ;
2020-02-27 08:40:39 +00:00
MIO_ASSERT ( mio , dev = = dnc - > tcp_sck ) ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - unable to receive dns response in time over TCP - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( reqmsg ) - > id ) ) ;
2020-02-27 08:40:39 +00:00
2020-02-28 07:41:25 +00:00
if ( MIO_LIKELY ( reqmsgxtn - > on_done ) ) reqmsgxtn - > on_done ( dnc , reqmsg , MIO_ETMOUT , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , reqmsg ) ;
2020-02-27 08:40:39 +00:00
}
2020-02-21 09:42:10 +00:00
static int on_tcp_write ( mio_dev_sck_t * dev , mio_iolen_t wrlen , void * wrctx , const mio_skad_t * dstaddr )
{
2020-02-27 08:40:39 +00:00
mio_t * mio = dev - > mio ;
mio_dns_msg_t * msg = ( mio_dns_msg_t * ) wrctx ;
dnc_dns_msg_xtn_t * msgxtn = dnc_dns_msg_getxtn ( msg ) ;
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
2020-02-27 17:34:27 +00:00
mio_errnum_t status ;
2020-02-27 08:40:39 +00:00
if ( wrlen < = - 1 )
{
2020-02-27 17:34:27 +00:00
/* send failure */
status = mio_geterrnum ( mio ) ;
goto finalize ;
2020-02-27 08:40:39 +00:00
}
else if ( mio_dns_msg_to_pkt ( msg ) - > qr = = 0 & & msgxtn - > rmaxtries > 0 )
{
/* question. schedule to wait for response */
mio_tmrjob_t tmrjob ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - sent dns question over tcp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-27 08:40:39 +00:00
MIO_MEMSET ( & tmrjob , 0 , MIO_SIZEOF ( tmrjob ) ) ;
tmrjob . ctx = msg ;
mio_gettime ( mio , & tmrjob . when ) ;
MIO_ADD_NTIME ( & tmrjob . when , & tmrjob . when , & msgxtn - > rtmout ) ;
tmrjob . handler = on_tcp_reply_timeout ;
tmrjob . idxptr = & msgxtn - > rtmridx ;
msgxtn - > rtmridx = mio_instmrjob ( mio , & tmrjob ) ;
if ( msgxtn - > rtmridx = = MIO_TMRIDX_INVALID )
{
/* call the callback to indicate this operation failure in the middle of transaction */
2020-02-27 17:34:27 +00:00
status = mio_geterrnum ( mio ) ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - unable to schedule tcp timeout - msgid: %d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-27 17:34:27 +00:00
goto finalize ;
2020-02-27 08:40:39 +00:00
}
2020-03-01 17:43:42 +00:00
2020-03-02 04:58:22 +00:00
MIO_ASSERT ( mio , msgxtn - > pending ! = 0 ) ;
2020-02-27 08:40:39 +00:00
}
else
{
2020-02-28 07:41:25 +00:00
/* no error. successfuly sent a message. no reply is expected */
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - sent dns message over tcp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-27 17:34:27 +00:00
status = MIO_ENOERR ;
goto finalize ;
2020-02-27 08:40:39 +00:00
}
2020-02-25 07:09:20 +00:00
2020-02-21 09:42:10 +00:00
return 0 ;
2020-02-27 17:34:27 +00:00
finalize :
2020-02-28 07:41:25 +00:00
if ( MIO_LIKELY ( msgxtn - > on_done ) ) msgxtn - > on_done ( dnc , msg , status , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , msg ) ;
2020-02-27 17:34:27 +00:00
return 0 ;
2020-02-21 09:42:10 +00:00
}
2020-03-01 17:43:42 +00:00
static int write_dns_msg_over_tcp ( mio_dev_sck_t * dev , mio_dns_msg_t * msg )
2020-02-21 09:42:10 +00:00
{
mio_t * mio = dev - > mio ;
2020-02-28 07:41:25 +00:00
dnc_dns_msg_xtn_t * msgxtn = dnc_dns_msg_getxtn ( msg ) ;
2020-02-21 09:42:10 +00:00
mio_uint16_t pktlen ;
2020-02-25 07:09:20 +00:00
mio_iovec_t iov [ 2 ] ;
2020-02-21 09:42:10 +00:00
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - sending dns message over tcp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-28 07:41:25 +00:00
pktlen = mio_hton16 ( msg - > pktlen ) ;
2020-02-21 09:42:10 +00:00
2020-02-28 07:41:25 +00:00
MIO_ASSERT ( mio , msgxtn - > rtries = = 0 ) ;
2020-03-01 17:43:42 +00:00
msgxtn - > rtries = 1 ;
2020-02-21 09:42:10 +00:00
2020-02-28 07:41:25 +00:00
/* TODO: Is it better to create 2 byte space when sending UDP and use it here instead of iov? */
2020-02-25 07:09:20 +00:00
iov [ 0 ] . iov_ptr = & pktlen ;
iov [ 0 ] . iov_len = MIO_SIZEOF ( pktlen ) ;
2020-02-28 07:41:25 +00:00
iov [ 1 ] . iov_ptr = mio_dns_msg_to_pkt ( msg ) ;
iov [ 1 ] . iov_len = msg - > pktlen ;
2020-03-01 17:43:42 +00:00
return mio_dev_sck_timedwritev ( dev , iov , MIO_COUNTOF ( iov ) , & msgxtn - > rtmout , msg , MIO_NULL ) ;
2020-02-28 07:41:25 +00:00
}
static void on_tcp_connect ( mio_dev_sck_t * dev )
{
mio_t * mio = dev - > mio ;
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
mio_dns_msg_t * reqmsg ;
MIO_ASSERT ( mio , dev = = dnc - > tcp_sck ) ;
reqmsg = dnc - > pending_req ;
while ( reqmsg )
{
dnc_dns_msg_xtn_t * reqmsgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
mio_dns_msg_t * nextreqmsg = reqmsgxtn - > next ;
2020-03-01 17:43:42 +00:00
if ( reqmsgxtn - > dev = = dev & & reqmsgxtn - > rtries = = 0 )
{
if ( write_dns_msg_over_tcp ( dev , reqmsg ) < = - 1 )
{
if ( MIO_LIKELY ( reqmsgxtn - > on_done ) ) reqmsgxtn - > on_done ( dnc , reqmsg , mio_geterrnum ( mio ) , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , reqmsg ) ;
}
}
2020-02-28 07:41:25 +00:00
reqmsg = nextreqmsg ;
2020-02-21 09:42:10 +00:00
}
}
static void on_tcp_disconnect ( mio_dev_sck_t * dev )
{
mio_t * mio = dev - > mio ;
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
2020-02-28 07:41:25 +00:00
mio_dns_msg_t * reqmsg ;
2020-02-21 09:42:10 +00:00
int status ;
/* UNABLE TO CONNECT or CONNECT TIMED OUT */
2020-02-27 08:40:39 +00:00
status = mio_geterrnum ( mio ) ;
2020-02-27 17:34:27 +00:00
if ( status = = MIO_ENOERR )
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG0 ( mio , " DNC - TCP DISCONNECTED \n " ) ;
2020-02-27 17:34:27 +00:00
}
else
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG2 ( mio , " DNC - TCP UNABLE TO CONNECT %d -> %js \n " , status , mio_errnum_to_errstr ( status ) ) ;
2020-02-27 17:34:27 +00:00
}
2020-02-21 09:42:10 +00:00
2020-02-28 07:41:25 +00:00
reqmsg = dnc - > pending_req ;
while ( reqmsg )
{
dnc_dns_msg_xtn_t * reqmsgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
mio_dns_msg_t * nextreqmsg = reqmsgxtn - > next ;
if ( reqmsgxtn - > dev = = dev )
{
if ( MIO_LIKELY ( reqmsgxtn - > on_done ) ) reqmsgxtn - > on_done ( dnc , reqmsg , MIO_ENORSP , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , reqmsg ) ;
}
reqmsg = nextreqmsg ;
}
2020-02-27 08:40:39 +00:00
/* let's forget about the tcp socket */
dnc - > tcp_sck = MIO_NULL ;
2020-02-21 09:42:10 +00:00
}
static int switch_reqmsg_transport_to_tcp ( mio_svc_dnc_t * dnc , mio_dns_msg_t * reqmsg )
{
mio_t * mio = dnc - > mio ;
dnc_dns_msg_xtn_t * reqmsgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
dnc_sck_xtn_t * sckxtn ;
mio_dev_sck_make_t mkinfo ;
mio_dev_sck_connect_t cinfo ;
2020-02-28 08:10:29 +00:00
/* TODO: more reliable way to check if connection is ok.
* even if tcp_sck is not null , the connection could have been torn down . . . */
2020-02-21 09:42:10 +00:00
if ( ! dnc - > tcp_sck )
{
MIO_MEMSET ( & mkinfo , 0 , MIO_SIZEOF ( mkinfo ) ) ;
switch ( mio_skad_family ( & reqmsgxtn - > servaddr ) )
{
case MIO_AF_INET :
mkinfo . type = MIO_DEV_SCK_TCP4 ;
break ;
case MIO_AF_INET6 :
mkinfo . type = MIO_DEV_SCK_TCP6 ;
break ;
default :
mio_seterrnum ( mio , MIO_EINTERN ) ;
return - 1 ;
}
mkinfo . on_write = on_tcp_write ;
mkinfo . on_read = on_tcp_read ;
mkinfo . on_connect = on_tcp_connect ;
mkinfo . on_disconnect = on_tcp_disconnect ;
dnc - > tcp_sck = mio_dev_sck_make ( mio , MIO_SIZEOF ( * sckxtn ) , & mkinfo ) ;
2020-02-25 07:09:20 +00:00
if ( ! dnc - > tcp_sck ) return - 1 ;
2020-02-21 09:42:10 +00:00
2020-02-28 07:41:25 +00:00
sckxtn = ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dnc - > tcp_sck ) ;
sckxtn - > dnc = dnc ;
2020-02-21 09:42:10 +00:00
2020-02-28 07:41:25 +00:00
MIO_MEMSET ( & cinfo , 0 , MIO_SIZEOF ( cinfo ) ) ;
cinfo . remoteaddr = reqmsgxtn - > servaddr ;
cinfo . connect_tmout = reqmsgxtn - > rtmout ; /* TOOD: create a separate connect timeout or treate rtmout as a whole transaction time and calculate the remaining time from the transaction start, and use it */
2020-02-21 09:42:10 +00:00
2020-02-28 07:41:25 +00:00
if ( mio_dev_sck_connect ( dnc - > tcp_sck , & cinfo ) < = - 1 )
{
mio_dev_sck_kill ( dnc - > tcp_sck ) ;
dnc - > tcp_sck = MIO_NULL ;
return - 1 ; /* the connect request hasn't been honored. */
}
2020-02-27 17:34:27 +00:00
}
2020-02-28 07:41:25 +00:00
2020-02-25 07:09:20 +00:00
/* switch the belonging device to the tcp socket since the connect request has been acknowledged. */
2020-02-27 08:40:39 +00:00
MIO_ASSERT ( mio , reqmsgxtn - > rtmridx = = MIO_TMRIDX_INVALID ) ; /* ensure no timer job scheduled at this moment */
2020-02-28 07:41:25 +00:00
reqmsgxtn - > dev = dnc - > tcp_sck ;
reqmsgxtn - > rtries = 0 ;
2020-03-01 17:43:42 +00:00
if ( ! reqmsgxtn - > pending & & mio_dns_msg_to_pkt ( reqmsg ) - > qr = = 0 ) chain_pending_dns_reqmsg ( dnc , reqmsg ) ;
2020-02-25 07:09:20 +00:00
2020-05-15 06:18:49 +00:00
MIO_DEBUG6 ( mio , " DNC - switched transport to tcp - msgid:%d %p %p %p %p %p \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( reqmsg ) - > id ) , reqmsg , reqmsgxtn , reqmsgxtn - > dev , dnc - > udp_sck , dnc - > tcp_sck ) ;
2020-02-28 07:41:25 +00:00
if ( MIO_DEV_SCK_GET_PROGRESS ( dnc - > tcp_sck ) & MIO_DEV_SCK_CONNECTED )
{
2020-03-01 17:43:42 +00:00
if ( write_dns_msg_over_tcp ( reqmsgxtn - > dev , reqmsg ) < = - 1 )
{
/* the caller must not use reqmsg from now because it's freed here */
if ( MIO_LIKELY ( reqmsgxtn - > on_done ) ) reqmsgxtn - > on_done ( dnc , reqmsg , mio_geterrnum ( mio ) , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , reqmsg ) ;
}
2020-02-28 07:41:25 +00:00
}
2020-02-25 07:09:20 +00:00
return 0 ;
2020-02-21 09:42:10 +00:00
}
/* ----------------------------------------------------------------------- */
2020-02-21 04:21:16 +00:00
static int on_udp_read ( mio_dev_sck_t * dev , const void * data , mio_iolen_t dlen , const mio_skad_t * srcaddr )
{
mio_t * mio = dev - > mio ;
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
mio_dns_pkt_t * pkt ;
mio_dns_msg_t * reqmsg ;
mio_uint16_t id ;
2020-02-27 17:34:27 +00:00
if ( MIO_UNLIKELY ( dlen < = - 1 ) )
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - dns read error ....%js \n " , mio_geterrmsg ( mio ) ) ; /* TODO: add source packet */
2020-02-27 17:34:27 +00:00
return 0 ;
}
2020-02-21 04:21:16 +00:00
if ( MIO_UNLIKELY ( dlen < MIO_SIZEOF ( * pkt ) ) )
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG0 ( mio , " DNC - dns packet too small from .... \n " ) ; /* TODO: add source packet */
2020-02-21 04:21:16 +00:00
return 0 ; /* drop */
}
pkt = ( mio_dns_pkt_t * ) data ;
if ( ! pkt - > qr )
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG0 ( mio , " DNC - dropping dns request received ... \n " ) ; /* TODO: add source info */
2020-02-21 04:21:16 +00:00
return 0 ; /* drop request */
}
id = mio_ntoh16 ( pkt - > id ) ;
/* if id doesn't match one of the pending requests sent, drop it */
/* TODO: improve performance of dns response matching*/
reqmsg = dnc - > pending_req ;
while ( reqmsg )
{
mio_dns_pkt_t * reqpkt = mio_dns_msg_to_pkt ( reqmsg ) ;
dnc_dns_msg_xtn_t * reqmsgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
2020-02-28 07:41:25 +00:00
if ( reqmsgxtn - > dev = = dev & & pkt - > id = = reqpkt - > id & & mio_equal_skads ( & reqmsgxtn - > servaddr , srcaddr , 0 ) )
2020-02-21 04:21:16 +00:00
{
2020-02-27 08:40:39 +00:00
if ( reqmsgxtn - > rtmridx ! = MIO_TMRIDX_INVALID )
{
/* unschedule a timer job if any */
mio_deltmrjob ( mio , reqmsgxtn - > rtmridx ) ;
MIO_ASSERT ( mio , reqmsgxtn - > rtmridx = = MIO_TMRIDX_INVALID ) ;
}
2020-02-21 09:42:10 +00:00
////////////////////////
2020-03-02 04:58:22 +00:00
// for simple testing without actual truncated dns response
//pkt->tc = 1;
2020-02-21 09:42:10 +00:00
////////////////////////
2020-02-25 07:09:20 +00:00
if ( MIO_UNLIKELY ( pkt - > tc ) )
2020-02-21 04:21:16 +00:00
{
/* TODO: add an option for this behavior */
2020-02-28 07:41:25 +00:00
if ( switch_reqmsg_transport_to_tcp ( dnc , reqmsg ) > = 0 ) return 0 ;
2020-02-25 07:09:20 +00:00
/* TODO: add an option to call an error callback with TRUNCATION error code instead of fallback to received UDP truncated message */
2020-02-21 04:21:16 +00:00
}
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - received dns response over udp for msgid:%d \n " , id ) ;
2020-02-27 17:34:27 +00:00
if ( MIO_LIKELY ( reqmsgxtn - > on_done ) ) reqmsgxtn - > on_done ( dnc , reqmsg , MIO_ENOERR , data , dlen ) ;
2020-02-21 04:21:16 +00:00
release_dns_msg ( dnc , reqmsg ) ;
return 0 ;
}
2020-02-27 17:34:27 +00:00
2020-02-21 04:21:16 +00:00
reqmsg = reqmsgxtn - > next ;
}
/* the response id didn't match the ID of pending requests - need to wait longer? */
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - unknown dns response over udp... msgid:%d \n " , id ) ; /* TODO: add source info */
2020-02-21 04:21:16 +00:00
return 0 ;
}
static void on_udp_reply_timeout ( mio_t * mio , const mio_ntime_t * now , mio_tmrjob_t * job )
{
mio_dns_msg_t * reqmsg = ( mio_dns_msg_t * ) job - > ctx ;
dnc_dns_msg_xtn_t * msgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
2020-02-28 07:41:25 +00:00
mio_dev_sck_t * dev = msgxtn - > dev ;
2020-02-21 04:21:16 +00:00
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
2020-02-27 17:34:27 +00:00
mio_errnum_t status = MIO_ETMOUT ;
2020-02-21 04:21:16 +00:00
MIO_ASSERT ( mio , msgxtn - > rtmridx = = MIO_TMRIDX_INVALID ) ;
2020-02-21 09:42:10 +00:00
MIO_ASSERT ( mio , dev = = dnc - > udp_sck ) ;
2020-02-21 04:21:16 +00:00
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - unable to receive dns response in time over udp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( reqmsg ) - > id ) ) ;
2020-02-21 04:21:16 +00:00
if ( msgxtn - > rtries < msgxtn - > rmaxtries )
{
mio_ntime_t * tmout ;
tmout = MIO_IS_POS_NTIME ( & msgxtn - > wtmout ) ? & msgxtn - > wtmout : MIO_NULL ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - sending dns question again over udp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( reqmsg ) - > id ) ) ;
2020-02-27 17:34:27 +00:00
if ( mio_dev_sck_timedwrite ( dev , mio_dns_msg_to_pkt ( reqmsg ) , reqmsg - > pktlen , tmout , reqmsg , & msgxtn - > servaddr ) > = 0 ) return ; /* resent */
2020-02-21 04:21:16 +00:00
2020-02-27 17:34:27 +00:00
/* retry failed */
status = mio_geterrnum ( mio ) ;
2020-02-21 04:21:16 +00:00
}
2020-02-27 17:34:27 +00:00
if ( MIO_LIKELY ( msgxtn - > on_done ) ) msgxtn - > on_done ( dnc , reqmsg , status , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , reqmsg ) ;
2020-02-21 04:21:16 +00:00
}
static int on_udp_write ( mio_dev_sck_t * dev , mio_iolen_t wrlen , void * wrctx , const mio_skad_t * dstaddr )
{
mio_t * mio = dev - > mio ;
mio_dns_msg_t * msg = ( mio_dns_msg_t * ) wrctx ;
dnc_dns_msg_xtn_t * msgxtn = dnc_dns_msg_getxtn ( msg ) ;
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
2020-02-27 17:34:27 +00:00
mio_errnum_t status ;
2020-02-21 04:21:16 +00:00
MIO_ASSERT ( mio , dev = = ( mio_dev_sck_t * ) msgxtn - > dev ) ;
if ( wrlen < = - 1 )
{
/* write has timed out or an error has occurred */
2020-02-27 17:34:27 +00:00
status = mio_geterrnum ( mio ) ;
goto finalize ;
2020-02-21 04:21:16 +00:00
}
else if ( mio_dns_msg_to_pkt ( msg ) - > qr = = 0 & & msgxtn - > rmaxtries > 0 )
{
/* question. schedule to wait for response */
mio_tmrjob_t tmrjob ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - sent dns question over udp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-21 04:21:16 +00:00
MIO_MEMSET ( & tmrjob , 0 , MIO_SIZEOF ( tmrjob ) ) ;
tmrjob . ctx = msg ;
mio_gettime ( mio , & tmrjob . when ) ;
MIO_ADD_NTIME ( & tmrjob . when , & tmrjob . when , & msgxtn - > rtmout ) ;
tmrjob . handler = on_udp_reply_timeout ;
tmrjob . idxptr = & msgxtn - > rtmridx ;
msgxtn - > rtmridx = mio_instmrjob ( mio , & tmrjob ) ;
if ( msgxtn - > rtmridx = = MIO_TMRIDX_INVALID )
{
/* call the callback to indicate this operation failure in the middle of transaction */
2020-02-27 17:34:27 +00:00
status = mio_geterrnum ( mio ) ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - unable to schedule udp timeout - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-27 17:34:27 +00:00
goto finalize ;
2020-02-21 04:21:16 +00:00
}
2020-05-15 06:18:49 +00:00
2020-02-27 17:34:27 +00:00
if ( msgxtn - > rtries = = 0 )
2020-02-21 04:21:16 +00:00
{
2020-02-27 17:34:27 +00:00
/* this is the first wait */
/* TODO: improve performance. hashing by id? */
/* chain it to the peing request list */
chain_pending_dns_reqmsg ( dnc , msg ) ;
2020-02-21 04:21:16 +00:00
}
2020-02-27 17:34:27 +00:00
msgxtn - > rtries + + ;
2020-02-21 04:21:16 +00:00
}
else
{
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - sent dns message over udp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-02-27 17:34:27 +00:00
/* sent an answer. however this may be a question if msgxtn->rmaxtries is 0. */
status = MIO_ENOERR ;
goto finalize ;
2020-02-21 04:21:16 +00:00
}
return 0 ;
2020-02-27 17:34:27 +00:00
finalize :
if ( MIO_LIKELY ( msgxtn - > on_done ) ) msgxtn - > on_done ( dnc , msg , status , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , msg ) ;
return 0 ;
2020-02-21 04:21:16 +00:00
}
static void on_udp_connect ( mio_dev_sck_t * dev )
{
}
static void on_udp_disconnect ( mio_dev_sck_t * dev )
{
2020-05-15 06:18:49 +00:00
/*mio_t* mio = dev->mio;*/
2020-02-28 07:41:25 +00:00
mio_svc_dnc_t * dnc = ( ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dev ) ) - > dnc ;
mio_dns_msg_t * reqmsg ;
reqmsg = dnc - > pending_req ;
while ( reqmsg )
{
dnc_dns_msg_xtn_t * reqmsgxtn = dnc_dns_msg_getxtn ( reqmsg ) ;
mio_dns_msg_t * nextreqmsg = reqmsgxtn - > next ;
if ( reqmsgxtn - > dev = = dev )
{
if ( MIO_LIKELY ( reqmsgxtn - > on_done ) ) reqmsgxtn - > on_done ( dnc , reqmsg , MIO_ENORSP , MIO_NULL , 0 ) ;
release_dns_msg ( dnc , reqmsg ) ;
}
reqmsg = nextreqmsg ;
}
2020-02-21 04:21:16 +00:00
}
2020-03-02 04:58:22 +00:00
mio_svc_dnc_t * mio_svc_dnc_start ( mio_t * mio , const mio_skad_t * serv_addr , const mio_skad_t * bind_addr , const mio_ntime_t * send_tmout , const mio_ntime_t * reply_tmout , mio_oow_t max_tries )
2020-02-21 04:21:16 +00:00
{
mio_svc_dnc_t * dnc = MIO_NULL ;
mio_dev_sck_make_t mkinfo ;
2020-02-29 16:56:36 +00:00
dnc_sck_xtn_t * sckxtn ;
2020-02-21 04:21:16 +00:00
dnc = ( mio_svc_dnc_t * ) mio_callocmem ( mio , MIO_SIZEOF ( * dnc ) ) ;
2020-05-05 15:12:08 +00:00
if ( MIO_UNLIKELY ( ! dnc ) ) goto oops ;
2020-02-21 04:21:16 +00:00
dnc - > mio = mio ;
2020-05-05 15:12:08 +00:00
dnc - > svc_stop = mio_svc_dnc_stop ;
2020-02-21 04:21:16 +00:00
dnc - > serv_addr = * serv_addr ;
dnc - > send_tmout = * send_tmout ;
dnc - > reply_tmout = * reply_tmout ;
2020-03-02 04:58:22 +00:00
dnc - > max_tries = max_tries ;
2020-02-21 04:21:16 +00:00
MIO_MEMSET ( & mkinfo , 0 , MIO_SIZEOF ( mkinfo ) ) ;
switch ( mio_skad_family ( serv_addr ) )
{
case MIO_AF_INET :
mkinfo . type = MIO_DEV_SCK_UDP4 ;
break ;
case MIO_AF_INET6 :
mkinfo . type = MIO_DEV_SCK_UDP6 ;
break ;
default :
mio_seterrnum ( mio , MIO_EINVAL ) ;
goto oops ;
}
mkinfo . on_write = on_udp_write ;
mkinfo . on_read = on_udp_read ;
mkinfo . on_connect = on_udp_connect ;
mkinfo . on_disconnect = on_udp_disconnect ;
2020-02-29 16:56:36 +00:00
dnc - > udp_sck = mio_dev_sck_make ( mio , MIO_SIZEOF ( * sckxtn ) , & mkinfo ) ;
2020-02-21 04:21:16 +00:00
if ( ! dnc - > udp_sck ) goto oops ;
2020-02-29 16:56:36 +00:00
sckxtn = ( dnc_sck_xtn_t * ) mio_dev_sck_getxtn ( dnc - > udp_sck ) ;
sckxtn - > dnc = dnc ;
2020-02-21 04:21:16 +00:00
if ( bind_addr ) /* TODO: get mio_dev_sck_bind_t? instead of bind_addr? */
{
mio_dev_sck_bind_t bi ;
MIO_MEMSET ( & bi , 0 , MIO_SIZEOF ( bi ) ) ;
bi . localaddr = * bind_addr ;
if ( mio_dev_sck_bind ( dnc - > udp_sck , & bi ) < = - 1 ) goto oops ;
}
2020-04-30 14:48:39 +00:00
MIO_SVCL_APPEND_SVC ( & mio - > actsvc , ( mio_svc_t * ) dnc ) ;
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - STARTED SERVICE %p \n " , dnc ) ;
2020-02-21 04:21:16 +00:00
return dnc ;
oops :
if ( dnc )
{
if ( dnc - > udp_sck ) mio_dev_sck_kill ( dnc - > udp_sck ) ;
mio_freemem ( mio , dnc ) ;
}
return MIO_NULL ;
}
void mio_svc_dnc_stop ( mio_svc_dnc_t * dnc )
{
mio_t * mio = dnc - > mio ;
2020-02-25 07:09:20 +00:00
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( mio , " DNC - STOPPING SERVICE %p \n " , dnc ) ;
2020-02-21 04:21:16 +00:00
if ( dnc - > udp_sck ) mio_dev_sck_kill ( dnc - > udp_sck ) ;
2020-02-25 07:09:20 +00:00
if ( dnc - > tcp_sck ) mio_dev_sck_kill ( dnc - > tcp_sck ) ;
2020-02-21 04:21:16 +00:00
while ( dnc - > pending_req ) release_dns_msg ( dnc , dnc - > pending_req ) ;
2020-04-30 14:48:39 +00:00
MIO_SVCL_UNLINK_SVC ( dnc ) ;
2020-02-21 04:21:16 +00:00
mio_freemem ( mio , dnc ) ;
}
2020-03-01 17:43:42 +00:00
2020-03-02 05:16:37 +00:00
static MIO_INLINE int send_dns_msg ( mio_svc_dnc_t * dnc , mio_dns_msg_t * msg , int send_flags )
2020-02-21 04:21:16 +00:00
{
2020-03-01 17:43:42 +00:00
dnc_dns_msg_xtn_t * msgxtn = dnc_dns_msg_getxtn ( msg ) ;
2020-02-21 04:21:16 +00:00
mio_ntime_t * tmout ;
2020-03-02 05:16:37 +00:00
if ( ( send_flags & MIO_SVC_DNC_SEND_FLAG_PREFER_TCP ) & & switch_reqmsg_transport_to_tcp ( dnc , msg ) > = 0 ) return 0 ;
2020-02-21 04:21:16 +00:00
2020-05-15 06:18:49 +00:00
MIO_DEBUG1 ( dnc - > mio , " DNC - sending dns message over udp - msgid:%d \n " , ( int ) mio_ntoh16 ( mio_dns_msg_to_pkt ( msg ) - > id ) ) ;
2020-03-01 17:43:42 +00:00
tmout = MIO_IS_POS_NTIME ( & msgxtn - > wtmout ) ? & msgxtn - > wtmout : MIO_NULL ;
/* TODO: optionally, override dnc->serv_addr and use the target address passed as a parameter */
return mio_dev_sck_timedwrite ( dnc - > udp_sck , mio_dns_msg_to_pkt ( msg ) , msg - > pktlen , tmout , msg , & msgxtn - > servaddr ) ;
}
2020-02-25 07:09:20 +00:00
2020-03-02 05:16:37 +00:00
mio_dns_msg_t * mio_svc_dnc_sendmsg ( mio_svc_dnc_t * dnc , mio_dns_bhdr_t * bdns , mio_dns_bqr_t * qr , mio_oow_t qr_count , mio_dns_brr_t * rr , mio_oow_t rr_count , mio_dns_bedns_t * edns , int send_flags , mio_svc_dnc_on_done_t on_done , mio_oow_t xtnsize )
2020-03-01 17:43:42 +00:00
{
/* send a request or a response */
mio_dns_msg_t * msg ;
2020-02-21 04:21:16 +00:00
2020-03-01 17:43:42 +00:00
msg = make_dns_msg ( dnc , bdns , qr , qr_count , rr , rr_count , edns , on_done , xtnsize ) ;
if ( ! msg ) return MIO_NULL ;
2020-03-02 05:16:37 +00:00
if ( send_dns_msg ( dnc , msg , send_flags ) < = - 1 )
2020-02-21 04:21:16 +00:00
{
release_dns_msg ( dnc , msg ) ;
return MIO_NULL ;
}
2020-03-01 17:43:42 +00:00
2020-02-21 04:21:16 +00:00
return msg ;
}
2020-03-02 05:16:37 +00:00
mio_dns_msg_t * mio_svc_dnc_sendreq ( mio_svc_dnc_t * dnc , mio_dns_bhdr_t * bdns , mio_dns_bqr_t * qr , mio_dns_bedns_t * edns , int send_flags , mio_svc_dnc_on_done_t on_done , mio_oow_t xtnsize )
2020-02-21 04:21:16 +00:00
{
/* send a request without resource records */
if ( bdns - > rcode ! = MIO_DNS_RCODE_NOERROR )
{
mio_seterrnum ( dnc - > mio , MIO_EINVAL ) ;
return MIO_NULL ;
}
2020-03-02 05:16:37 +00:00
return mio_svc_dnc_sendmsg ( dnc , bdns , qr , 1 , MIO_NULL , 0 , edns , send_flags , on_done , xtnsize ) ;
2020-02-21 04:21:16 +00:00
}
/* ----------------------------------------------------------------------- */
struct dnc_dns_msg_resolve_xtn_t
{
mio_dns_rrt_t qtype ;
int flags ;
2020-07-02 14:21:47 +00:00
mio_uint8_t client_cookie [ MIO_DNS_COOKIE_CLIENT_LEN ] ;
2020-02-21 04:21:16 +00:00
mio_svc_dnc_on_resolve_t on_resolve ;
} ;
typedef struct dnc_dns_msg_resolve_xtn_t dnc_dns_msg_resolve_xtn_t ;
# if defined(MIO_HAVE_INLINE)
static MIO_INLINE dnc_dns_msg_resolve_xtn_t * dnc_dns_msg_resolve_getxtn ( mio_dns_msg_t * msg ) { return ( ( dnc_dns_msg_resolve_xtn_t * ) ( ( mio_uint8_t * ) dnc_dns_msg_getxtn ( msg ) + MIO_SIZEOF ( dnc_dns_msg_xtn_t ) ) ) ; }
# else
# define dnc_dns_msg_resolve_getxtn(msg) ((dnc_dns_msg_resolve_xtn_t*)((mio_uint8_t*)dnc_dns_msg_getxtn(msg) + MIO_SIZEOF(dnc_dns_msg_xtn_t)))
# endif
static void on_dnc_resolve ( mio_svc_dnc_t * dnc , mio_dns_msg_t * reqmsg , mio_errnum_t status , const void * data , mio_oow_t dlen )
{
mio_t * mio = mio_svc_dnc_getmio ( dnc ) ;
mio_dns_pkt_info_t * pi = MIO_NULL ;
2020-07-02 14:21:47 +00:00
dnc_dns_msg_resolve_xtn_t * resolxtn = dnc_dns_msg_resolve_getxtn ( reqmsg ) ;
2020-02-21 04:21:16 +00:00
if ( data )
{
mio_uint32_t i ;
MIO_ASSERT ( mio , status = = MIO_ENOERR ) ;
pi = mio_dns_make_packet_info ( mio , data , dlen ) ;
if ( ! pi )
{
status = mio_geterrnum ( mio ) ;
2020-02-27 17:34:27 +00:00
goto no_data ;
2020-02-21 04:21:16 +00:00
}
2020-07-02 14:21:47 +00:00
if ( resolxtn - > flags & MIO_SVC_DNC_RESOLVE_FLAG_COOKIE )
{
/* ------------------------------------------------- */
if ( pi - > edns . cookie . client_len > 0 )
{
if ( MIO_MEMCMP ( resolxtn - > client_cookie , pi - > edns . cookie . data . client , pi - > edns . cookie . client_len ) = = 0 )
{
pi - > edns . cookie_verified = 1 ; /* UGLY to set data in mio_dns_pkt_info_t */
}
}
if ( pi - > edns . cookie . server_len > 0 )
{
MIO_MEMCPY ( dnc - > cookie . data . server , pi - > edns . cookie . data . server , pi - > edns . cookie . server_len ) ;
dnc - > cookie . server_len = pi - > edns . cookie . server_len ;
}
#if 0
if ( pi - > hdr . rcode = = MIO_DNS_RCODE_BADCOOKIE )
{
/* TODO: retry it */
#if 0
if ( mio_svc_dnc_resolve ( dnc , qname , resolxtn - > qtype , resolxtn - > flags , on_dnc_resolve , resolxtn - > xtnsize ) < = - 1 )
{
}
# endif
/*how to retry?*/
}
# endif
/* ------------------------------------------------- */
}
if ( ! ( resolxtn - > flags & MIO_SVC_DNC_RESOLVE_FLAG_BRIEF ) )
{
/* the full reply packet is requested. */
if ( resolxtn - > on_resolve ) resolxtn - > on_resolve ( dnc , reqmsg , status , pi , 0 ) ;
goto done ;
}
2020-07-01 09:40:34 +00:00
2020-02-21 04:21:16 +00:00
if ( pi - > hdr . rcode ! = MIO_DNS_RCODE_NOERROR )
{
status = MIO_EINVAL ;
2020-02-27 17:34:27 +00:00
goto no_data ;
2020-02-21 04:21:16 +00:00
}
2020-02-27 17:34:27 +00:00
if ( pi - > ancount < 0 ) goto no_data ;
2020-02-21 04:21:16 +00:00
/* in the brief mode, we inspect the answer section only */
2020-07-02 14:21:47 +00:00
if ( resolxtn - > qtype = = MIO_DNS_RRT_Q_ANY )
2020-02-21 04:21:16 +00:00
{
/* return A or AAAA for ANY in the brief mode */
for ( i = 0 ; i < pi - > ancount ; i + + )
{
if ( pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_A | | pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_AAAA )
{
match_found :
2020-07-02 14:21:47 +00:00
if ( resolxtn - > on_resolve ) resolxtn - > on_resolve ( dnc , reqmsg , status , & pi - > rr . an [ i ] , MIO_SIZEOF ( pi - > rr . an [ i ] ) ) ;
2020-02-21 04:21:16 +00:00
goto done ;
}
}
}
for ( i = 0 ; i < pi - > ancount ; i + + )
{
/* it is a bit time taking to retreive the query type from the packet
* bundled in reqmsg as it requires parsing of the packet . let me use
* the query type i stored in the extension space . */
2020-07-02 14:21:47 +00:00
switch ( resolxtn - > qtype )
2020-02-21 04:21:16 +00:00
{
case MIO_DNS_RRT_Q_ANY :
case MIO_DNS_RRT_Q_AFXR : /* AFXR doesn't make sense in the brief mode. just treat it like ANY */
/* no A or AAAA found. so give the first entry in the answer */
goto match_found ;
case MIO_DNS_RRT_Q_MAILA :
/* if you want to get the full RRs, don't use the brief mode. */
if ( pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_MD | | pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_MF ) goto match_found ;
break ;
case MIO_DNS_RRT_Q_MAILB :
/* if you want to get the full RRs, don't use the brief mode. */
if ( pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_MB | | pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_MG | |
pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_MR | | pi - > rr . an [ i ] . rrtype = = MIO_DNS_RRT_MINFO ) goto match_found ;
break ;
default :
2020-07-02 14:21:47 +00:00
if ( pi - > rr . an [ i ] . rrtype = = resolxtn - > qtype ) goto match_found ;
2020-02-21 04:21:16 +00:00
break ;
}
}
2020-02-27 17:34:27 +00:00
goto no_data ;
2020-02-21 04:21:16 +00:00
}
else
{
2020-02-27 17:34:27 +00:00
no_data :
2020-07-02 14:21:47 +00:00
if ( resolxtn - > on_resolve ) resolxtn - > on_resolve ( dnc , reqmsg , status , MIO_NULL , 0 ) ;
2020-02-21 04:21:16 +00:00
}
done :
if ( pi ) mio_dns_free_packet_info ( mio_svc_dnc_getmio ( dnc ) , pi ) ;
}
2020-03-02 05:16:37 +00:00
mio_dns_msg_t * mio_svc_dnc_resolve ( mio_svc_dnc_t * dnc , const mio_bch_t * qname , mio_dns_rrt_t qtype , int resolve_flags , mio_svc_dnc_on_resolve_t on_resolve , mio_oow_t xtnsize )
2020-02-21 04:21:16 +00:00
{
static mio_dns_bhdr_t qhdr =
{
- 1 , /* id */
0 , /* qr */
MIO_DNS_OPCODE_QUERY , /* opcode */
0 , /* aa */
0 , /* tc */
1 , /* rd */
0 , /* ra */
0 , /* ad */
0 , /* cd */
MIO_DNS_RCODE_NOERROR /* rcode */
} ;
2020-07-01 15:54:23 +00:00
mio_dns_bedns_t qedns =
2020-02-21 04:21:16 +00:00
{
4096 , /* uplen */
0 , /* edns version */
0 , /* dnssec ok */
0 , /* number of edns options */
MIO_NULL
} ;
2020-07-01 15:54:23 +00:00
mio_dns_beopt_t beopt_cookie ;
2020-02-21 04:21:16 +00:00
mio_dns_bqr_t qr ;
mio_dns_msg_t * reqmsg ;
2020-03-01 17:43:42 +00:00
dnc_dns_msg_resolve_xtn_t * resolxtn ;
2020-02-21 04:21:16 +00:00
qr . qname = ( mio_bch_t * ) qname ;
qr . qtype = qtype ;
qr . qclass = MIO_DNS_RRC_IN ;
2020-07-01 15:54:23 +00:00
if ( resolve_flags & MIO_SVC_DNC_RESOLVE_FLAG_COOKIE )
{
beopt_cookie . code = MIO_DNS_EOPT_COOKIE ;
2020-07-02 14:21:47 +00:00
beopt_cookie . dlen = MIO_DNS_COOKIE_CLIENT_LEN ;
if ( dnc - > cookie . server_len > 0 ) beopt_cookie . dlen + = dnc - > cookie . server_len ;
beopt_cookie . dptr = & dnc - > cookie . data ;
2020-07-01 15:54:23 +00:00
qedns . beonum = 1 ;
qedns . beoptr = & beopt_cookie ;
}
2020-03-01 17:43:42 +00:00
reqmsg = make_dns_msg ( dnc , & qhdr , & qr , 1 , MIO_NULL , 0 , & qedns , on_dnc_resolve , MIO_SIZEOF ( * resolxtn ) + xtnsize ) ;
2020-02-21 04:21:16 +00:00
if ( reqmsg )
{
2020-03-02 05:16:37 +00:00
int send_flags ;
2020-07-02 14:21:47 +00:00
#if 0
if ( ( resolve_flags & MIO_SVC_DNC_RESOLVE_FLAG_COOKIE ) & & dnc - > cookie . server_len = = 0 )
2020-07-01 15:54:23 +00:00
{
2020-07-02 14:21:47 +00:00
/* Exclude the server cookie from the packet when the server cookie is not available.
*
* ASSUMPTIONS :
2020-07-01 15:54:23 +00:00
* the eopt entries are at the back of the packet .
* only 1 eopt entry ( MIO_DNS_EOPT_COOKIE ) has been added .
*
* manipulate the data length of the EDNS0 RR and COOKIE option
* as if the server cookie data has not been added .
*/
mio_dns_rrtr_t * edns_rrtr ;
mio_dns_eopt_t * eopt ;
edns_rrtr = ( mio_dns_rrtr_t * ) ( ( mio_uint8_t * ) mio_dns_msg_to_pkt ( reqmsg ) + reqmsg - > ednsrrtroff ) ;
2020-07-02 14:21:47 +00:00
reqmsg - > pktlen - = MIO_DNS_COOKIE_SERVER_MAX_LEN ;
2020-07-01 15:54:23 +00:00
MIO_ASSERT ( dnc - > mio , edns_rrtr - > rrtype = = MIO_CONST_HTON16 ( MIO_DNS_RRT_OPT ) ) ;
2020-07-02 14:21:47 +00:00
MIO_ASSERT ( dnc - > mio , edns_rrtr - > dlen = = MIO_CONST_HTON16 ( MIO_SIZEOF ( mio_dns_eopt_t ) + MIO_DNS_COOKIE_MAX_LEN ) ) ;
edns_rrtr - > dlen = MIO_CONST_HTON16 ( MIO_SIZEOF ( mio_dns_eopt_t ) + MIO_DNS_COOKIE_CLIENT_LEN ) ;
2020-07-01 15:54:23 +00:00
eopt = ( mio_dns_eopt_t * ) ( edns_rrtr + 1 ) ;
2020-07-02 14:21:47 +00:00
MIO_ASSERT ( dnc - > mio , eopt - > dlen = = MIO_CONST_HTON16 ( MIO_DNS_COOKIE_MAX_LEN ) ) ;
eopt - > dlen = MIO_CONST_HTON16 ( MIO_DNS_COOKIE_CLIENT_LEN ) ;
2020-07-01 15:54:23 +00:00
}
2020-07-02 14:21:47 +00:00
# endif
2020-07-01 15:54:23 +00:00
2020-03-01 17:43:42 +00:00
resolxtn = dnc_dns_msg_resolve_getxtn ( reqmsg ) ;
resolxtn - > on_resolve = on_resolve ;
resolxtn - > qtype = qtype ;
2020-03-02 05:16:37 +00:00
resolxtn - > flags = resolve_flags ;
2020-07-02 14:21:47 +00:00
/* store in the extension area the client cookie set in the packet */
MIO_MEMCPY ( resolxtn - > client_cookie , dnc - > cookie . data . client , MIO_DNS_COOKIE_CLIENT_LEN ) ;
2020-03-01 17:43:42 +00:00
2020-03-02 05:16:37 +00:00
send_flags = ( resolve_flags & MIO_SVC_DNC_SEND_FLAG_ALL ) ;
if ( MIO_UNLIKELY ( qtype = = MIO_DNS_RRT_Q_AFXR ) ) send_flags | = MIO_SVC_DNC_SEND_FLAG_PREFER_TCP ;
if ( send_dns_msg ( dnc , reqmsg , send_flags ) < = - 1 )
2020-03-01 17:43:42 +00:00
{
release_dns_msg ( dnc , reqmsg ) ;
return MIO_NULL ;
}
2020-02-21 04:21:16 +00:00
}
return reqmsg ;
}
2020-03-03 08:49:24 +00:00
/* TODO: upon startup, read /etc/hosts. setup inotify or find a way to detect file changes..
* in resolve , add an option to use entries from / etc / hosts */
/* TODO: afxr client ... */
/* TODO: trace function to do its own recursive resolution?...
mio_dns_msg_t * mio_svc_dnc_trace ( mio_svc_dnc_t * dnc , const mio_bch_t * qname , mio_dns_rrt_t qtype , int resolve_flags , mio_svc_dnc_on_resolve_t on_resolve , mio_oow_t xtnsize )
{
}
*/