我的splice程序在Ubuntu正常,在CentOS就出错。各位大哥帮我看看

我要思考

1. #ifndef SPLICE_H
   
2. #define SPLICE_H
   
3. 
   
4. #include <errno.h>
   
5. #include <sys/uio.h>
   
6. #include <sys/stat.h>
   
7. #include <linux/unistd.h>
   
8. 
   
9. #if defined(__i386__)
   
10. #define __NR_sys_splice                313
    
11. #define __NR_sys_tee                315
    
12. #define __NR_sys_vmsplice        316
    
13. #elif defined(__x86_64__)
    
14. #define __NR_sys_splice                275
    
15. #define __NR_sys_tee                276
    
16. #define __NR_sys_vmsplice        278
    
17. #elif defined(__powerpc__) || defined(__powerpc64__)
    
18. #define __NR_sys_splice                283
    
19. #define __NR_sys_tee                284
    
20. #define __NR_sys_vmsplice        285
    
21. #elif defined(__ia64__)
    
22. #define __NR_sys_splice                1297
    
23. #define __NR_sys_tee                1301
    
24. #define __NR_sys_vmsplice        1302
    
25. #else
    
26. #error unsupported arch
    
27. #endif
    
28. 
    
29. #ifndef SPLICE_F_MOVE
    
30. 
    
31. #define SPLICE_F_MOVE        (0x01)        /* move pages instead of copying */
    
32. #define SPLICE_F_NONBLOCK (0x02) /* don't block on the pipe splicing (but */
    
33.                                  /* we may still block on the fd we splice */
    
34.                                  /* from/to, of course */
    
35. #define SPLICE_F_MORE        (0x04)        /* expect more data */
    
36. #define SPLICE_F_GIFT   (0x08)  /* pages passed in are a gift */
    
37. 
    
38. #endif /* SPLICE_F_MOVE defined */
    
39. 
    
40. /*
    
41. * SPLICE_F_UNMAP was introduced later, so check for that seperately
    
42. */
    
43. #ifndef SPLICE_F_UNMAP
    
44. #define SPLICE_F_UNMAP        (0x10)        /* undo vmsplice map */
    
45. #endif
    
46. 
    
47. static inline int ssplice(int fdin, loff_t *off_in, int fdout, loff_t *off_out,
    
48.                           size_t len, unsigned int flags)
    
49. {
    
50.        
    
51.         return syscall(__NR_sys_splice, fdin, off_in, fdout, off_out, len, flags);
    
52. }
    
53. 
    
54. static inline int stee(int fdin, int fdout, size_t len, unsigned int flags)
    
55. {
    
56.         return syscall(__NR_sys_tee, fdin, fdout, len, flags);
    
57. }
    
58. 
    
59. static inline int svmsplice(int fd, const struct iovec *iov,
    
60.                             unsigned long nr_segs, unsigned int flags)
    
61. {
    
62.         return syscall(__NR_sys_vmsplice, fd, iov, nr_segs, flags);
    
63. }
    
64. 
    
65. #define SPLICE_SIZE        (64*1024)
    
66. 
    
67. #define BUG_ON(c) assert(!(c))
    
68. 
    
69. #define min(x,y) ({ \
    
70.         typeof(x) _x = (x);     \
    
71.         typeof(y) _y = (y);     \
    
72.         (void) (&_x == &_y);            \
    
73.         _x < _y ? _x : _y; })
    
74. 
    
75. #define max(x,y) ({ \
    
76.         typeof(x) _x = (x);     \
    
77.         typeof(y) _y = (y);     \
    
78.         (void) (&_x == &_y);            \
    
79.         _x > _y ? _x : _y; })
    
80. 
    
81. static inline int error(const char *n)
    
82. {
    
83.         perror(n);
    
84.         return -1;
    
85. }
    
86. 
    
87. static int __check_pipe(int pfd)
    
88. {
    
89.         struct stat sb;
    
90. 
    
91.         if (fstat(pfd, &sb) < 0)
    
92.                 return error("stat");
    
93.         if (!S_ISFIFO(sb.st_mode))
    
94.                 return 1;
    
95. 
    
96.         return 0;
    
97. }
    
98. 
    
99. static inline int check_input_pipe(void)
    
100. {
     
101.         if (!__check_pipe(STDIN_FILENO))
     
102.                 return 0;
     
103. 
     
104.         fprintf(stderr, "stdin must be a pipe\n");
     
105.         return 1;
     
106. }
     
107. 
     
108. static inline int check_output_pipe(void)
     
109. {
     
110.         if (!__check_pipe(STDOUT_FILENO))
     
111.                 return 0;
     
112. 
     
113.         fprintf(stderr, "stdout must be a pipe\n");
     
114.         return 1;
     
115. }
     
116. 
     
117. #endif
     

复制代码

1. #include <time.h>
   
2. #include <stdio.h>
   
3. #include <errno.h>
   
4. #include <fcntl.h>
   
5. #include <netdb.h>
   
