libaddns/dnssock.c

ソースコードを見る。

関数
static int	destroy_dns_connection (struct dns_connection *conn)
static DNS_ERROR	dns_tcp_open (const char *nameserver, TALLOC_CTX *mem_ctx, struct dns_connection **result)
static DNS_ERROR	dns_udp_open (const char *nameserver, TALLOC_CTX *mem_ctx, struct dns_connection **result)
DNS_ERROR	dns_open_connection (const char *nameserver, int32 dwType, TALLOC_CTX *mem_ctx, struct dns_connection **conn)
static DNS_ERROR	write_all (int fd, uint8 *data, size_t len)
static DNS_ERROR	dns_send_tcp (struct dns_connection *conn, const struct dns_buffer *buf)
static DNS_ERROR	dns_send_udp (struct dns_connection *conn, const struct dns_buffer *buf)
DNS_ERROR	dns_send (struct dns_connection *conn, const struct dns_buffer *buf)
static DNS_ERROR	read_all (int fd, uint8 *data, size_t len)
static DNS_ERROR	dns_receive_tcp (TALLOC_CTX *mem_ctx, struct dns_connection *conn, struct dns_buffer **presult)
static DNS_ERROR	dns_receive_udp (TALLOC_CTX *mem_ctx, struct dns_connection *conn, struct dns_buffer **presult)
DNS_ERROR	dns_receive (TALLOC_CTX *mem_ctx, struct dns_connection *conn, struct dns_buffer **presult)
DNS_ERROR	dns_transaction (TALLOC_CTX *mem_ctx, struct dns_connection *conn, const struct dns_request *req, struct dns_request **resp)
DNS_ERROR	dns_update_transaction (TALLOC_CTX *mem_ctx, struct dns_connection *conn, struct dns_update_request *up_req, struct dns_update_request **up_resp)

---

関数

static int destroy_dns_connection

(

struct dns_connection *

conn

)

[static]

dnssock.c の 31 行で定義されています。

参照先 dns_connection::s.

参照元 dns_tcp_open()・dns_udp_open().

{
        return close(conn->s);
}

static DNS_ERROR dns_tcp_open	(	const char *	nameserver,
		TALLOC_CTX *	mem_ctx,
		struct dns_connection **	result
	)			[static]

dnssock.c の 39 行で定義されています。

参照先 destroy_dns_connection()・dns_connection::hType・result・dns_connection::s.

参照元 dns_open_connection().

{
        uint32_t ulAddress;
        struct hostent *pHost;
        struct sockaddr_in s_in;
        struct dns_connection *conn;
        int res;

        if (!(conn = talloc(mem_ctx, struct dns_connection))) {
                return ERROR_DNS_NO_MEMORY;
        }

        if ( (ulAddress = inet_addr( nameserver )) == INADDR_NONE ) {
                if ( (pHost = gethostbyname( nameserver )) == NULL ) {
                        TALLOC_FREE(conn);
                        return ERROR_DNS_INVALID_NAME_SERVER;
                }
                memcpy( &ulAddress, pHost->h_addr, pHost->h_length );
        }

        conn->s = socket( PF_INET, SOCK_STREAM, 0 );
        if (conn->s == -1) {
                TALLOC_FREE(conn);
                return ERROR_DNS_CONNECTION_FAILED;
        }

        talloc_set_destructor(conn, destroy_dns_connection);

        s_in.sin_family = AF_INET;
        s_in.sin_addr.s_addr = ulAddress;
        s_in.sin_port = htons( DNS_TCP_PORT );

        res = connect(conn->s, (struct sockaddr*)&s_in, sizeof( s_in ));
        if (res == -1) {
                TALLOC_FREE(conn);
                return ERROR_DNS_CONNECTION_FAILED;
        }

        conn->hType = DNS_TCP;

        *result = conn;
        return ERROR_DNS_SUCCESS;
}

static DNS_ERROR dns_udp_open	(	const char *	nameserver,
		TALLOC_CTX *	mem_ctx,
		struct dns_connection **	result
	)			[static]

dnssock.c の 88 行で定義されています。

参照先 destroy_dns_connection()・dns_connection::hType・dns_connection::RecvAddr・result・dns_connection::s.

参照元 dns_open_connection().

