Linux RPC测试

swordhui

RPC是glibc提供的函数参数/返回值封装服务, 并将封装结果通过网络传到服务器.

RPC服务端首先要启动portmapper服务.

测试一个简单的RPC传输示例, 先定义一个模板文件test.x

1. program TESTPROG{
   
2.         version VERSION{
   
3.                 int int_echo(int)=1;
   
4.                 int get_str_len(string)=2;
   
5.                 int add ( int, int ) = 3;
   
6.         }=1;
   
7. }=30000;
   

复制代码

内含3个函数, 注意其中一个有2个参数.

然后可以用rpcgen生成一个Makefile:

1. rpcgen -a -N test.x

复制代码

这会生成Makefile, 客户端和服务端的程序, 和函数示例.

我们手工修改一下Makefile

1. # This is a template Makefile generated by rpcgen
   
2. 
   
3. # Parameters
   
4. 
   
5. CLIENT = test_client
   
6. SERVER = test_server
   
7. 
   
8. SOURCES_CLNT.c =
   
9. SOURCES_CLNT.h =
   
10. SOURCES_SVC.c =
    
11. SOURCES_SVC.h =
    
12. SOURCES.x = test.x
    
13. 
    
14. TARGETS_SVC.c = test_svc.c test_server.c test_xdr.c
    
15. TARGETS_CLNT.c = test_clnt.c test_client.c test_xdr.c
    
16. TARGETS = test.h test_xdr.c test_clnt.c test_svc.c
    
17. 
    
18. OBJECTS_CLNT = $(SOURCES_CLNT.c:%.c=%.o) $(TARGETS_CLNT.c:%.c=%.o)
    
19. OBJECTS_SVC = $(SOURCES_SVC.c:%.c=%.o) $(TARGETS_SVC.c:%.c=%.o)
    
20. # Compiler flags
    
21. 
    
22. CFLAGS += -g -pipe
    
23. LDLIBS += -lnsl
    
24. RPCGENFLAGS = -N
    
25. 
    
26. # Targets
    
27. 
    
28. all : $(CLIENT) $(SERVER)
    
29. 
    
30. $(TARGETS) : $(SOURCES.x)
    
31.         rpcgen $(RPCGENFLAGS) $(SOURCES.x)
    
32. 
    
33. $(OBJECTS_CLNT) : $(SOURCES_CLNT.c) $(SOURCES_CLNT.h) $(TARGETS_CLNT.c)
    
34. 
    
35. $(OBJECTS_SVC) : $(SOURCES_SVC.c) $(SOURCES_SVC.h) $(TARGETS_SVC.c)
    
36. 
    
37. $(CLIENT) : $(OBJECTS_CLNT)
    
38.         $(LINK.c) -o $(CLIENT) $(OBJECTS_CLNT) $(LDLIBS)
    
39. 
    
40. $(SERVER) : $(OBJECTS_SVC)
    
41.         $(LINK.c) -o $(SERVER) $(OBJECTS_SVC) $(LDLIBS)
    
42. 
    
43. clean:
    
44.          $(RM) core $(TARGETS) $(OBJECTS_CLNT) $(OBJECTS_SVC) $(CLIENT) $(SERVER)
    

复制代码

修改test_server.c服务端的处理函数, 提供3种服务:

1. /*
   
2. * This is sample code generated by rpcgen.
   
3. * These are only templates and you can use them
   
4. * as a guideline for developing your own functions.
   
5. */
   
6. 
   
7. #include "test.h"
   
8. 
   
9. int *
   
10. int_echo_1_svc(int arg1,  struct svc_req *rqstp)
    
11. {
    
12.         static int  result;
    
13. 
    
14.         //echo.
    
15.         result=arg1;
    
16.         printf("[RPC1] source=%d, echo=%d\n", arg1, result);
    
17. 
    
18.         return &result;
    
19. }
    
20. 
    
21. int *
    
22. get_str_len_1_svc(char *arg1,  struct svc_req *rqstp)
    
