share_fd.c
上传用户:tjescc
上传日期:2021-02-23
资源大小:419k
文件大小:4k
源码类别:

Telnet服务器

开发平台:

Unix_Linux

  1. #include <includes.h>
  2. /*
  3.  * send_fd() and recv_fd() come from OpenSSH-23/openssh/monitor_fdpass.c.
  4.  * These two functions have been written by Niels Provos :
  5.  *
  6.  * Copyright 2001 Niels Provos <provos@citi.umich.edu>
  7.  * All rights reserved.
  8.  *
  9.  * Redistribution and use in source and binary forms, with or without
  10.  * modification, are permitted provided that the following conditions
  11.  * are met:
  12.  * 1. Redistributions of source code must retain the above copyright
  13.  *    notice, this list of conditions and the following disclaimer.
  14.  * 2. Redistributions in binary form must reproduce the above copyright
  15.  *    notice, this list of conditions and the following disclaimer in the
  16.  *    documentation and/or other materials provided with the distribution.
  17.  *
  18.  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  19.  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  20.  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  21.  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  22.  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  23.  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24.  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25.  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26.  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27.  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28.  */
  29. int send_fd(int socket, int fd)
  30. {
  31. #if defined(HAVE_SENDMSG) && (defined(HAVE_ACCRIGHTS_IN_MSGHDR) || defined(HAVE_CONTROL_IN_MSGHDR))
  32. struct msghdr msg;
  33. struct iovec vec;
  34. char ch = '';
  35. int n;
  36. #ifndef HAVE_ACCRIGHTS_IN_MSGHDR
  37. char tmp[CMSG_SPACE(sizeof(int))];
  38. struct cmsghdr *cmsg;
  39. #endif
  40. memset(&msg, 0, sizeof(msg));
  41. #ifdef HAVE_ACCRIGHTS_IN_MSGHDR
  42. msg.msg_accrights = (caddr_t)&fd;
  43. msg.msg_accrightslen = sizeof(fd);
  44. #else
  45. msg.msg_control = (caddr_t)tmp;
  46. msg.msg_controllen = CMSG_LEN(sizeof(int));
  47. cmsg = CMSG_FIRSTHDR(&msg);
  48. cmsg->cmsg_len = CMSG_LEN(sizeof(int));
  49. cmsg->cmsg_level = SOL_SOCKET;
  50. cmsg->cmsg_type = SCM_RIGHTS;
  51. *(int *)CMSG_DATA(cmsg) = fd;
  52. #endif
  53. vec.iov_base = &ch;
  54. vec.iov_len = 1;
  55. msg.msg_iov = &vec;
  56. msg.msg_iovlen = 1;
  57. if ((n = sendmsg(socket, &msg, 0)) == -1)
  58. fprintf(stderr, "send_fd(): sendmsg(%d): %s", fd,
  59.     strerror(errno));
  60. if (n != 1)
  61. fprintf(stderr, "send_fd(): sendmsg: expected sent 1 got %d", n);
  62.  return 0;
  63. #else
  64.  return -1;
  65. #endif
  66. }
  67. int
  68. recv_fd(int socket)
  69. {
  70. #if defined(HAVE_RECVMSG) && (defined(HAVE_ACCRIGHTS_IN_MSGHDR) || defined(HAVE_CONTROL_IN_MSGHDR))
  71. struct msghdr msg;
  72. struct iovec vec;
  73. char ch;
  74. int fd, n;
  75. #ifndef HAVE_ACCRIGHTS_IN_MSGHDR
  76. char tmp[CMSG_SPACE(sizeof(int))];
  77. struct cmsghdr *cmsg;
  78. #endif
  79. memset(&msg, 0, sizeof(msg));
  80. vec.iov_base = &ch;
  81. vec.iov_len = 1;
  82. msg.msg_iov = &vec;
  83. msg.msg_iovlen = 1;
  84. #ifdef HAVE_ACCRIGHTS_IN_MSGHDR
  85. msg.msg_accrights = (caddr_t)&fd;
  86. msg.msg_accrightslen = sizeof(fd);
  87. #else
  88. msg.msg_control = tmp;
  89. msg.msg_controllen = sizeof(tmp);
  90. #endif
  91. if ((n = recvmsg(socket, &msg, 0)) == -1)
  92. printf("%s: recvmsg: %s", __func__, strerror(errno));
  93. if (n != 1)
  94. printf("%s: recvmsg: expected received 1 got %d",
  95.     __func__, n);
  96. #ifdef HAVE_ACCRIGHTS_IN_MSGHDR
  97. if (msg.msg_accrightslen != sizeof(fd))
  98. printf("recv_fd(): no fdn");
  99. #else
  100. cmsg = CMSG_FIRSTHDR(&msg);
  101. if (cmsg->cmsg_type != SCM_RIGHTS)
  102. printf("recv_fd():  expected type %d got %d",
  103.     SCM_RIGHTS, cmsg->cmsg_type);
  104. fd = (*(int *)CMSG_DATA(cmsg));
  105. #endif
  106. return fd;
  107. #else
  108. return -1;
  109. #endif
  110. }