{
        unsigned long ulAddress;
        struct hostent *pHost;
        struct sockaddr_in RecvAddr;
        struct dns_connection *conn;

        if (!(conn = talloc(NULL, struct dns_connection))) {
                return ERROR_DNS_NO_MEMORY;
        }

        if ( (ulAddress = inet_addr( nameserver )) == INADDR_NONE ) {
                if ( (pHost = gethostbyname( nameserver )) == NULL ) {
                        TALLOC_FREE(conn);
                        return ERROR_DNS_INVALID_NAME_SERVER;
                }
                memcpy( &ulAddress, pHost->h_addr, pHost->h_length );
        }
        
        /* Create a socket for sending data */

        conn->s = socket( AF_INET, SOCK_DGRAM, IPPROTO_UDP );
        if (conn->s == -1) {
                TALLOC_FREE(conn);
                return ERROR_DNS_CONNECTION_FAILED;
        }

        talloc_set_destructor(conn, destroy_dns_connection);

        /* Set up the RecvAddr structure with the IP address of
           the receiver (in this example case "123.456.789.1")
           and the specified port number. */

        RecvAddr.sin_family = AF_INET;
        RecvAddr.sin_port = htons( DNS_UDP_PORT );
        RecvAddr.sin_addr.s_addr = ulAddress;

        conn->hType = DNS_UDP;
        memcpy( &conn->RecvAddr, &RecvAddr, sizeof( struct sockaddr_in ) );

        *result = conn;
        return ERROR_DNS_SUCCESS;
}

DNS_ERROR dns_open_connection	(	const char *	nameserver,
		int32	dwType,
		TALLOC_CTX *	mem_ctx,
		struct dns_connection **	conn
	)

dnssock.c の 137 行で定義されています。

参照先 dns_tcp_open()・dns_udp_open().

参照元 dns_negotiate_sec_ctx()・do_gethostbyname()・DoDNSUpdate().

{
        switch ( dwType ) {
        case DNS_TCP:
                return dns_tcp_open( nameserver, mem_ctx, conn );
        case DNS_UDP:
                return dns_udp_open( nameserver, mem_ctx, conn );
        }
        
        return ERROR_DNS_INVALID_PARAMETER;
}

static DNS_ERROR write_all	(	int	fd,
		uint8 *	data,
		size_t	len
	)			[static]

dnssock.c の 151 行で定義されています。

参照先 total.

参照元 dns_send_tcp().

{
        size_t total = 0;

        while (total < len) {

                ssize_t ret = write(fd, data + total, len - total);

                if (ret <= 0) {
                        /*
                         * EOF or error
                         */
                        return ERROR_DNS_SOCKET_ERROR;
                }

                total += ret;
        }

        return ERROR_DNS_SUCCESS;
}

static DNS_ERROR dns_send_tcp	(	struct dns_connection *	conn,
		const struct dns_buffer *	buf
	)			[static]

dnssock.c の 172 行で定義されています。

参照先 buf・err・len・dns_connection::s・write_all().

参照元 dns_send().

{
        uint16 len = htons(buf->offset);
        DNS_ERROR err;

        err = write_all(conn->s, (uint8 *)&len, sizeof(len));
        if (!ERR_DNS_IS_OK(err)) return err;

        return write_all(conn->s, buf->data, buf->offset);
}

static DNS_ERROR dns_send_udp	(	struct dns_connection *	conn,
		const struct dns_buffer *	buf
	)			[static]

dnssock.c の 184 行で定義されています。

参照先 buf・dns_connection::RecvAddr・dns_connection::s.

参照元 dns_send().

{
        ssize_t ret;

        ret = sendto(conn->s, buf->data, buf->offset, 0,
                     (struct sockaddr *)&conn->RecvAddr,
                     sizeof(conn->RecvAddr));

        if (ret != buf->offset) {
                return ERROR_DNS_SOCKET_ERROR;
        }

        return ERROR_DNS_SUCCESS;
}

DNS_ERROR dns_send	(	struct dns_connection *	conn,
		const struct dns_buffer *	buf
	)

dnssock.c の 200 行で定義されています。

参照先 buf・dns_send_tcp()・dns_send_udp()・dns_connection::hType.

参照元 dns_negotiate_gss_ctx_int()・dns_transaction().

{
        if (conn->hType == DNS_TCP) {
                return dns_send_tcp(conn, buf);
        }

        if (conn->hType == DNS_UDP) {
                return dns_send_udp(conn, buf);
        }

        return ERROR_DNS_INVALID_PARAMETER;
}

static DNS_ERROR read_all	(	int	fd,
		uint8 *	data,
		size_t	len
	)			[static]

dnssock.c の 213 行で定義されています。

参照先 total.

参照元 dns_receive_tcp().

{
        size_t total = 0;
        fd_set rfds;
        struct timeval tv;

        while (total < len) {
                ssize_t ret;
                int fd_ready;
                
                FD_ZERO( &rfds );
                FD_SET( fd, &rfds );

                /* 10 second timeout */
                tv.tv_sec = 10;
                tv.tv_usec = 0;
                
                fd_ready = select( fd+1, &rfds, NULL, NULL, &tv );
                if ( fd_ready == 0 ) {
                        /* read timeout */
                        return ERROR_DNS_SOCKET_ERROR;
                }

                ret = read(fd, data + total, len - total);
                if (ret <= 0) {
                        /* EOF or error */
                        return ERROR_DNS_SOCKET_ERROR;
                }

                total += ret;
        }

        return ERROR_DNS_SUCCESS;
}

