+++ /dev/null
-/*
- * socket_tests.c
- * xnu_quick_test
- *
- * Created by Jerry Cottingham on 4/12/05.
- * Copyright 2005 Apple Computer Inc. All rights reserved.
- *
- */
-
-#include "tests.h"
-#include <poll.h>
-#include <mach/mach.h>
-
-extern char g_target_path[ PATH_MAX ];
-
-/* **************************************************************************************************************
- * Test accept, bind, connect, listen, socket, recvmsg, sendmsg, recvfrom, sendto, getpeername, getsockname
- * system calls.
- * WARNING - I don't do networking - this should get a good look from a networking stud.
- * **************************************************************************************************************
- */
-int socket_tests( void * the_argp )
-{
- int my_err, my_status, my_len;
- pid_t my_pid, my_wait_pid;
- int my_socket_fd = -1;
- int my_accepted_socket = -1;
- char * my_parent_pathp = NULL;
- char * my_child_pathp = NULL;
- socklen_t my_accept_len;
- struct sockaddr *my_sockaddr;
- ssize_t my_result;
- off_t my_current_offset;
- char my_parent_socket_name[sizeof(struct sockaddr) + 64];
- char my_child_socket_name[sizeof(struct sockaddr) + 64];
- char my_accept_buffer[sizeof(struct sockaddr) + 64];
- kern_return_t my_kr;
-
- /* generate 2 names for binding to the sockets (one socket in the parent and one in the child) */
- my_kr = vm_allocate((vm_map_t) mach_task_self(), (vm_address_t*)&my_parent_pathp, 128, VM_FLAGS_ANYWHERE);
- if(my_kr != KERN_SUCCESS){
- printf( "vm_allocate failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-
- my_kr = vm_allocate((vm_map_t) mach_task_self(), (vm_address_t*)&my_child_pathp, 128, VM_FLAGS_ANYWHERE);
- if(my_kr != KERN_SUCCESS){
- printf( "vm_allocate failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-
- *my_parent_pathp = 0x00;
- strcat( my_parent_pathp, "/tmp/" );
-
- /* get a unique name for our testing */
- my_err = create_random_name( my_parent_pathp, 0 );
- if ( my_err != 0 ) {
- goto test_failed_exit;
- }
- strcpy( my_child_pathp, my_parent_pathp );
- strcat( my_parent_pathp, "p" ); /* append 'p' to mean "parent" */
- strcat( my_child_pathp, "c" ); /* append 'c' to mean "child" */
-
- memset( &my_parent_socket_name[0], 0, sizeof(my_parent_socket_name) );
- memset( &my_child_socket_name[0], 0, sizeof(my_child_socket_name) );
-
- /* use unique names we generated in /tmp/ */
- my_sockaddr = (struct sockaddr *) &my_parent_socket_name[0];
- my_len = sizeof(*my_sockaddr) - sizeof(my_sockaddr->sa_data) + strlen(my_parent_pathp);
- my_sockaddr->sa_len = my_len;
- my_sockaddr->sa_family = AF_UNIX;
- strcpy( &my_sockaddr->sa_data[0], my_parent_pathp );
-
- my_sockaddr = (struct sockaddr *) &my_child_socket_name[0];
- my_len = sizeof(*my_sockaddr) - sizeof(my_sockaddr->sa_data) + strlen(my_child_pathp);
- my_sockaddr->sa_len = my_len;
- my_sockaddr->sa_family = AF_UNIX;
- strcpy( &my_sockaddr->sa_data[0], my_child_pathp );
-
- /* set up socket for parent side */
- my_socket_fd = socket( AF_UNIX, SOCK_STREAM, 0 );
- if ( my_socket_fd == -1 ) {
- printf( "socket call in parent failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
- my_sockaddr = (struct sockaddr *) &my_parent_socket_name[0];
- my_err = bind( my_socket_fd, my_sockaddr, my_sockaddr->sa_len );
- if ( my_err == -1 ) {
- printf( "bind call in child failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-
- /* test getsockname */
- my_sockaddr = (struct sockaddr *) &my_accept_buffer[0];
- my_accept_len = sizeof(my_accept_buffer);
- my_err = getsockname( my_socket_fd, my_sockaddr, &my_accept_len );
- if ( my_err == -1 ) {
- printf( "getsockname call in child failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
- if ( my_sockaddr->sa_family != SOCK_STREAM ) {
- printf( "getsockname test failed - did not get correct socket name data \n" );
- goto test_failed_exit;
- }
-
- /* make sure we can't seek on a socket */
- my_current_offset = lseek( my_socket_fd, 0, SEEK_CUR );
- if ( my_current_offset != -1 ) {
- printf( "lseek on socket should fail but did not \n" );
- goto test_failed_exit;
- }
-
- /*
- * spin off a child process that we communicate with via sockets.
- */
- my_pid = fork( );
- if ( my_pid == -1 ) {
- printf( "fork failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
- if ( my_pid == 0 ) {
- /*
- * child process - open a socket and use it to talk to our parent.
- */
- int my_child_fd = -1;
- struct msghdr my_msghdr;
- struct iovec my_iov[4];
- char my_buffer[128];
-
- my_child_fd = socket( AF_UNIX, SOCK_STREAM, 0 );
- if ( my_child_fd == -1 ) {
- printf( "socket call in child failed with error %d - \"%s\" \n", errno, strerror( errno) );
- exit( -1 );
- }
-
- my_sockaddr = (struct sockaddr *) &my_child_socket_name[0];
- my_err = bind( my_child_fd, my_sockaddr, my_sockaddr->sa_len );
- if ( my_err == -1 ) {
- close( my_child_fd );
- printf( "bind call in child failed with error %d - \"%s\" \n", errno, strerror( errno) );
- exit( -1 );
- }
- sleep(2);
-
- /* connect to socket in our parent */
- my_sockaddr = (struct sockaddr *) &my_parent_socket_name[0];
- my_err = connect( my_child_fd, my_sockaddr, my_sockaddr->sa_len );
- if ( my_err == -1 ) {
- close( my_child_fd );
- printf( "connect call in child failed with error %d - \"%s\" \n", errno, strerror( errno) );
- exit( -1 );
- }
-
- /* get some data from the child via socket and test socket peer data */
- {
- socklen_t my_buffer_len;
- struct sockaddr * my_sockaddr;
- char my_parent_buffer[256];
-
- my_sockaddr = (struct sockaddr *) &my_parent_buffer[0];
- my_buffer_len = sizeof(my_parent_buffer);
- my_err = getpeername( my_child_fd, my_sockaddr, &my_buffer_len );
- if ( my_err == -1 ) {
- printf( "getpeername call in parent failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-
- /* test results - should be sa_family == SOCK_STREAM and name should match my_child_pathp */
- if ( my_sockaddr->sa_family != SOCK_STREAM ) {
- printf( "getpeername test failed - did not get correct peer data \n" );
- goto test_failed_exit;
- }
- }
-
- my_buffer[0] = 'j';
- my_iov[0].iov_base = &my_buffer[0];
- my_iov[0].iov_len = 1;
-
- my_sockaddr = (struct sockaddr *) &my_parent_socket_name[0];
- my_msghdr.msg_name = my_sockaddr;
- my_msghdr.msg_namelen = my_sockaddr->sa_len;
- my_msghdr.msg_iov = &my_iov[0];
- my_msghdr.msg_iovlen = 1;
- my_msghdr.msg_control = NULL;
- my_msghdr.msg_controllen = 0;
- my_msghdr.msg_flags = 0;
-
- my_result = sendmsg( my_child_fd, &my_msghdr, 0 );
- if ( my_result == -1 ) {
- printf( "sendmsg failed with error %d - \"%s\" \n", errno, strerror( errno) );
- close( my_child_fd );
- exit( -1 );
- }
-
-#if 1
- /* get data from our parent */
- my_result = recvfrom( my_child_fd, &my_buffer[0], 1,
- MSG_WAITALL, NULL, NULL );
- if ( my_result == -1 ) {
- printf( "recvfrom failed with error %d - \"%s\" \n", errno, strerror( errno) );
- close( my_child_fd );
- exit( -1 );
- }
-
- /* verify that we got the correct message from our child */
- if ( my_buffer[0] != 'e' ) {
- printf( "test failed - did not get correct data from child \n" );
- close( my_child_fd );
- exit( -1 );
- }
-#endif
-
- /* sendfile test. Open libsystem, set up some headers, and send it */
- struct sf_hdtr my_sf_hdtr;
- int my_libsys_fd;
- off_t my_libsys_len;
-
- my_libsys_fd = open("/usr/lib/libSystem.dylib", O_RDONLY, 0644);
- if (my_libsys_fd < 0) {
- printf( "test failed - could not open /usr/lib/libSystem.dylib\n" );
- close ( my_child_fd );
- exit ( -1 );
- }
-
- my_libsys_len = 7+2; /* 2 bytes of header */
- my_buffer[0] = 's';
- my_iov[0].iov_base = &my_buffer[0];
- my_iov[0].iov_len = 1;
- my_buffer[1] = 'e';
- my_iov[1].iov_base = &my_buffer[1];
- my_iov[1].iov_len = 1;
- my_buffer[2] = 'n';
- my_iov[2].iov_base = &my_buffer[2];
- my_iov[2].iov_len = 1;
- my_buffer[3] = 'd';
- my_iov[3].iov_base = &my_buffer[3];
- my_iov[3].iov_len = 1;
-
- my_sf_hdtr.headers = &my_iov[0];
- my_sf_hdtr.hdr_cnt = 2;
- my_sf_hdtr.trailers = &my_iov[2];
- my_sf_hdtr.trl_cnt = 2;
-
- my_result = sendfile(my_libsys_fd, my_child_fd, 3, &my_libsys_len, &my_sf_hdtr, 0);
- if (my_result < 0 || my_libsys_len != 11) {
- printf( "sendfile failed with error %d - \"%s\" \n", errno, strerror( errno) );
- close( my_child_fd );
- exit( -1 );
- }
-
- my_result = close ( my_libsys_fd );
- if ( my_libsys_fd < 0 ) {
- printf ( "close failed with error %d - \"%s\" \n", errno, strerror( errno) );
- close ( my_child_fd );
- exit ( -1 );
- }
-
- /* tell parent we're done */
- my_result = write( my_child_fd, "all done", 8 );
- if ( my_result == -1 ) {
- close( my_child_fd );
- exit( -1 );
- }
-
- close( my_child_fd );
- exit(0);
- }
-
- /*
- * parent process - listen for connection requests
- */
- my_err = listen( my_socket_fd, 10 );
- if ( my_err == -1 ) {
- printf( "listen call in parent failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-
- /* accept connection from child */
- my_sockaddr = (struct sockaddr *) &my_accept_buffer[0];
- my_accepted_socket = accept( my_socket_fd, my_sockaddr, &my_accept_len );
- if ( my_accepted_socket == -1 ) {
- printf( "accept call in parent failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-
- /* get some data from the child via socket and test socket peer data */
- {
- //socklen_t my_buffer_len;
- struct msghdr my_msghdr;
- struct iovec my_iov;
- char my_parent_buffer[128];
-
- my_parent_buffer[0] = 'x';
- my_iov.iov_base = &my_parent_buffer[0];
- my_iov.iov_len = 1;
-
- my_msghdr.msg_name = &my_accept_buffer[0];
- my_msghdr.msg_namelen = my_accept_len;
- my_msghdr.msg_iov = &my_iov;
- my_msghdr.msg_iovlen = 1;
- my_msghdr.msg_control = NULL;
- my_msghdr.msg_controllen = 0;
- my_msghdr.msg_flags = 0;
-
- my_result = recvmsg( my_accepted_socket, &my_msghdr, MSG_WAITALL );
- if ( my_result == -1 ) {
- printf( "recvmsg failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-
- /* verify that we got the correct message from our child */
- if ( my_parent_buffer[0] != 'j' ) {
- printf( "test failed - did not get correct data from child \n" );
- goto test_failed_exit;
- }
-
-#if 1
- /* now send some data to our child */
- my_parent_buffer[0] = 'e';
- my_sockaddr = (struct sockaddr *) &my_child_socket_name[0];
- my_result = sendto( my_accepted_socket, &my_parent_buffer[0], 1, 0, my_sockaddr,
- my_sockaddr->sa_len );
- if ( my_result == -1 ) {
- printf( "sendto failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
-#endif
-
- size_t neededBytes = 11;
-
- /* Check for sendfile output */
- bzero( (void *)&my_parent_buffer[0], sizeof(my_parent_buffer) );
- while (neededBytes > 0) {
- my_result = read( my_accepted_socket, &my_parent_buffer[11-neededBytes], neededBytes );
- if ( my_result == -1 ) {
- printf( "read call failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- } else if (my_result == 0) {
- break;
- }
- neededBytes -= my_result;
- }
-
- if ( neededBytes > 0 ) {
- printf( "read call returned %ld bytes instead of 11\n", 11 - neededBytes );
- goto test_failed_exit;
- }
-
- if ( ! (my_parent_buffer[0] == 's' && my_parent_buffer[1] == 'e' && my_parent_buffer[9] == 'n' && my_parent_buffer[10] == 'd') ) {
- printf( "read wrong sendfile message from child \n" );
- goto test_failed_exit;
- }
-
-
- /* see if child is done. bzero so that string is NUL terminated */
- bzero( (void *)&my_parent_buffer[0], sizeof(my_parent_buffer) );
- my_result = read( my_accepted_socket, &my_parent_buffer[0], sizeof(my_parent_buffer) );
- if ( my_result == -1 ) {
- printf( "read call failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
- if ( strcmp( "all done", &my_parent_buffer[0] ) != 0 ) {
- printf( "read wrong message from child \n" );
- goto test_failed_exit;
- }
- }
-
- /* wait for child to exit */
- my_wait_pid = wait4( my_pid, &my_status, 0, NULL );
- if ( my_wait_pid == -1 ) {
- printf( "wait4 failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
-
- if ( WIFEXITED( my_status ) && WEXITSTATUS( my_status ) != 0 ) {
- goto test_failed_exit;
- }
-
- my_err = 0;
- goto test_passed_exit;
-
-test_failed_exit:
- my_err = -1;
-
-test_passed_exit:
- if ( my_socket_fd != -1 )
- close( my_socket_fd );
- if ( my_accepted_socket != -1 )
- close( my_accepted_socket );
- if ( my_parent_pathp != NULL ) {
- remove( my_parent_pathp );
- vm_deallocate(mach_task_self(), (vm_address_t)my_parent_pathp, 128);
- }
- if ( my_child_pathp != NULL ) {
- remove( my_child_pathp );
- vm_deallocate(mach_task_self(), (vm_address_t)my_child_pathp, 128);
- }
- return( my_err );
-}
-
-/* **************************************************************************************************************
- * Test fsync, getsockopt, poll, select, setsockopt, socketpair system calls.
- * **************************************************************************************************************
- */
-int socket2_tests( void * the_argp )
-{
- int my_err, my_status;
- int my_sockets[ 2 ] = {-1, -1};
- pid_t my_pid, my_wait_pid;
- ssize_t my_count;
- socklen_t my_socklen;
- struct timeval * my_tvp;
- struct timeval my_orig_tv;
- char my_buffer[ 32 ];
-
- my_err = socketpair( AF_UNIX, SOCK_STREAM, 0, &my_sockets[0] );
- if ( my_err == -1 ) {
- printf( "socketpair failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
-
- /* test getsockopt and setsockopt */
- my_socklen = sizeof( my_buffer );
- my_err = getsockopt( my_sockets[0], SOL_SOCKET, SO_TYPE, &my_buffer[0], &my_socklen);
- if ( my_err == -1 ) {
- printf( "getsockopt - SO_TYPE - failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
- if ( SOCK_STREAM != *((int *)&my_buffer[0]) ) {
- printf( "getsockopt returned incorrect socket type \n" );
- goto test_failed_exit;
- }
-
- /* get and set receive timeout */
- my_socklen = sizeof( my_buffer );
- my_err = getsockopt( my_sockets[0], SOL_SOCKET, SO_RCVTIMEO, &my_buffer[0], &my_socklen);
- if ( my_err == -1 ) {
- printf( "getsockopt - SO_RCVTIMEO - failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
- my_tvp = (struct timeval *) &my_buffer[0];
- my_orig_tv.tv_sec = my_tvp->tv_sec;
- my_orig_tv.tv_usec = my_tvp->tv_usec;
-
- my_tvp->tv_sec += 60;
- my_err = setsockopt( my_sockets[0], SOL_SOCKET, SO_RCVTIMEO, &my_buffer[0], sizeof(struct timeval) );
- if ( my_err == -1 ) {
- printf( "setsockopt - SO_RCVTIMEO - failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
-
- /* verify we set it */
- my_socklen = sizeof( my_buffer );
- my_err = getsockopt( my_sockets[0], SOL_SOCKET, SO_RCVTIMEO, &my_buffer[0], &my_socklen);
- if ( my_err == -1 ) {
- printf( "getsockopt - SO_RCVTIMEO - failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
- my_tvp = (struct timeval *) &my_buffer[0];
- if ( my_tvp->tv_sec != (my_orig_tv.tv_sec + 60) || my_tvp->tv_usec != my_orig_tv.tv_usec ) {
- printf( "setsockopt - SO_RCVTIMEO - did not set correct timeval \n" );
- goto test_failed_exit;
- }
-
- /* set back to original receive timeout */
- my_err = setsockopt( my_sockets[0], SOL_SOCKET, SO_RCVTIMEO, &my_orig_tv, sizeof(struct timeval) );
- if ( my_err == -1 ) {
- printf( "setsockopt - SO_RCVTIMEO - failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
-
- /* test fsync - should fail when used with a socket fd */
- errno = 0;
- my_err = fsync( my_sockets[0] );
- if ( my_err == -1 && errno != ENOTSUP ) {
- printf( "fsync failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
- else if ( my_err != -1 ) {
- printf( "fsync should have failed with errno ENOTSUP \n" );
- goto test_failed_exit;
- }
-
- /*
- * spin off a child process that we will talk to via our socketpair.
- */
- my_pid = fork( );
- if ( my_pid == -1 ) {
- printf( "fork failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
- if ( my_pid == 0 ) {
- /*
- * child process - tell parent we are ready to go.
- */
- char my_buffer[ 32 ];
- struct pollfd my_pollfd;
-
- my_count = write( my_sockets[1], "r", 1 );
- if ( my_count == -1 ) {
- printf( "write call failed. got errno %d - %s. \n", errno, strerror( errno ) );
- exit( -1 );
- }
-
- /* test select by using it to wait for message from parent */
- for ( ;; ) {
- fd_set my_read_set;
- struct timeval my_timeout;
-
- FD_ZERO( &my_read_set );
- FD_SET( my_sockets[1], &my_read_set );
- timerclear( &my_timeout );
- my_timeout.tv_sec = 1;
-
- /* check to see if we are done, if no message is ready after a second
- * return and try again...
- */
- my_err = select( (my_sockets[1] + 1), &my_read_set, NULL, NULL, &my_timeout );
- if ( my_err == -1 ) {
- printf( "select call failed with error %d - \"%s\" \n", errno, strerror( errno) );
- exit( -1 );
- }
- else if ( my_err > 0 ) {
- /* we're done */
- break;
- }
- }
-
- /* test poll too */
- my_pollfd.fd = my_sockets[1];
- my_pollfd.events = (POLLIN | POLLPRI);
- my_pollfd.revents = 0;
- my_err = poll( &my_pollfd, 1, 500 );
- if ( my_err == -1 ) {
- printf( "poll call failed with error %d - \"%s\" \n", errno, strerror( errno) );
- exit( -1 );
- }
- /* should be ready for read */
- if ( (my_pollfd.revents & POLLIN) == 0 ) {
- printf( "poll should have returned ready for read \n" );
- exit( -1 );
- }
-
- my_count = read( my_sockets[1], &my_buffer[0], sizeof(my_buffer) );
- if ( my_count == -1 ) {
- printf( "read call failed with error %d - \"%s\" \n", errno, strerror( errno) );
- exit( -1 );
- }
- if ( my_buffer[0] != 'd' ) {
- printf( "read call on socket failed to get \"all done\" message \n" );
- exit( -1 );
- }
-
- exit(0);
- }
-
- /*
- * parent process - wait for child to spin up
- */
- my_count = read( my_sockets[0], &my_buffer[0], sizeof(my_buffer) );
- if ( my_count == -1 ) {
- printf( "read call failed with error %d - \"%s\" \n", errno, strerror( errno) );
- goto test_failed_exit;
- }
- if ( my_buffer[0] != 'r' ) {
- printf( "read call on socket failed to get \"ready to go message\" \n" );
- goto test_failed_exit;
- }
-
- /* tell child we're done */
- write( my_sockets[0], "d", 1 );
-
- my_wait_pid = wait4( my_pid, &my_status, 0, NULL );
- if ( my_wait_pid == -1 ) {
- printf( "wait4 failed with errno %d - %s \n", errno, strerror( errno ) );
- goto test_failed_exit;
- }
-
- /* wait4 should return our child's pid when it exits */
- if ( my_wait_pid != my_pid ) {
- printf( "wait4 did not return child pid - returned %d should be %d \n", my_wait_pid, my_pid );
- goto test_failed_exit;
- }
-
- if ( WIFEXITED( my_status ) && WEXITSTATUS( my_status ) != 0 ) {
- printf( "wait4 returned wrong exit status - 0x%02X \n", my_status );
- goto test_failed_exit;
- }
-
- my_err = 0;
- goto test_passed_exit;
-
-test_failed_exit:
- my_err = -1;
-
-test_passed_exit:
- if ( my_sockets[0] != -1 )
- close( my_sockets[0] );
- if ( my_sockets[1] != -1 )
- close( my_sockets[1] );
- return( my_err );
-}
-
projects / apple / xnu.git / blobdiff