2005-01-19 06:20:20 +03:00
/*
Unix SMB / CIFS implementation .
local testing of socket routines .
Copyright ( C ) Andrew Tridgell 2005
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 2 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 , write to the Free Software
Foundation , Inc . , 675 Mass Ave , Cambridge , MA 0213 9 , USA .
*/
# include "includes.h"
2005-02-10 09:59:29 +03:00
# include "lib/socket/socket.h"
2005-06-17 03:19:35 +04:00
# include "lib/events/events.h"
2006-03-07 14:07:23 +03:00
# include "system/network.h"
# include "netif/netif.h"
2006-03-25 19:23:42 +03:00
# include "torture/torture.h"
2005-01-19 06:20:20 +03:00
/*
basic testing of udp routines
*/
2006-06-17 02:48:50 +04:00
static BOOL test_udp ( struct torture_context * test , const void * data )
2005-01-19 06:20:20 +03:00
{
struct socket_context * sock1 , * sock2 ;
NTSTATUS status ;
2006-01-10 01:12:53 +03:00
struct socket_address * srv_addr , * from_addr , * localhost ;
2005-01-19 06:20:20 +03:00
size_t size = 100 + ( random ( ) % 100 ) ;
DATA_BLOB blob , blob2 ;
size_t sent , nread ;
BOOL ret = True ;
status = socket_create ( " ip " , SOCKET_TYPE_DGRAM , & sock1 , 0 ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " creating DGRAM IP socket 1 " ) ;
2006-06-17 02:48:50 +04:00
talloc_steal ( test , sock1 ) ;
2005-01-19 06:20:20 +03:00
status = socket_create ( " ip " , SOCKET_TYPE_DGRAM , & sock2 , 0 ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " creating DGRAM IP socket 1 " ) ;
2006-06-17 02:48:50 +04:00
talloc_steal ( test , sock2 ) ;
2005-01-19 06:20:20 +03:00
2006-01-10 01:12:53 +03:00
localhost = socket_address_from_strings ( sock1 , sock1 - > backend_name ,
iface_best_ip ( " 127.0.0.1 " ) , 0 ) ;
2006-06-17 02:48:50 +04:00
torture_assert ( test , localhost , " Localhost not found " ) ;
2006-01-10 01:12:53 +03:00
status = socket_listen ( sock1 , localhost , 0 , 0 ) ;
2006-06-26 23:23:21 +04:00
torture_assert_ntstatus_ok ( test , status , " listen on socket 1 " ) ;
2005-01-19 06:20:20 +03:00
2006-06-17 02:48:50 +04:00
srv_addr = socket_get_my_addr ( sock1 , test ) ;
2006-01-10 01:12:53 +03:00
if ( srv_addr = = NULL | | strcmp ( srv_addr - > addr , iface_best_ip ( " 127.0.0.1 " ) ) ! = 0 ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Expected server address of %s but got %s " ,
2006-01-10 01:12:53 +03:00
iface_best_ip ( " 127.0.0.1 " ) , srv_addr ? srv_addr - > addr : NULL ) ;
2005-01-19 06:20:20 +03:00
return False ;
}
2006-06-17 02:48:50 +04:00
torture_comment ( test , " server port is %d " , srv_addr - > port ) ;
2005-01-19 06:20:20 +03:00
2006-06-17 02:48:50 +04:00
blob = data_blob_talloc ( test , NULL , size ) ;
blob2 = data_blob_talloc ( test , NULL , size ) ;
2005-01-19 06:20:20 +03:00
generate_random_buffer ( blob . data , blob . length ) ;
sent = size ;
2006-04-30 09:58:31 +04:00
status = socket_sendto ( sock2 , & blob , & sent , srv_addr ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " sendto() on socket 2 " ) ;
2005-01-19 06:20:20 +03:00
2006-04-30 09:58:31 +04:00
status = socket_recvfrom ( sock1 , blob2 . data , size , & nread ,
2006-01-10 01:12:53 +03:00
sock1 , & from_addr ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " recvfrom() on socket 1 " ) ;
2005-01-19 06:20:20 +03:00
2006-01-10 01:12:53 +03:00
if ( strcmp ( from_addr - > addr , srv_addr - > addr ) ! = 0 ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Unexpected recvfrom addr %s " , from_addr - > addr ) ;
return False ;
2005-01-19 06:20:20 +03:00
}
if ( nread ! = size ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Unexpected recvfrom size %d should be %d \n " ,
( int ) nread , ( int ) size ) ;
return False ;
2005-01-19 06:20:20 +03:00
}
2006-06-17 02:48:50 +04:00
torture_assert ( test , memcmp ( blob2 . data , blob . data , size ) = = 0 ,
" Bad data in recvfrom " ) ;
2005-01-19 06:20:20 +03:00
generate_random_buffer ( blob . data , blob . length ) ;
2006-04-30 09:58:31 +04:00
status = socket_sendto ( sock1 , & blob , & sent , from_addr ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " sendto() on socket 1 " ) ;
2005-01-19 06:20:20 +03:00
2006-04-30 09:58:31 +04:00
status = socket_recvfrom ( sock2 , blob2 . data , size , & nread ,
2006-01-10 01:12:53 +03:00
sock2 , & from_addr ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " recvfrom() on socket 2 " ) ;
2006-01-10 01:12:53 +03:00
if ( strcmp ( from_addr - > addr , srv_addr - > addr ) ! = 0 ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Unexpected recvfrom addr %s \n " , from_addr - > addr ) ;
return False ;
2005-01-19 06:20:20 +03:00
}
2006-06-17 02:48:50 +04:00
2005-01-19 06:20:20 +03:00
if ( nread ! = size ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Unexpected recvfrom size %d should be %d \n " ,
( int ) nread , ( int ) size ) ;
return False ;
2005-01-19 06:20:20 +03:00
}
2006-06-17 02:48:50 +04:00
2006-01-10 01:12:53 +03:00
if ( from_addr - > port ! = srv_addr - > port ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Unexpected recvfrom port %d should be %d \n " ,
2006-01-10 01:12:53 +03:00
from_addr - > port , srv_addr - > port ) ;
2006-06-17 02:48:50 +04:00
return False ;
2005-01-19 06:20:20 +03:00
}
2006-06-17 02:48:50 +04:00
torture_assert ( test , memcmp ( blob2 . data , blob . data , size ) = = 0 ,
" Bad data in recvfrom " ) ;
2005-01-19 06:20:20 +03:00
talloc_free ( sock1 ) ;
talloc_free ( sock2 ) ;
return ret ;
}
2005-01-19 06:33:49 +03:00
/*
basic testing of tcp routines
*/
2006-06-17 02:48:50 +04:00
static BOOL test_tcp ( struct torture_context * test , const void * data )
2005-01-19 06:33:49 +03:00
{
struct socket_context * sock1 , * sock2 , * sock3 ;
NTSTATUS status ;
2006-01-10 01:12:53 +03:00
struct socket_address * srv_addr , * from_addr , * localhost ;
2005-01-19 06:33:49 +03:00
size_t size = 100 + ( random ( ) % 100 ) ;
DATA_BLOB blob , blob2 ;
size_t sent , nread ;
2006-06-17 02:48:50 +04:00
struct event_context * ev = event_context_init ( test ) ;
2005-01-19 06:33:49 +03:00
status = socket_create ( " ip " , SOCKET_TYPE_STREAM , & sock1 , 0 ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " creating IP stream socket 1 " ) ;
2006-06-17 02:48:50 +04:00
talloc_steal ( test , sock1 ) ;
2005-01-19 06:33:49 +03:00
status = socket_create ( " ip " , SOCKET_TYPE_STREAM , & sock2 , 0 ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " creating IP stream socket 1 " ) ;
2006-06-17 02:48:50 +04:00
talloc_steal ( test , sock2 ) ;
2005-01-19 06:33:49 +03:00
2006-01-10 01:12:53 +03:00
localhost = socket_address_from_strings ( sock1 , sock1 - > backend_name ,
iface_best_ip ( " 127.0.0.1 " ) , 0 ) ;
2006-06-17 02:48:50 +04:00
torture_assert ( test , localhost , " Localhost not found " ) ;
2006-01-10 01:12:53 +03:00
status = socket_listen ( sock1 , localhost , 0 , 0 ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " listen on socket 1 " ) ;
2005-01-19 06:33:49 +03:00
2006-06-17 02:48:50 +04:00
srv_addr = socket_get_my_addr ( sock1 , test ) ;
torture_assert ( test , srv_addr & & srv_addr - > addr ,
" Unexpected socket_get_my_addr NULL \n " ) ;
2006-01-10 01:12:53 +03:00
if ( strcmp ( srv_addr - > addr , iface_best_ip ( " 127.0.0.1 " ) ) ! = 0 ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Expected server address of %s but got %s \n " ,
2006-01-10 01:12:53 +03:00
iface_best_ip ( " 127.0.0.1 " ) , srv_addr ? srv_addr - > addr : NULL ) ;
2005-01-19 06:33:49 +03:00
return False ;
}
2006-06-17 02:48:50 +04:00
torture_comment ( test , " server port is %d " , srv_addr - > port ) ;
2005-01-19 06:33:49 +03:00
2006-01-10 01:12:53 +03:00
status = socket_connect_ev ( sock2 , NULL , srv_addr , 0 , ev ) ;
2006-06-26 23:23:21 +04:00
torture_assert_ntstatus_ok ( test , status , " connect() on socket 2 " ) ;
2005-01-19 06:33:49 +03:00
status = socket_accept ( sock1 , & sock3 ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " accept() on socket 1 " ) ;
2006-06-17 02:48:50 +04:00
talloc_steal ( test , sock3 ) ;
2005-01-19 06:33:49 +03:00
talloc_free ( sock1 ) ;
2006-06-17 02:48:50 +04:00
blob = data_blob_talloc ( test , NULL , size ) ;
blob2 = data_blob_talloc ( test , NULL , size ) ;
2005-01-19 06:33:49 +03:00
generate_random_buffer ( blob . data , blob . length ) ;
sent = size ;
2006-04-30 09:58:31 +04:00
status = socket_send ( sock2 , & blob , & sent ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " send() on socket 2 " ) ;
2005-01-19 06:33:49 +03:00
2006-04-30 09:58:31 +04:00
status = socket_recv ( sock3 , blob2 . data , size , & nread ) ;
2006-06-24 01:05:45 +04:00
torture_assert_ntstatus_ok ( test , status , " recv() on socket 3 " ) ;
2005-01-19 06:33:49 +03:00
2006-06-17 02:48:50 +04:00
from_addr = socket_get_peer_addr ( sock3 , test ) ;
torture_assert ( test , from_addr & & from_addr - > addr ,
" Unexpected recvfrom addr NULL " ) ;
2005-01-19 06:33:49 +03:00
2006-01-10 01:12:53 +03:00
if ( strcmp ( from_addr - > addr , srv_addr - > addr ) ! = 0 ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Unexpected recvfrom addr %s \n " ,
from_addr ? from_addr - > addr : NULL ) ;
return False ;
2005-01-19 06:33:49 +03:00
}
if ( nread ! = size ) {
2006-06-17 02:48:50 +04:00
torture_fail ( test , " Unexpected recvfrom size %d should be %d \n " ,
( int ) nread , ( int ) size ) ;
return False ;
2005-01-19 06:33:49 +03:00
}
2006-06-17 02:48:50 +04:00
torture_assert ( test ,
memcmp ( blob2 . data , blob . data , size ) = = 0 ,
" Bad data in recv " ) ;
2005-01-19 06:33:49 +03:00
2006-06-17 02:48:50 +04:00
return True ;
2005-01-19 06:33:49 +03:00
}
2006-06-17 04:17:50 +04:00
struct torture_suite * torture_local_socket ( TALLOC_CTX * mem_ctx )
2005-01-19 06:20:20 +03:00
{
2006-06-17 04:17:50 +04:00
struct torture_suite * suite = torture_suite_create ( mem_ctx ,
2006-06-17 02:48:50 +04:00
" LOCAL-SOCKET " ) ;
2005-01-19 06:20:20 +03:00
2006-06-17 02:48:50 +04:00
torture_suite_add_simple_tcase ( suite , " udp " , test_udp , NULL ) ;
torture_suite_add_simple_tcase ( suite , " tcp " , test_tcp , NULL ) ;
2005-01-19 06:20:20 +03:00
2006-06-17 04:17:50 +04:00
return suite ;
2005-01-19 06:20:20 +03:00
}