1/*
2 * by Manuel Bouyer (bouyer@ensta.fr)
3 *
4 * There is no copyright, you can use it as you want.
5 */
6
7#include <sys/cdefs.h>
8__FBSDID("$FreeBSD$");
9
10#include <sys/param.h>
11#include <sys/mount.h>
12#include <sys/file.h>
13#include <sys/stat.h>
14#include <sys/socket.h>
15
16#include <ufs/ufs/quota.h>
17#include <rpc/rpc.h>
18#include <rpcsvc/rquota.h>
19#include <arpa/inet.h>
20#include <netdb.h>
21
22#include <ctype.h>
23#include <errno.h>
24#include <fstab.h>
25#include <grp.h>
26#include <libutil.h>
27#include <pwd.h>
28#include <signal.h>
29#include <stdio.h>
30#include <stdlib.h>
31#include <string.h>
32#include <syslog.h>
33#include <unistd.h>
34
35static void rquota_service(struct svc_req *request, SVCXPRT *transp);
36static void sendquota(struct svc_req *request, SVCXPRT *transp);
37static void initfs(void);
38static int getfsquota(long id, char *path, struct dqblk *dqblk);
39
40static struct quotafile **qfa;	/* array of qfs */
41static int nqf, szqf;		/* number of qfs and size of array */
42static int from_inetd = 1;
43
44static void
45cleanup(int sig)
46{
47
48	(void)sig;
49	(void)rpcb_unset(RQUOTAPROG, RQUOTAVERS, NULL);
50	exit(0);
51}
52
53int
54main(void)
55{
56	SVCXPRT *transp;
57	int ok;
58	struct sockaddr_storage from;
59	socklen_t fromlen;
60
61	fromlen = sizeof(from);
62	if (getsockname(0, (struct sockaddr *)&from, &fromlen) < 0)
63		from_inetd = 0;
64
65	if (!from_inetd) {
66		daemon(0, 0);
67		(void)rpcb_unset(RQUOTAPROG, RQUOTAVERS, NULL);
68		(void)signal(SIGINT, cleanup);
69		(void)signal(SIGTERM, cleanup);
70		(void)signal(SIGHUP, cleanup);
71	}
72
73	openlog("rpc.rquotad", LOG_CONS|LOG_PID, LOG_DAEMON);
74
75	/* create and register the service */
76	if (from_inetd) {
77		transp = svc_tli_create(0, NULL, NULL, 0, 0);
78		if (transp == NULL) {
79			syslog(LOG_ERR, "couldn't create udp service.");
80			exit(1);
81		}
82		ok = svc_reg(transp, RQUOTAPROG, RQUOTAVERS,
83		    rquota_service, NULL);
84	} else {
85		ok = svc_create(rquota_service,
86		    RQUOTAPROG, RQUOTAVERS, "udp");
87	}
88	if (!ok) {
89		syslog(LOG_ERR,
90		    "unable to register (RQUOTAPROG, RQUOTAVERS, %s)",
91		    from_inetd ? "(inetd)" : "udp");
92		exit(1);
93	}
94
95	initfs();
96	svc_run();
97	syslog(LOG_ERR, "svc_run returned");
98	exit(1);
99}
100
101static void
102rquota_service(struct svc_req *request, SVCXPRT *transp)
103{
104
105	switch (request->rq_proc) {
106	case NULLPROC:
107		(void)svc_sendreply(transp, (xdrproc_t)xdr_void, (char *)NULL);
108		break;
109	case RQUOTAPROC_GETQUOTA:
110	case RQUOTAPROC_GETACTIVEQUOTA:
111		sendquota(request, transp);
112		break;
113	default:
114		svcerr_noproc(transp);
115		break;
116	}
117	if (from_inetd)
118		exit(0);
119}
120
121/* read quota for the specified id, and send it */
122static void
123sendquota(struct svc_req *request, SVCXPRT *transp)
124{
125	struct getquota_args getq_args;
126	struct getquota_rslt getq_rslt;
127	struct dqblk dqblk;
128	struct timeval timev;
129	int scale;
130
131	bzero(&getq_args, sizeof(getq_args));
132	if (!svc_getargs(transp, (xdrproc_t)xdr_getquota_args, &getq_args)) {
133		svcerr_decode(transp);
134		return;
135	}
136	if (request->rq_cred.oa_flavor != AUTH_UNIX) {
137		/* bad auth */
138		getq_rslt.status = Q_EPERM;
139	} else if (!getfsquota(getq_args.gqa_uid, getq_args.gqa_pathp, &dqblk)) {
140		/* failed, return noquota */
141		getq_rslt.status = Q_NOQUOTA;
142	} else {
143		gettimeofday(&timev, NULL);
144		getq_rslt.status = Q_OK;
145		getq_rslt.getquota_rslt_u.gqr_rquota.rq_active = TRUE;
146		scale = 1 << flsll(dqblk.dqb_bhardlimit >> 32);
147		getq_rslt.getquota_rslt_u.gqr_rquota.rq_bsize =
148		    DEV_BSIZE * scale;
149		getq_rslt.getquota_rslt_u.gqr_rquota.rq_bhardlimit =
150		    dqblk.dqb_bhardlimit / scale;
151		getq_rslt.getquota_rslt_u.gqr_rquota.rq_bsoftlimit =
152		    dqblk.dqb_bsoftlimit / scale;
153		getq_rslt.getquota_rslt_u.gqr_rquota.rq_curblocks =
154		    dqblk.dqb_curblocks / scale;
155		getq_rslt.getquota_rslt_u.gqr_rquota.rq_fhardlimit =
156		    dqblk.dqb_ihardlimit;
157		getq_rslt.getquota_rslt_u.gqr_rquota.rq_fsoftlimit =
158		    dqblk.dqb_isoftlimit;
159		getq_rslt.getquota_rslt_u.gqr_rquota.rq_curfiles =
160		    dqblk.dqb_curinodes;
161		getq_rslt.getquota_rslt_u.gqr_rquota.rq_btimeleft =
162		    dqblk.dqb_btime - timev.tv_sec;
163		getq_rslt.getquota_rslt_u.gqr_rquota.rq_ftimeleft =
164		    dqblk.dqb_itime - timev.tv_sec;
165	}
166	if (!svc_sendreply(transp, (xdrproc_t)xdr_getquota_rslt, &getq_rslt))
167		svcerr_systemerr(transp);
168	if (!svc_freeargs(transp, (xdrproc_t)xdr_getquota_args, &getq_args)) {
169		syslog(LOG_ERR, "unable to free arguments");
170		exit(1);
171	}
172}
173
174static void
175initfs(void)
176{
177	struct fstab *fs;
178
179	setfsent();
180	szqf = 8;
181	if ((qfa = malloc(szqf * sizeof *qfa)) == NULL)
182		goto enomem;
183	while ((fs = getfsent())) {
184		if (strcmp(fs->fs_vfstype, "ufs"))
185			continue;
186		if (nqf >= szqf) {
187			szqf *= 2;
188			if ((qfa = reallocf(qfa, szqf * sizeof *qfa)) == NULL)
189				goto enomem;
190		}
191		if ((qfa[nqf] = quota_open(fs, USRQUOTA, O_RDONLY)) == NULL) {
192			if (errno != EOPNOTSUPP)
193				goto fserr;
194			continue;
195		}
196		++nqf;
197		/* XXX */
198	}
199	endfsent();
200	return;
201enomem:
202	syslog(LOG_ERR, "out of memory");
203	exit(1);
204fserr:
205	syslog(LOG_ERR, "%s: %s", fs->fs_file, strerror(errno));
206	exit(1);
207}
208
209/*
210 * gets the quotas for id, filesystem path.
211 * Return 0 if fail, 1 otherwise
212 */
213static int
214getfsquota(long id, char *path, struct dqblk *dqblk)
215{
216	int i;
217
218	for (i = 0; i < nqf; ++i)
219		if (quota_check_path(qfa[i], path) == 1)
220			return (quota_read(qfa[i], dqblk, id) == 0);
221	return (0);
222}
223