1*4d495c6eSApple OSS Distributions /* 2*4d495c6eSApple OSS Distributions * Copyright (c) 2024 Apple Inc. All rights reserved. 3*4d495c6eSApple OSS Distributions * 4*4d495c6eSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ 5*4d495c6eSApple OSS Distributions * 6*4d495c6eSApple OSS Distributions * This file contains Original Code and/or Modifications of Original Code 7*4d495c6eSApple OSS Distributions * as defined in and that are subject to the Apple Public Source License 8*4d495c6eSApple OSS Distributions * Version 2.0 (the 'License'). You may not use this file except in 9*4d495c6eSApple OSS Distributions * compliance with the License. The rights granted to you under the License 10*4d495c6eSApple OSS Distributions * may not be used to create, or enable the creation or redistribution of, 11*4d495c6eSApple OSS Distributions * unlawful or unlicensed copies of an Apple operating system, or to 12*4d495c6eSApple OSS Distributions * circumvent, violate, or enable the circumvention or violation of, any 13*4d495c6eSApple OSS Distributions * terms of an Apple operating system software license agreement. 14*4d495c6eSApple OSS Distributions * 15*4d495c6eSApple OSS Distributions * Please obtain a copy of the License at 16*4d495c6eSApple OSS Distributions * http://www.opensource.apple.com/apsl/ and read it before using this file. 17*4d495c6eSApple OSS Distributions * 18*4d495c6eSApple OSS Distributions * The Original Code and all software distributed under the License are 19*4d495c6eSApple OSS Distributions * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20*4d495c6eSApple OSS Distributions * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21*4d495c6eSApple OSS Distributions * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22*4d495c6eSApple OSS Distributions * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23*4d495c6eSApple OSS Distributions * Please see the License for the specific language governing rights and 24*4d495c6eSApple OSS Distributions * limitations under the License. 25*4d495c6eSApple OSS Distributions * 26*4d495c6eSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27*4d495c6eSApple OSS Distributions */ 28*4d495c6eSApple OSS Distributions 29*4d495c6eSApple OSS Distributions #include <vsock_helpers.h> 30*4d495c6eSApple OSS Distributions 31*4d495c6eSApple OSS Distributions T_GLOBAL_META( 32*4d495c6eSApple OSS Distributions T_META_RUN_CONCURRENTLY(true), 33*4d495c6eSApple OSS Distributions T_META_NAMESPACE("xnu.vsock") 34*4d495c6eSApple OSS Distributions ); 35*4d495c6eSApple OSS Distributions 36*4d495c6eSApple OSS Distributions T_DECL(vsock_private_connect_with_entitlement, "vsock private connect should succeed with entitlement") 37*4d495c6eSApple OSS Distributions { 38*4d495c6eSApple OSS Distributions const uint32_t port = 1234; 39*4d495c6eSApple OSS Distributions 40*4d495c6eSApple OSS Distributions struct sockaddr_vm listen_addr; 41*4d495c6eSApple OSS Distributions int listen_socket; 42*4d495c6eSApple OSS Distributions int result = vsock_private_listen(VMADDR_CID_ANY, port, &listen_addr, 1, &listen_socket); 43*4d495c6eSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(result, "vsock listen with entitlement"); 44*4d495c6eSApple OSS Distributions 45*4d495c6eSApple OSS Distributions const uint32_t connection_cid = vsock_get_local_cid(listen_socket); 46*4d495c6eSApple OSS Distributions 47*4d495c6eSApple OSS Distributions int connected_socket = vsock_private_new_socket(); 48*4d495c6eSApple OSS Distributions struct sockaddr_vm addr = (struct sockaddr_vm) { 49*4d495c6eSApple OSS Distributions .svm_cid = connection_cid, 50*4d495c6eSApple OSS Distributions .svm_port = port, 51*4d495c6eSApple OSS Distributions }; 52*4d495c6eSApple OSS Distributions result = connect(connected_socket, (struct sockaddr *)&addr, sizeof(addr)); 53*4d495c6eSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(result, "vsock connection with entitlement"); 54*4d495c6eSApple OSS Distributions 55*4d495c6eSApple OSS Distributions vsock_close(connected_socket); 56*4d495c6eSApple OSS Distributions vsock_close(listen_socket); 57*4d495c6eSApple OSS Distributions } 58*4d495c6eSApple OSS Distributions 59*4d495c6eSApple OSS Distributions T_DECL(vsock_private_bind_with_entitlement, "vsock private bind should succeed with entitlement") 60*4d495c6eSApple OSS Distributions { 61*4d495c6eSApple OSS Distributions int socket; 62*4d495c6eSApple OSS Distributions struct sockaddr_vm addr; 63*4d495c6eSApple OSS Distributions int result = vsock_private_bind(VMADDR_CID_ANY, 1234, &addr, &socket); 64*4d495c6eSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(result, "vsock bind with entitlement"); 65*4d495c6eSApple OSS Distributions } 66