6. #include <stdint.h>
   
7. #include <stdlib.h>
   
8. #include <string.h>
   
9. #include <unistd.h>
   
10. #include <sys/stat.h>
    
11. #include <sys/time.h>
    
12. #include <sys/epoll.h>
    
13. #include <sys/types.h>
    
14. #include <arpa/inet.h>
    
15. #include <sys/ioctl.h>
    
16. #include <sys/select.h>
    
17. #include <sys/socket.h>
    
18. #include <netinet/in.h>
    
19. #include <netinet/ip.h>
    
20. #include <netinet/tcp.h>
    
21. #include <sys/resource.h>
    
22. 
    
23. 
    
24. #include "splice.h"
    
25. 
    
26. 
    
27. #define DestPort 80
    
28. 
    
29. int pipefd[2];
    
30. 
    
31. // Support for non-blocking I/O
    
32. static int set_nonblocking(const int sockfd)
    
33. {
    
34.         int flags = 1;
    
35. 
    
36.         if (ioctl(sockfd, FIONBIO, &flags) && ((flags = fcntl(sockfd, F_GETFL, 0)) < 0 || fcntl(sockfd, F_SETFL, flags | O_NONBLOCK) < 0))
    
37.         {
    
38.                 close(sockfd);
    
39.                 return -1;
    
40.         }
    
41. }
    
42. 
    
43. 
    
44. 
    
45. 
    
46. 
    
47. 
    
48. int main(int argc, char *argv[])
    
49. {
    
50.         char *ptr;
    
51.         fd_set readfds;
    
52.         struct timeval tv;
    
53.         char buffer[1024];
    
54.         int sockfd, efd, nbytes, nfds, i;
    
55.         char ch[]     = "GET /opensource/2010/0422/528617.shtml HTTP/1.1\r\nAccept: image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, application/x-shockwave-flash, application/vnd.ms-excel, application/vnd.ms-powerpoint, application/msword, */*\r\nAccept-Language: zh-cn\r\nUser-Agent: Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1)\r\nHost: news.chinaunix.net\r\nConnection: Keep-Alive\r\n\r\n";
    
56.         char server[] = "news.chinaunix.net";
    
57.         struct hostent *he;
    
58.         struct in_addr inadr;
    
59.         struct sockaddr_in dest_addr;
    
60.        
    
61.         long read_incoming, write_incoming, write_outcoming;
    
62. 
    
63. 
    
64.         int fp;
    
65.         if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
    
66.         {
    
67.                 perror("socket");
    
68.                 exit(1);
    
69.         }
    
70.        
    
71.         if (inet_aton(server, &inadr) == 0)
    
72.         {
    
73.                 if ((he = gethostbyname(server)) == NULL)
    
74.                 {
    
75.                         printf("unable to resolve: %s\n", server);
    
76.                         return (-1);
    
77.                 }
    
78. 
    
79.                 memcpy(&inadr.s_addr, he->h_addr_list[0], he->h_length);
    
80.         }
    
81. 
    
82.         dest_addr.sin_family      = AF_INET;
    
83.         dest_addr.sin_port        = htons(DestPort);
    
84. //        dest_addr.sin_addr.s_addr = inet_addr(DestIp);
    
85.        
    
86.         memcpy(&dest_addr.sin_addr, &inadr.s_addr, 4);
    
87. 
    
88.         // Create and setup the connection
    
89.         if (connect(sockfd, (struct sockaddr *)&dest_addr,sizeof(struct sockaddr)) == -1)
    
90.         {
    
91.                 perror("connect");
    
92.                 exit(1);
    
93.         }
    
94. 
    
95.         set_nonblocking(sockfd);
    
96.        
    
97. 
    
98.        
    
99.         nbytes = write(sockfd, ch, strlen(ch));
    
100.         printf("nbytes = %d\n", nbytes);
     
101. 
     
102.         fp = open("./chunked.html", O_WRONLY | O_CREAT | O_TRUNC, 0644);
     
103.        
     
104.         if ( pipe(pipefd) < 0 )
     
105.         {
     
106.                 perror("pipe");
     
107.                 exit(1);
     
108.         }
     
109.        
     
110.         do {
     
111.                 read_incoming = ssplice(sockfd, NULL, pipefd[1], NULL, SPLICE_SIZE, SPLICE_F_NONBLOCK | SPLICE_F_MORE | SPLICE_F_MOVE);
     
112. 
     
113.                 if(read_incoming < 0)
     
114.                 {
     
115.                         if(errno == EINTR || errno == EAGAIN)
     
116.                         {
     
117.                                 usleep(100);
     
118.                                 continue;
     
119.                         }
     
120.                         error("1splice");
     
121.                         break;
     
122.                 }
     
123.                 else if(read_incoming == 0)
     
124.                 {
     
125.                         break;
     
126.                 }
     
127. 
     
128.                 write_outcoming = read_incoming;
     
129.                 while(write_outcoming > 0)
     
130.                 {
     
131. 
     
132.                         write_incoming = splice(pipefd[0], NULL, fp, NULL, write_outcoming, SPLICE_F_NONBLOCK | SPLICE_F_MORE | SPLICE_F_MOVE);
     
133.                         if(write_incoming < 0)
     
134.                         {
     
135.                                 if(write_incoming == -EAGAIN)
     
136.                                 {
     
137.                                         continue;
     
138.                                 }
     
139.                                 error("2splice");
     
140.                                 break;
     
141.                         }
     
142. 
     
143.                         write_outcoming -= write_incoming;
     
144.                 }
     
145.         } while (1);
     
146. 
     
147. 
     
148.         close(fp);
     
149. 
     
150.         close(sockfd);
     
151. }
     

