1185493Speter#include <sys/cdefs.h>
2185493Speter__FBSDID("$FreeBSD$");
3185493Speter
4186512Srwatson#include <sys/param.h>
5185493Speter#include <sys/user.h>
6185493Speter#include <sys/sysctl.h>
7185493Speter#include <stdlib.h>
8185493Speter#include <string.h>
9185493Speter
10185493Speter#include "libutil.h"
11185493Speter
12185493Speterstruct kinfo_file *
13185493Speterkinfo_getfile(pid_t pid, int *cntp)
14185493Speter{
15185493Speter	int mib[4];
16185493Speter	int error;
17185493Speter	int cnt;
18185493Speter	size_t len;
19185493Speter	char *buf, *bp, *eb;
20185493Speter	struct kinfo_file *kif, *kp, *kf;
21185493Speter
22186314Smarcus	*cntp = 0;
23185493Speter	len = 0;
24185493Speter	mib[0] = CTL_KERN;
25185493Speter	mib[1] = KERN_PROC;
26185493Speter	mib[2] = KERN_PROC_FILEDESC;
27185493Speter	mib[3] = pid;
28185493Speter
29185493Speter	error = sysctl(mib, 4, NULL, &len, NULL, 0);
30185493Speter	if (error)
31186314Smarcus		return (NULL);
32185493Speter	len = len * 4 / 3;
33185493Speter	buf = malloc(len);
34185493Speter	if (buf == NULL)
35186314Smarcus		return (NULL);
36185493Speter	error = sysctl(mib, 4, buf, &len, NULL, 0);
37185493Speter	if (error) {
38185493Speter		free(buf);
39186314Smarcus		return (NULL);
40185493Speter	}
41185493Speter	/* Pass 1: count items */
42185493Speter	cnt = 0;
43185493Speter	bp = buf;
44185493Speter	eb = buf + len;
45185493Speter	while (bp < eb) {
46185553Speter		kf = (struct kinfo_file *)(uintptr_t)bp;
47185493Speter		bp += kf->kf_structsize;
48185493Speter		cnt++;
49185493Speter	}
50185493Speter
51185493Speter	kif = calloc(cnt, sizeof(*kif));
52185493Speter	if (kif == NULL) {
53185493Speter		free(buf);
54186314Smarcus		return (NULL);
55185493Speter	}
56185493Speter	bp = buf;
57185493Speter	eb = buf + len;
58185493Speter	kp = kif;
59185493Speter	/* Pass 2: unpack */
60185493Speter	while (bp < eb) {
61185553Speter		kf = (struct kinfo_file *)(uintptr_t)bp;
62185493Speter		/* Copy/expand into pre-zeroed buffer */
63185493Speter		memcpy(kp, kf, kf->kf_structsize);
64185493Speter		/* Advance to next packed record */
65185493Speter		bp += kf->kf_structsize;
66185493Speter		/* Set field size to fixed length, advance */
67185493Speter		kp->kf_structsize = sizeof(*kp);
68185493Speter		kp++;
69185493Speter	}
70185493Speter	free(buf);
71185493Speter	*cntp = cnt;
72185493Speter	return (kif);	/* Caller must free() return value */
73185493Speter}
74