1/* 2 * Copyright (c) 2000-2001,2003-2004 Apple Computer, Inc. All Rights Reserved. 3 * 4 * @APPLE_LICENSE_HEADER_START@ 5 * 6 * This file contains Original Code and/or Modifications of Original Code 7 * as defined in and that are subject to the Apple Public Source License 8 * Version 2.0 (the 'License'). You may not use this file except in 9 * compliance with the License. Please obtain a copy of the License at 10 * http://www.opensource.apple.com/apsl/ and read it before using this 11 * file. 12 * 13 * The Original Code and all software distributed under the License are 14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 18 * Please see the License for the specific language governing rights and 19 * limitations under the License. 20 * 21 * @APPLE_LICENSE_HEADER_END@ 22 */ 23 24 25// 26// fdmover - send/receive file descriptors over a UNIX domain socket connection 27// 28#include "fdmover.h" 29#include <iterator> 30#include <security_utilities/debugging.h> 31 32 33namespace Security { 34namespace IPPlusPlus { 35 36 37void *FdMover::Element::operator new (size_t base, size_t more) 38{ 39 Element *element = (Element *)::malloc(CMSG_SPACE(more)); 40 element->cmsg_len = (socklen_t)CMSG_LEN(more); 41 return element; 42} 43 44void FdMover::Element::operator delete (void *data, size_t base) 45{ 46 ::free(data); 47} 48 49FdMover::Element::Element(int level, int type) 50{ 51 cmsg_level = level; 52 cmsg_type = type; 53} 54 55 56FdMover::Message::Message(const void *data, size_t length) 57 : iovec(data, length) 58{ 59 msg_name = NULL; 60 msg_namelen = 0; 61 msg_iov = &iovec; 62 msg_iovlen = 1; 63 msg_control = NULL; 64 msg_controllen = 0; 65 msg_flags = 0; 66} 67 68void FdMover::Message::set(Element *elem) 69{ 70 msg_control = (caddr_t)elem; 71 msg_controllen = elem->cmsg_len; 72} 73 74 75size_t FdMover::send(const void *data, size_t length, const FdVector &fds) 76{ 77 auto_ptr<Element> elem(new (fds.size() * sizeof(int)) Element (SOL_SOCKET, SCM_RIGHTS)); 78 copy(fds.begin(), fds.end(), &elem.get()->payload<int>()); 79 Message msg(data, length); 80 msg.set(elem.get()); 81 ssize_t rc = ::sendmsg(fd(), &msg, 0); 82 checkError(rc); 83 return rc; 84} 85 86 87size_t FdMover::receive(void *data, size_t length, FdVector &fds) 88{ 89 static const int maxFds = 20; // arbitrary limit 90 Message msg(data, length); 91 auto_ptr<Element> elem(new (maxFds * sizeof(int)) Element); 92 msg.set(elem.get()); 93 ssize_t rc = ::recvmsg(fd(), &msg, 0); 94 checkError(rc); 95 size_t count = elem.get()->payloadSize() / sizeof(int); 96 FdVector result; 97 copy(&elem.get()->payload<int>(), &elem.get()->payload<int>() + count, back_inserter(result)); 98 swap(fds, result); 99 return rc; 100} 101 102 103} // end namespace IPPlusPlus 104} // end namespace Security 105