复制代码

这个程序 作为http客户端,直接把socket数据 存储到磁盘中去。
在我的Ubuntu 就正常。

在CentOS 就报错1splice: Invalid argument

不知道什么原因? 各位大哥帮我看看

CentOS
Linux version 2.6.18-128.7.1.el5 (gcc version 4.1.2 20080704 (Red Hat 4.1.2-44)) #1 SMP Mon Aug 24 08:21:56 EDT 2009

Ubuntu

Linux version 2.6.31-14-generic-pae (buildd@rothera) (gcc version 4.4.1 (Ubuntu 4.4.1-4ubuntu ) #48-Ubuntu SMP Fri Oct 16 15:22:42 UTC 2009

没本

#if defined(__i386__)
#define __NR_sys_splice                313
#define __NR_sys_tee                315
#define __NR_sys_vmsplice        316
这个不见得适用于Kernel 2.6.18吧。而且sys_splice的参数个数是不是改过你得查查。
上面的预定义查CentOS的内核源码，有但是数字不同的话用上对应的，没有那就是不支持了。

我要思考

1. 
   
2. /*
   
3. * Splice stdin to net
   
4. */
   
5. #include <stdio.h>
   
6. #include <stdlib.h>
   
7. #include <unistd.h>
   
8. #include <netdb.h>
   
9. #include <unistd.h>
   
10. #include <fcntl.h>
    
11. #include <signal.h>
    
12. #include <netinet/in.h>
    
13. #include <arpa/inet.h>
    
14. #include <string.h>
    
15. #include <sys/time.h>
    
16. #include <errno.h>
    
17. 
    
18. #include "splice.h"
    
19. 
    
20. static int usage(char *name)
    
21. {
    
22.         fprintf(stderr, "%s: target port\n", name);
    
23.         return 1;
    
24. }
    
25. 
    
26. int main(int argc, char *argv[])
    
27. {
    
28.         struct sockaddr_in addr;
    
29.         unsigned short port;
    
30.         int fd, ret;
    
31. 
    
32.         if (argc < 3)
    
33.                 return usage(argv[0]);
    
34. 
    
35.         if (check_input_pipe())
    
36.                 return usage(argv[0]);
    
37. 
    
38.         port = atoi(argv[2]);
    
39. 
    
40.         memset(&addr, 0, sizeof(addr));
    
41.         addr.sin_family = AF_INET;
    
42.         addr.sin_port = htons(port);
    
43. 
    
44.         if (inet_aton(argv[1], &addr.sin_addr) != 1) {
    
45.                 struct hostent *hent = gethostbyname(argv[1]);
    
46. 
    
47.                 if (!hent)
    
48.                         return error("gethostbyname");
    
49. 
    
50.                 memcpy(&addr.sin_addr, hent->h_addr, 4);
    
51.         }
    
52. 
    
53.         printf("Connecting to %s/%d\n", argv[1], port);
    
54. 
    
55.         fd = socket(AF_INET, SOCK_STREAM, 0);
    
56.         if (fd < 0)
    
57.                 return error("socket");
    
58. 
    
59.         if (connect(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0)
    
60.                 return error("connect");
    
61. 
    
62.         do {
    
63.                 ret = ssplice(STDIN_FILENO, NULL, fd, NULL, SPLICE_SIZE, SPLICE_F_NONBLOCK);
    
64.                 if (ret < 0) {
    
65.                         if (errno == EAGAIN) {
    
66.                                 usleep(100);
    
67.                                 continue;
    
68.                         }
    
69.                         return error("splice");
    
70.                 } else if (!ret)
    
71.                         break;
    
72.         } while (1);
    
73. 
    
74.         close(fd);
    
75.         return 0;
    
76. }
    
77. 
    

复制代码

我用这个代码,就可以在CentOS正常运行。这个是 从本地到socket

没本

那你把标志参数一个个去掉看是哪个标志不支持吧。你本地不是只用了一个吗。SPLICE_F_NONBLOCK。或者升级CentOS的内核，可以试试把Ubuntu的内核复制过去。

我要思考

我把errno 打印出来
报 errno  EINVAL 错误
我查man

EINVAL
Target file system doesn't support splicing; neither of the descriptors refers to a pipe; or offset given for non-seekable device.