FIFO 及 unix域通信代码,C代码库,德仔网

代码语言
代码分类
【C】 FIFO 及 unix域通信
作者:ljfabcd / 发布于2013/10/23/ 479

	#include <stdio.h>
	#include <stdlib.h>
	#include <string.h>
	#include <sys/types.h>
	#include <sys/stat.h>
	#include <sys/ioctl.h>
	#include <sys/poll.h>
	#include <sys/time.h>
	#include <fcntl.h>
	#include <errno.h>
	#include <pthread.h>
	#include <math.h>
	#include <unistd.h>
	#include <sys/mman.h>
	#include <sys/resource.h>
	#include <sys/types.h>
	#include <sys/wait.h>
	#include <sys/signal.h>
	#include <sys/time.h>
	#include <sys/socket.h>
	#include <stdint.h>

	
	//block until server open
	int open_client_fifo(char* server)
	{
	int fd_file;
	
	fd_file = open(server, O_WRONLY);
	if(fd_file <= 0)
	{
	 perror("create fifo error");
	 return -1;
	}

	return fd_file;
	}

	int open_server_fifo(char* server)
	{
	int fd_file;
	
	remove(server);
	mkfifo(server, 0666);

	fd_file = open(server, O_RDONLY | O_NONBLOCK);
	if(fd_file <= 0)
	{
	 perror("create fifo error");
	 return -1;
	}

	if(fd_file > 0)
	 fcntl(fd_file, F_SETFL, fcntl(fd_file, F_GETFL) & (~O_NONBLOCK));
	
	return fd_file;
	}


	/*
	* Create a client endpoint and connect to a server.
	* Returns fd if all OK, <0 on error.
	*/
	int
	cli_conn(const char *server, const char *client)
	{
	int fd, len, err, rval;
	struct sockaddr_un un;

	errno=0;
	/* create a UNIX domain stream socket */
	if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
	{
	 perror("cli_conn");
	 return(-1);
	}
	
	//DEBUG("&&&&&cli_conn fd=%d\n",fd);
	/* fill socket address structure with our address */
	memset(&un, 0, sizeof(un));
	un.sun_family = AF_UNIX;
	sprintf(un.sun_path, "%sd", client, getpid());
	len = offsetof(struct sockaddr_un, sun_path) + strlen(un.sun_path);

	unlink(un.sun_path); /* in case it already exists */
	if (bind(fd, (struct sockaddr *)&un, len) < 0) {
	 rval = -2;
	 goto errout;
	}
	if (chmod(un.sun_path, CLI_PERM) < 0) {
	 rval = -3;
	 goto errout;
	}

	/* fill socket address structure with server's address */
	memset(&un, 0, sizeof(un));
	un.sun_family = AF_UNIX;
	strcpy(un.sun_path, server);
	len = offsetof(struct sockaddr_un, sun_path) + strlen(server);
	if (connect(fd, (struct sockaddr *)&un, len) < 0) {
	 rval = -4;
	 goto errout;
	}
	return(fd);

	errout:
	err = errno;
	if(fd>0)
	 close(fd);
	errno = err;
	return(rval);
	}

	
	ssize_t /* Read "n" bytes from a descriptor. */
	recvn(int fd, void *vptr, size_t n)
	{
	size_t nleft;
	ssize_t nread;
	char *ptr;

	ptr = vptr;
	nleft = n;
	while (nleft > 0) {
	 if ( (nread = read(fd, ptr, nleft)) < 0) {
	 if (errno == EINTR)
	 nread = 0; /* and call read() again */
	 else
	 return(-1);
	 } else if (nread == 0)
	 break; /* EOF */

	 nleft -= nread;
	 ptr += nread;
	}
	
	return(n - nleft); /* return >= 0 */
	}


	ssize_t /* Write "n" bytes to a descriptor. */
	sendn(int fd, const void *vptr, size_t n)
	{
	size_t nleft;
	ssize_t nwritten;
	const char *ptr;

	ptr = vptr;
	nleft = n;
	while (nleft > 0) {
	 if ( (nwritten = write(fd, ptr, nleft)) <= 0) {
	 if (nwritten < 0 && errno == EINTR)
	 nwritten = 0; /* and call write() again */
	 else
	 return(-1); /* error */
	 }

	 nleft -= nwritten;
	 ptr += nwritten;
	}
	return(n);
	}

	

	/*
	* Wait for a client connection to arrive, and accept it.
	* We also obtain the client's user ID from the pathname
	* that it must bind before calling us.
	* Returns new fd if all OK, <0 on error
	*/
	int
	serv_accept(int listenfd, uid_t *uidptr)
	{
	int clifd, len, err, rval;
	struct sockaddr_un un;
	struct stat statbuf;

	len = sizeof(un);
	if ((clifd = accept(listenfd, (struct sockaddr *)&un, &len)) < 0)
	 return(-1); /* often errno=EINTR, if signal caught */

	/* obtain the client's uid from its calling address */
	len -= offsetof(struct sockaddr_un, sun_path); /* len of pathname */
	un.sun_path[len] = 0; /* null terminate */

	if (stat(un.sun_path, &statbuf) < 0) {
	 rval = -2;
	 goto errout;
	}
	#ifdef S_ISSOCK /* not defined for SVR4 */
	if (S_ISSOCK(statbuf.st_mode) == 0) {
	 rval = -3; /* not a socket */
	 goto errout;
	}
	#endif
	if ((statbuf.st_mode & (S_IRWXG | S_IRWXO)) ||
	 (statbuf.st_mode & S_IRWXU) != S_IRWXU) {
	 rval = -4; /* is not rwx------ */
	 goto errout;
	}
	#if 0
	staletime = time(NULL) - STALE;
	if (statbuf.st_atime < staletime ||
	 statbuf.st_ctime < staletime ||
	 statbuf.st_mtime < staletime) {
	 rval = -5; /* i-node is too old */
	 goto errout;
	}
	#endif
	if (uidptr != NULL)
	 *uidptr = statbuf.st_uid; /* return uid of caller */
	unlink(un.sun_path); /* we're done with pathname now */
	return(clifd);

	errout:
	err = errno;
	if(clifd>0)
	 close(clifd);
	errno = err;
	return(rval);
	}

	

	/*
	* Create a server endpoint of a connection.
	* Returns fd if all OK, <0 on error.
	*/
	int
	serv_listen(const char *name)
	{
	int fd, len, err, rval;
	struct sockaddr_un un;

	/* create a UNIX domain stream socket */
	if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
	 return(-1);

	unlink(name); /* in case it already exists */

	/* fill in socket address structure */
	memset(&un, 0, sizeof(un));
	un.sun_family = AF_UNIX;
	strcpy(un.sun_path, name);
	len = offsetof(struct sockaddr_un, sun_path) + strlen(name);

	/* bind the name to the descriptor */
	if (bind(fd, (struct sockaddr *)&un, len) < 0) {
	 rval = -2;
	 goto errout;
	}
	if (listen(fd, QLEN) < 0) { /* tell kernel we're a server */
	 rval = -3;
	 goto errout;
	}
	return(fd);

	errout:
	err = errno;
	if(fd>0)
	 close(fd);
	errno = err;
	return(rval);
	}
评论列表
本站所提供的代码，版权归原作者所有，若有侵犯作者版权，请与我们联系，我们将立即删除或修改。谢谢!
本站所有代码发布及提供者。
试试其它关键字
　FIFO　
同语言下
可能有用的
ljfabcd贡献的其它代码(1)
.FIFO 及 unix域通信