Home | History | Annotate | Download | only in tirpc_expertlevel_rpcb_rmtcall
      1 /*
      2 * Copyright (c) Bull S.A.  2007 All Rights Reserved.
      3 *
      4 * This program is free software; you can redistribute it and/or modify it
      5 * under the terms of version 2 of the GNU General Public License as
      6 * published by the Free Software Foundation.
      7 *
      8 * This program is distributed in the hope that it would be useful, but
      9 * WITHOUT ANY WARRANTY; without even the implied warranty of
     10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
     11 *
     12 * Further, this software is distributed without any warranty that it is
     13 * free of the rightful claim of any third person regarding infringement
     14 * or the like.  Any license provided herein, whether implied or
     15 * otherwise, applies only to this software file.  Patent licenses, if
     16 * any, provided herein do not apply to combinations of this program with
     17 * other software, or any other product whatsoever.
     18 *
     19 * You should have received a copy of the GNU General Public License along
     20 * with this program; if not, write the Free Software Foundation, Inc.,
     21 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
     22 *
     23 * History:
     24 * Created by: Cyril Lacabanne (Cyril.Lacabanne (at) bull.net)
     25 *
     26 */
     27 
     28 #include <stdio.h>
     29 #include <stdlib.h>
     30 #include <time.h>
     31 #include <pthread.h>
     32 #include <tirpc/netconfig.h>
     33 #include <tirpc/rpc/rpc.h>
     34 #include <tirpc/rpc/types.h>
     35 #include <tirpc/rpc/xdr.h>
     36 #include <tirpc/rpc/svc.h>
     37 #include <errno.h>
     38 
     39 //Standard define
     40 #define CALCTHREADPROC	1000
     41 #define VERSNUM 1
     42 
     43 #define ADDRBUFSIZE 100
     44 
     45 static int *thread_array_result;
     46 int run_mode;
     47 int progNum;
     48 char *hostname;
     49 char *nettype;
     50 
     51 struct RES {
     52 	double locRes;
     53 	double svcRes;
     54 };
     55 
     56 struct RES *resTbl;
     57 
     58 struct datas {
     59 	double a;
     60 	double b;
     61 	double c;
     62 };
     63 
     64 bool_t xdr_datas(XDR * pt_xdr, struct datas *pt)
     65 {
     66 	return (xdr_double(pt_xdr, &(pt->a)) &&
     67 		xdr_double(pt_xdr, &(pt->b)) && xdr_double(pt_xdr, &(pt->c)));
     68 }
     69 
     70 double getRand(void)
     71 {
     72 	return (drand48() * 1000);
     73 }
     74 
     75 void *my_thread_process(void *arg)
     76 {
     77 	struct datas vars;
     78 	struct timeval total_timeout;
     79 	struct netconfig *nconf = NULL;
     80 	struct netbuf svcaddr;
     81 	char addrbuf[ADDRBUFSIZE];
     82 
     83 	total_timeout.tv_sec = 1;
     84 	total_timeout.tv_usec = 1;
     85 
     86 	nconf = getnetconfigent("udp");
     87 
     88 	if ((struct netconfig *)nconf == NULL) {
     89 		//Test failed
     90 		printf("5\n");
     91 		pthread_exit((void*)5l);
     92 	}
     93 
     94 	svcaddr.len = 0;
     95 	svcaddr.maxlen = ADDRBUFSIZE;
     96 	svcaddr.buf = addrbuf;
     97 
     98 	if (svcaddr.buf == NULL) {
     99 		printf("5\n");
    100 		pthread_exit((void*)5l);
    101 	}
    102 
    103 	if (!rpcb_getaddr(progNum, VERSNUM, nconf, &svcaddr, hostname)) {
    104 		fprintf(stderr, "rpcb_getaddr failed!!\n");
    105 		printf("5\n");
    106 		pthread_exit((void*)5l);
    107 	}
    108 
    109 	if (run_mode == 1) {
    110 		fprintf(stderr, "Thread %ld\n", (long)arg);
    111 	}
    112 
    113 	vars.a = getRand();
    114 	vars.b = getRand();
    115 	vars.c = getRand();
    116 
    117 	resTbl[atoi(arg)].locRes = vars.a + (vars.b * vars.c);
    118 
    119 	rpcb_rmtcall(nconf, hostname, progNum, VERSNUM, CALCTHREADPROC,
    120 		     (xdrproc_t) xdr_datas, (char *)&vars,
    121 		     (xdrproc_t) xdr_double, (char *)&resTbl[atoi(arg)].svcRes,
    122 		     total_timeout, &svcaddr);
    123 
    124 	thread_array_result[atoi(arg)] =
    125 	    (resTbl[atoi(arg)].svcRes == resTbl[atoi(arg)].locRes) ? 0 : 1;
    126 
    127 	if (run_mode == 1) {
    128 		fprintf(stderr, "Thread #%d calc : %lf, received : %lf\n",
    129 			atoi(arg), resTbl[atoi(arg)].locRes,
    130 			resTbl[atoi(arg)].svcRes);
    131 	}
    132 
    133 	pthread_exit(0);
    134 }
    135 
    136 int main(int argn, char *argc[])
    137 {
    138 	//Program parameters : argc[1] : HostName or Host IP
    139 	//                                         argc[2] : Server Program Number
    140 	//                                         argc[3] : Number of threads
    141 	//                                         other arguments depend on test case
    142 
    143 	//run_mode can switch into stand alone program or program launch by shell script
    144 	//1 : stand alone, debug mode, more screen information
    145 	//0 : launch by shell script as test case, only one printf -> result status
    146 	run_mode = 0;
    147 	int test_status = 0;	//Default test result set to FAILED
    148 	int threadNb = atoi(argc[3]);
    149 	long i;
    150 	pthread_t *pThreadArray;
    151 	void *ret;
    152 
    153 	hostname = argc[1];
    154 	nettype = "VISIBLE";
    155 
    156 	resTbl = malloc(threadNb * sizeof(struct RES));
    157 
    158 	progNum = atoi(argc[2]);
    159 
    160 	if (run_mode == 1) {
    161 		printf("Server #%d\n", progNum);
    162 		printf("Thread to create %d\n", threadNb);
    163 	}
    164 	//Initialization : create threads results array, init elements to 0
    165 	//Each thread will put function result (pas/fail) into array
    166 	thread_array_result = malloc(threadNb * sizeof(int));
    167 	memset(&thread_array_result[0], 0, threadNb * sizeof(int));
    168 
    169 	//Create all threads
    170 	//Run all threads
    171 	pThreadArray = malloc(threadNb * sizeof(pthread_t));
    172 	for (i = 0; i < threadNb; i++) {
    173 		if (run_mode == 1)
    174 			fprintf(stderr, "Try to create thread %ld\n", i);
    175 		if (pthread_create(&pThreadArray[i], NULL, my_thread_process, (void*)i)
    176 		    < 0) {
    177 			fprintf(stderr, "pthread_create error for thread 1\n");
    178 			exit(1);
    179 		}
    180 	}
    181 
    182 	//Clean threads
    183 	for (i = 0; i < threadNb; i++) {
    184 		(void)pthread_join(pThreadArray[i], &ret);
    185 	}
    186 
    187 	//Check if all threads results are ok
    188 	test_status = 0;
    189 	for (i = 0; i < threadNb; i++) {
    190 		if (thread_array_result[i] != 0) {
    191 			test_status = 1;
    192 			break;
    193 		}
    194 	}
    195 
    196 	if (run_mode == 1) {
    197 		for (i = 0; i < threadNb; i++) {
    198 			fprintf(stderr, "Result[%ld]=%d\n", i,
    199 				thread_array_result[i]);
    200 		}
    201 	}
    202 	//This last printf gives the result status to the tests suite
    203 	//normally should be 0: test has passed or 1: test has failed
    204 	printf("%d\n", test_status);
    205 
    206 	return test_status;
    207 }
    208