root/usr.sbin/rpc.statd/test.c
/*      $NetBSD: test.c,v 1.3 2018/01/23 21:06:26 sevan Exp $   */

#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: test.c,v 1.3 2018/01/23 21:06:26 sevan Exp $");
#endif

#include <stdio.h>
#include <rpc/rpc.h>
#include <rpcsvc/sm_inter.h>


/* Default timeout can be changed using clnt_control() */
static struct timeval TIMEOUT = {25, 0};

struct sm_stat_res *
sm_stat_1(struct sm_name *argp, CLIENT *clnt)
{
        static struct sm_stat_res res;

        bzero((char *) &res, sizeof(res));
        if (clnt_call(clnt, SM_STAT, xdr_sm_name, argp, xdr_sm_stat_res,
            &res, TIMEOUT) != RPC_SUCCESS)
                return (NULL);
        return (&res);
}


struct sm_stat_res *
sm_mon_1(struct mon *argp, CLIENT *clnt)
{
        static struct sm_stat_res res;

        bzero((char *) &res, sizeof(res));
        if (clnt_call(clnt, SM_MON, xdr_mon, argp, xdr_sm_stat_res,
            &res, TIMEOUT) != RPC_SUCCESS)
                return (NULL);
        return (&res);
}


struct sm_stat *
sm_unmon_1(struct mon_id *argp, CLIENT *clnt)
{
        static struct sm_stat res;

        bzero((char *) &res, sizeof(res));
        if (clnt_call(clnt, SM_UNMON, xdr_mon_id, argp, xdr_sm_stat,
            &res, TIMEOUT) != RPC_SUCCESS)
                return (NULL);
        return (&res);
}


struct sm_stat *
sm_unmon_all_1(struct my_id *argp, CLIENT *clnt)
{
        static struct sm_stat res;

        bzero((char *) &res, sizeof(res));
        if (clnt_call(clnt, SM_UNMON_ALL, xdr_my_id, argp, xdr_sm_stat,
            &res, TIMEOUT) != RPC_SUCCESS)
                return (NULL);
        return (&res);
}


void *
sm_simu_crash_1(void   *argp, CLIENT *clnt)
{
        static char res;

        bzero((char *) &res, sizeof(res));
        if (clnt_call(clnt, SM_SIMU_CRASH, xdr_void, argp, xdr_void,
            &res, TIMEOUT) != RPC_SUCCESS)
                return (NULL);
        return ((void *) &res);
}


int 
main(int argc, char *argv[])
{
        CLIENT *cli;
        char dummy;
        void *out;
        struct mon mon;

        if (argc < 2) {
                warnx("usage: test {<hostname> | crash}");
                errx(1, "Always talks to statd at localhost");
        }
        printf("Creating client for localhost\n");
        cli = clnt_create("localhost", SM_PROG, SM_VERS, "udp");
        if (!cli) {
                errx(1, "Failed to create client");
        }
        mon.mon_id.mon_name = argv[1];
        mon.mon_id.my_id.my_name = argv[1];
        mon.mon_id.my_id.my_prog = SM_PROG;
        mon.mon_id.my_id.my_vers = SM_VERS;
        mon.mon_id.my_id.my_proc = 1;   /* have it call sm_stat() !!! */

        if (strcmp(argv[1], "crash")) {
                /* Hostname given */
                struct sm_stat_res *res;
                if (res = sm_mon_1(&mon, cli))
                        printf("Success!\n");
                else
                        printf("Fail\n");
        } else {
                if (out = sm_simu_crash_1(&dummy, cli))
                        printf("Success!\n");
                else
                        printf("Fail\n");
        }

        return 0;
}