summaryrefslogtreecommitdiff
path: root/fd.c
blob: 0f7028f0b12caa44c0e89839adedb9ffdb7dbcc9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include "types.h"
#include "param.h"
#include "x86.h"
#include "mmu.h"
#include "proc.h"
#include "defs.h"
#include "fd.h"
#include "spinlock.h"

struct spinlock fd_table_lock;

struct fd fds[NFD];

/*
 * allocate a file descriptor number for curproc.
 */
int
fd_ualloc()
{
  int fd;
  struct proc *p = curproc[cpu()];
  for(fd = 0; fd < NOFILE; fd++)
    if(p->fds[fd] == 0)
      return fd;
  return -1;
}

/*
 * allocate a file descriptor structure
 */
struct fd *
fd_alloc()
{
  int i;

  acquire(&fd_table_lock);
  for(i = 0; i < NFD; i++){
    if(fds[i].type == FD_CLOSED){
      fds[i].type = FD_NONE;
      fds[i].count = 1;
      release(&fd_table_lock);
      return fds + i;
    }
  }
  release(&fd_table_lock);
  return 0;
}

/*
 * addr is a kernel address, pointing into some process's p->mem.
 */
int
fd_write(struct fd *fd, char *addr, int n)
{
  if(fd->writeable == 0)
    return -1;
  if(fd->type == FD_PIPE){
    return pipe_write(fd->pipe, addr, n);
  } else {
    panic("fd_write");
    return -1;
  }
}

int
fd_read(struct fd *fd, char *addr, int n)
{
  if(fd->readable == 0)
    return -1;
  if(fd->type == FD_PIPE){
    return pipe_read(fd->pipe, addr, n);
  } else {
    panic("fd_read");
    return -1;
  }
}

void
fd_close(struct fd *fd)
{
  acquire(&fd_table_lock);

  if(fd->count < 1 || fd->type == FD_CLOSED)
    panic("fd_close");

  fd->count -= 1;

  if(fd->count == 0){
    if(fd->type == FD_PIPE){
      pipe_close(fd->pipe, fd->writeable);
    } else {
      panic("fd_close");
    }
    fd->count = 0;
    fd->type = FD_CLOSED;
  }
  
  release(&fd_table_lock);
}

void
fd_reference(struct fd *fd)
{
  acquire(&fd_table_lock);
  if(fd->count < 1 || fd->type == FD_CLOSED)
    panic("fd_reference");
  fd->count += 1;
  release(&fd_table_lock);
}