upstream/mercurial-mirror Files · rust/chg/src/sendfds.c

tests: use sha256line.py instead of /dev/random in test-censor.t (issue6858)...

tests: use sha256line.py instead of /dev/random in test-censor.t (issue6858) Sometimes the systems that run our test suite don't have enough entropy and they cannot produce target file of the expected size using /dev/random, which results in test failures. Switching to /dev/urandom would give us way more available data at the cost of it being less "random", but we don't really need to use entropy for this task at all, since we only care if the file size after compression is big enough to not be stored inline in the revlog. So let's use something that we already have used to generate this kind of data in other tests.

Yuya Nishihara - - Load All Authors

File last commit:

r40005:208cb7a9 default


                r52255:e7be2ddf

stable

Download file

             sendfds.c
        
                    51 lines
            
             | 1.3 KiB
            
                | text/x-c
            
             |
                CLexer
            
             / rust / chg / src / sendfds.c
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      /*

       * Utility to send fds via Unix domain socket

       *

       * Copyright 2011, 2018 Yuya Nishihara <yuya@tcha.org>

       *

       * This software may be used and distributed according to the terms of the

       * GNU General Public License version 2 or any later version.

       */

      #include <errno.h>

      #include <stdlib.h>

      #include <string.h>

      #include <sys/socket.h>

      #include <sys/types.h>

      #define MAX_FD_LEN 10

      /*

       * Sends the given fds with 1-byte dummy payload.

       *

       * Returns the number of bytes sent on success, -1 on error and errno is set

       * appropriately.

       */

      ssize_t sendfds(int sockfd, const int *fds, size_t fdlen)

      {

      	char dummy[1] = {0};

      	struct iovec iov = {dummy, sizeof(dummy)};

      	char fdbuf[CMSG_SPACE(sizeof(fds[0]) * MAX_FD_LEN)];

      	struct msghdr msgh;

      	struct cmsghdr *cmsg;

      	/* just use a fixed-size buffer since we'll never send tons of fds */

      	if (fdlen > MAX_FD_LEN) {

      		errno = EINVAL;

      		return -1;

      	}

      	memset(&msgh, 0, sizeof(msgh));

      	msgh.msg_iov = &iov;

      	msgh.msg_iovlen = 1;

      	msgh.msg_control = fdbuf;

      	msgh.msg_controllen = CMSG_SPACE(sizeof(fds[0]) * fdlen);

      	cmsg = CMSG_FIRSTHDR(&msgh);

      	cmsg->cmsg_level = SOL_SOCKET;

      	cmsg->cmsg_type = SCM_RIGHTS;

      	cmsg->cmsg_len = CMSG_LEN(sizeof(fds[0]) * fdlen);

      	memcpy(CMSG_DATA(cmsg), fds, sizeof(fds[0]) * fdlen);

      	msgh.msg_controllen = cmsg->cmsg_len;

      	return sendmsg(sockfd, &msgh, 0);

      }

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				/*
				* Utility to send fds via Unix domain socket
				*
				* Copyright 2011, 2018 Yuya Nishihara <yuya@tcha.org>
				*
				* This software may be used and distributed according to the terms of the
				* GNU General Public License version 2 or any later version.
				*/

				#include <errno.h>
				#include <stdlib.h>
				#include <string.h>
				#include <sys/socket.h>
				#include <sys/types.h>

				#define MAX_FD_LEN 10

				/*
				* Sends the given fds with 1-byte dummy payload.
				*
				* Returns the number of bytes sent on success, -1 on error and errno is set
				* appropriately.
				*/
				ssize_t sendfds(int sockfd, const int *fds, size_t fdlen)
				{
				char dummy[1] = {0};
				struct iovec iov = {dummy, sizeof(dummy)};
				char fdbuf[CMSG_SPACE(sizeof(fds[0]) * MAX_FD_LEN)];
				struct msghdr msgh;
				struct cmsghdr *cmsg;

				/* just use a fixed-size buffer since we'll never send tons of fds */
				if (fdlen > MAX_FD_LEN) {
				errno = EINVAL;
				return -1;
				}

				memset(&msgh, 0, sizeof(msgh));
				msgh.msg_iov = &iov;
				msgh.msg_iovlen = 1;
				msgh.msg_control = fdbuf;
				msgh.msg_controllen = CMSG_SPACE(sizeof(fds[0]) * fdlen);

				cmsg = CMSG_FIRSTHDR(&msgh);
				cmsg->cmsg_level = SOL_SOCKET;
				cmsg->cmsg_type = SCM_RIGHTS;
				cmsg->cmsg_len = CMSG_LEN(sizeof(fds[0]) * fdlen);
				memcpy(CMSG_DATA(cmsg), fds, sizeof(fds[0]) * fdlen);
				msgh.msg_controllen = cmsg->cmsg_len;
				return sendmsg(sockfd, &msgh, 0);
				}