xv6-cs450/fd.c

133 lines
2.2 KiB
C
Raw Normal View History

2006-06-27 16:35:53 +02:00
#include "types.h"
#include "param.h"
#include "x86.h"
#include "mmu.h"
#include "proc.h"
#include "defs.h"
#include "fd.h"
#include "spinlock.h"
#include "dev.h"
struct spinlock fd_table_lock;
struct devsw devsw[NDEV];
2006-06-27 16:35:53 +02:00
struct fd fds[NFD];
void
fd_init(void)
{
initlock(&fd_table_lock, "fd_table");
}
2006-06-27 16:35:53 +02:00
/*
* allocate a file descriptor number for curproc.
*/
int
2006-07-17 03:25:22 +02:00
fd_ualloc(void)
2006-06-27 16:35:53 +02:00
{
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
*/
2006-06-27 16:35:53 +02:00
struct fd *
2006-07-17 03:25:22 +02:00
fd_alloc(void)
2006-06-27 16:35:53 +02:00
{
int i;
acquire(&fd_table_lock);
2006-06-27 16:35:53 +02:00
for(i = 0; i < NFD; i++){
if(fds[i].type == FD_CLOSED){
fds[i].type = FD_NONE;
2006-07-16 04:04:58 +02:00
fds[i].ref = 1;
release(&fd_table_lock);
2006-06-27 16:35:53 +02:00
return fds + i;
}
}
release(&fd_table_lock);
2006-06-27 16:35:53 +02:00
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 if (fd->type == FD_FILE) {
ilock(fd->ip);
int r = writei (fd->ip, addr, fd->off, n);
if (r > 0) {
fd->off += r;
}
iunlock(fd->ip);
return r;
2006-06-27 16:35:53 +02:00
} 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 if(fd->type == FD_FILE){
ilock(fd->ip);
int cc = readi(fd->ip, addr, fd->off, n);
if(cc > 0)
fd->off += cc;
iunlock(fd->ip);
return cc;
2006-06-27 16:35:53 +02:00
} else {
panic("fd_read");
return -1;
}
}
void
fd_close(struct fd *fd)
{
acquire(&fd_table_lock);
2006-07-16 04:04:58 +02:00
if(fd->ref < 1 || fd->type == FD_CLOSED)
panic("fd_close");
2006-07-16 04:04:58 +02:00
if(--fd->ref == 0){
if(fd->type == FD_PIPE){
pipe_close(fd->pipe, fd->writeable);
2006-07-29 11:35:02 +02:00
} else if(fd->type == FD_FILE){
idecref(fd->ip);
} else {
panic("fd_close");
}
2006-07-16 04:04:58 +02:00
fd->ref = 0;
fd->type = FD_CLOSED;
}
release(&fd_table_lock);
}
void
fd_incref(struct fd *fd)
{
acquire(&fd_table_lock);
2006-07-16 04:04:58 +02:00
if(fd->ref < 1 || fd->type == FD_CLOSED)
panic("fd_incref");
fd->ref++;
release(&fd_table_lock);
}