1*43a90889SApple OSS Distributions #include <sys/cdefs.h>
2*43a90889SApple OSS Distributions #include <sys/proc_internal.h>
3*43a90889SApple OSS Distributions #include <sys/vnode_internal.h>
4*43a90889SApple OSS Distributions #include <sys/ubc_internal.h>
5*43a90889SApple OSS Distributions #include <sys/file_internal.h>
6*43a90889SApple OSS Distributions #include <sys/vnode.h>
7*43a90889SApple OSS Distributions
8*43a90889SApple OSS Distributions // From vm_parameter_validation_kern.c
9*43a90889SApple OSS Distributions void testprintf(const char *format, ...);
10*43a90889SApple OSS Distributions
11*43a90889SApple OSS Distributions struct file_control_return {
12*43a90889SApple OSS Distributions void * control;
13*43a90889SApple OSS Distributions struct fileproc * fp;
14*43a90889SApple OSS Distributions struct vnode * vp;
15*43a90889SApple OSS Distributions int fd;
16*43a90889SApple OSS Distributions };
17*43a90889SApple OSS Distributions struct file_control_return get_control_from_fd(int fd);
18*43a90889SApple OSS Distributions
19*43a90889SApple OSS Distributions struct file_control_return
get_control_from_fd(int fd)20*43a90889SApple OSS Distributions get_control_from_fd(int fd)
21*43a90889SApple OSS Distributions {
22*43a90889SApple OSS Distributions struct fileproc *fp = NULL;
23*43a90889SApple OSS Distributions struct vnode *vp = NULL;
24*43a90889SApple OSS Distributions size_t file_size;
25*43a90889SApple OSS Distributions off_t fs;
26*43a90889SApple OSS Distributions memory_object_control_t file_control = NULL;
27*43a90889SApple OSS Distributions int error;
28*43a90889SApple OSS Distributions struct file_control_return ret = {NULL, NULL, NULL, fd};
29*43a90889SApple OSS Distributions
30*43a90889SApple OSS Distributions
31*43a90889SApple OSS Distributions proc_t p = current_proc();
32*43a90889SApple OSS Distributions
33*43a90889SApple OSS Distributions /* get file structure from file descriptor */
34*43a90889SApple OSS Distributions error = fp_get_ftype(p, fd, DTYPE_VNODE, EINVAL, &fp);
35*43a90889SApple OSS Distributions if (error) {
36*43a90889SApple OSS Distributions testprintf("%s: [%d(%s)]: fp_get_ftype() failed, error %d\n",
37*43a90889SApple OSS Distributions __func__, proc_getpid(p), p->p_comm, error);
38*43a90889SApple OSS Distributions return ret;
39*43a90889SApple OSS Distributions }
40*43a90889SApple OSS Distributions ret.fp = fp;
41*43a90889SApple OSS Distributions
42*43a90889SApple OSS Distributions /* We need at least read permission on the file */
43*43a90889SApple OSS Distributions if (!(fp->fp_glob->fg_flag & FREAD)) {
44*43a90889SApple OSS Distributions testprintf("%s: [%d(%s)]: not readable\n",
45*43a90889SApple OSS Distributions __func__, proc_getpid(p), p->p_comm);
46*43a90889SApple OSS Distributions return ret;
47*43a90889SApple OSS Distributions }
48*43a90889SApple OSS Distributions
49*43a90889SApple OSS Distributions /* Get the vnode from file structure */
50*43a90889SApple OSS Distributions vp = (struct vnode *)fp_get_data(fp);
51*43a90889SApple OSS Distributions error = vnode_getwithref(vp);
52*43a90889SApple OSS Distributions if (error) {
53*43a90889SApple OSS Distributions testprintf("%s: [%d(%s)]: failed to get vnode, error %d\n",
54*43a90889SApple OSS Distributions __func__, proc_getpid(p), p->p_comm, error);
55*43a90889SApple OSS Distributions return ret;
56*43a90889SApple OSS Distributions }
57*43a90889SApple OSS Distributions ret.vp = vp;
58*43a90889SApple OSS Distributions
59*43a90889SApple OSS Distributions /* Make sure the vnode is a regular file */
60*43a90889SApple OSS Distributions if (vp->v_type != VREG) {
61*43a90889SApple OSS Distributions testprintf("%s: [%d(%s)]: vnode not VREG\n",
62*43a90889SApple OSS Distributions __func__, proc_getpid(p), p->p_comm);
63*43a90889SApple OSS Distributions return ret;
64*43a90889SApple OSS Distributions }
65*43a90889SApple OSS Distributions
66*43a90889SApple OSS Distributions /* get vnode size */
67*43a90889SApple OSS Distributions error = vnode_size(vp, &fs, vfs_context_current());
68*43a90889SApple OSS Distributions if (error) {
69*43a90889SApple OSS Distributions return ret;
70*43a90889SApple OSS Distributions }
71*43a90889SApple OSS Distributions file_size = fs;
72*43a90889SApple OSS Distributions
73*43a90889SApple OSS Distributions /* get the file's memory object handle */
74*43a90889SApple OSS Distributions file_control = ubc_getobject(vp, UBC_HOLDOBJECT);
75*43a90889SApple OSS Distributions if (file_control == MEMORY_OBJECT_CONTROL_NULL) {
76*43a90889SApple OSS Distributions testprintf("%s: [%d(%s)]: no memory object\n",
77*43a90889SApple OSS Distributions __func__, proc_getpid(p), p->p_comm);
78*43a90889SApple OSS Distributions return ret;
79*43a90889SApple OSS Distributions }
80*43a90889SApple OSS Distributions ret.control = file_control;
81*43a90889SApple OSS Distributions
82*43a90889SApple OSS Distributions return ret;
83*43a90889SApple OSS Distributions }
84*43a90889SApple OSS Distributions
85*43a90889SApple OSS Distributions void cleanup_control_related_data(struct file_control_return info);
86*43a90889SApple OSS Distributions
87*43a90889SApple OSS Distributions void
cleanup_control_related_data(struct file_control_return info)88*43a90889SApple OSS Distributions cleanup_control_related_data(struct file_control_return info)
89*43a90889SApple OSS Distributions {
90*43a90889SApple OSS Distributions if (info.fp != NULL) {
91*43a90889SApple OSS Distributions /* release the file descriptor */
92*43a90889SApple OSS Distributions fp_drop(current_proc(), info.fd, info.fp, 0);
93*43a90889SApple OSS Distributions }
94*43a90889SApple OSS Distributions if (info.vp != NULL) {
95*43a90889SApple OSS Distributions (void)vnode_put(info.vp);
96*43a90889SApple OSS Distributions }
97*43a90889SApple OSS Distributions }
98