23. {
    
24.         static int  result;
    
25. 
    
26.         //get strlen.
    
27.         result=strlen(arg1);
    
28.         printf("[PRC2] str=%s, len=%d\n", arg1, result);
    
29.         return &result;
    
30. }
    
31. 
    
32. int *
    
33. add_1_svc(int arg1, int arg2,  struct svc_req *rqstp)
    
34. {
    
35.         static int  result;
    
36. 
    
37.         result=arg1+arg2;
    
38.         printf("[RPC3] %d+%d=%d\n", arg1, arg2, result);
    
39.         return &result;
    
40. }

复制代码

客户端test_client.c, 调用这三种服务:

1. /*
   
2. * This is sample code generated by rpcgen.
   
3. * These are only templates and you can use them
   
4. * as a guideline for developing your own functions.
   
5. */
   
6. 
   
7. #include "test.h"
   
8. 
   
9. 
   
10. void
    
11. testprog_1(char *host)
    
12. {
    
13.         CLIENT *clnt;
    
14.         int  *result_1;
    
15.         int int_echo_1_arg1=55;
    
16.         int  *result_2;
    
17.         char *get_str_len_1_arg1="Hello, world";
    
18.         int  *result_3;
    
19.         int add_1_arg1=10;
    
20.         int add_1_arg2=20;
    
21. 
    
22.         clnt = clnt_create (host, TESTPROG, VERSION, "udp");
    
23.         if (clnt == NULL) {
    
24.                 clnt_pcreateerror (host);
    
25.                 exit (1);
    
26.         }
    
27. 
    
28.         result_1 = int_echo_1(int_echo_1_arg1, clnt);
    
29.         if (result_1 == (int *) NULL) {
    
30.                 clnt_perror (clnt, "call failed");
    
31.         }
    
32.         else
    
33.                 printf("[PRC1] echo %d, source %d\n", *result_1,
    
34.                         int_echo_1_arg1);
    
35. 
    
36.         result_2 = get_str_len_1(get_str_len_1_arg1, clnt);
    
37.         if (result_2 == (int *) NULL) {
    
38.                 clnt_perror (clnt, "call failed");
    
39.         }
    
40.         else
    
41.                 printf("[RPC2] return %d, should %d\n", *result_2,
    
42.                         strlen(get_str_len_1_arg1));
    
43.         result_3 = add_1(add_1_arg1, add_1_arg2, clnt);
    
44.         if (result_3 == (int *) NULL) {
    
45.                 clnt_perror (clnt, "call failed");
    
46.         }
    
47.         else
    
48.                 printf("[PRC3] %d+%d=%d\n", add_1_arg1, add_1_arg2,
    
49.                         *result_3);
    
50.         clnt_destroy (clnt);
    
51. }

复制代码

OK, 可以调用make了.
生成可执行程序test_server和test_client.

我们启动./test_server, 用rpcinfo看看:

$rpcinfo -p 127.0.0.1
   program vers proto   port
    100000    2   tcp    111  portmapper
     30000    1   udp  36307
     30000    1   tcp  34883

Bingo! 启动成功.
再开个终端, 尝试一下调用.

./test_client 127.0.0.1
[PRC1] echo 55, source 55
[RPC2] return 12, should 12
[PRC3] 10+20=30

正是我们期望的.

------------------------------------------
原创作品, 张力辉 <swordhui@263.net>
------------------------------------------

realtang

swordhui兄好！
是我给加的精华。顺便问些背景知识，RPC是如何确保安全性有一套机制吗？

聚焦深空

这么懒的方法，好！

swordhui

Post by realtang;2030252
swordhui兄好！
是我给加的精华。顺便问些背景知识，RPC是如何确保安全性有一套机制吗？

好久不来了，今天偶然发现居然留了篇精，谢谢:flash:

我对RPC安全性了解不深，有空翻翻glibc的源码，看看对rpc安全性方面做了什么工作。