static DNS_ERROR dns_receive_tcp	(	TALLOC_CTX *	mem_ctx,
		struct dns_connection *	conn,
		struct dns_buffer **	presult
	)			[static]

dnssock.c の 248 行で定義されています。

参照先 buf・err・len・read_all()・dns_connection::s.

参照元 dns_receive().

{
        struct dns_buffer *buf;
        DNS_ERROR err;
        uint16 len;

        if (!(buf = TALLOC_ZERO_P(mem_ctx, struct dns_buffer))) {
                return ERROR_DNS_NO_MEMORY;
        }

        err = read_all(conn->s, (uint8 *)&len, sizeof(len));
        if (!ERR_DNS_IS_OK(err)) {
                return err;
        }

        buf->size = ntohs(len);

        if (buf->size) {
                if (!(buf->data = TALLOC_ARRAY(buf, uint8, buf->size))) {
                        TALLOC_FREE(buf);
                        return ERROR_DNS_NO_MEMORY;
                }
        } else {
                buf->data = NULL;
        }

        err = read_all(conn->s, buf->data, buf->size);
        if (!ERR_DNS_IS_OK(err)) {
                TALLOC_FREE(buf);
                return err;
        }

        *presult = buf;
        return ERROR_DNS_SUCCESS;
}

static DNS_ERROR dns_receive_udp	(	TALLOC_CTX *	mem_ctx,
		struct dns_connection *	conn,
		struct dns_buffer **	presult
	)			[static]

dnssock.c の 286 行で定義されています。

参照先 buf・dns_connection::s.

参照元 dns_receive().

{
        struct dns_buffer *buf;
        ssize_t received;

        if (!(buf = TALLOC_ZERO_P(mem_ctx, struct dns_buffer))) {
                return ERROR_DNS_NO_MEMORY;
        }

        /*
         * UDP based DNS can only be 512 bytes
         */

        if (!(buf->data = TALLOC_ARRAY(buf, uint8, 512))) {
                TALLOC_FREE(buf);
                return ERROR_DNS_NO_MEMORY;
        }

        received = recv(conn->s, (void *)buf->data, 512, 0);

        if (received == -1) {
                TALLOC_FREE(buf);
                return ERROR_DNS_SOCKET_ERROR;
        }

        if (received > 512) {
                TALLOC_FREE(buf);
                return ERROR_DNS_BAD_RESPONSE;
        }

        buf->size = received;
        buf->offset = 0;

        *presult = buf;
        return ERROR_DNS_SUCCESS;
}

DNS_ERROR dns_receive	(	TALLOC_CTX *	mem_ctx,
		struct dns_connection *	conn,
		struct dns_buffer **	presult
	)

dnssock.c の 325 行で定義されています。

参照先 dns_receive_tcp()・dns_receive_udp()・dns_connection::hType.

参照元 dns_negotiate_gss_ctx_int()・dns_transaction().

{
        if (conn->hType == DNS_TCP) {
                return dns_receive_tcp(mem_ctx, conn, presult);
        }

        if (conn->hType == DNS_UDP) {
                return dns_receive_udp(mem_ctx, conn, presult);
        }

        return ERROR_DNS_INVALID_PARAMETER;
}

DNS_ERROR dns_transaction	(	TALLOC_CTX *	mem_ctx,
		struct dns_connection *	conn,
		const struct dns_request *	req,
		struct dns_request **	resp
	)

dnssock.c の 339 行で定義されています。

参照先 buf・dns_marshall_request()・dns_receive()・dns_send()・dns_unmarshall_request()・err・error.

参照元 dns_update_transaction()・do_gethostbyname().

{
        struct dns_buffer *buf = NULL;
        DNS_ERROR err;

        err = dns_marshall_request(conn, req, &buf);
        if (!ERR_DNS_IS_OK(err)) goto error;

        err = dns_send(conn, buf);
        if (!ERR_DNS_IS_OK(err)) goto error;
        TALLOC_FREE(buf);

        err = dns_receive(mem_ctx, conn, &buf);
        if (!ERR_DNS_IS_OK(err)) goto error;

        err = dns_unmarshall_request(mem_ctx, buf, resp);

 error:
        TALLOC_FREE(buf);
        return err;
}

DNS_ERROR dns_update_transaction	(	TALLOC_CTX *	mem_ctx,
		struct dns_connection *	conn,
		struct dns_update_request *	up_req,
		struct dns_update_request **	up_resp
	)

dnssock.c の 363 行で定義されています。

参照先 dns_request2update()・dns_transaction()・dns_update2request()・err.

参照元 DoDNSUpdate().

{
        struct dns_request *resp;
        DNS_ERROR err;

        err = dns_transaction(mem_ctx, conn, dns_update2request(up_req),
                              &resp);

        if (!ERR_DNS_IS_OK(err)) return err;

        *up_resp = dns_request2update(resp);
        return ERROR_DNS_SUCCESS;
}

---