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