mirror of
https://github.com/Mathewnd/Astral.git
synced 2026-10-29 21:58:20 +00:00
Matheus
120 lines
2.2 KiB
C
120 lines
2.2 KiB
C
#include <kernel/syscalls.h>
|
|
#include <kernel/socket.h>
|
|
#include <kernel/fd.h>
|
|
#include <arch/cls.h>
|
|
#include <arch/spinlock.h>
|
|
#include <kernel/ustring.h>
|
|
|
|
extern fs_t kerneltmpfs;
|
|
|
|
syscallret syscall_accept(int sockfd, void* addr, socklen_t* addrlen){
|
|
|
|
syscallret retv;
|
|
retv.ret = -1;
|
|
|
|
if(addr > USER_SPACE_END){
|
|
retv.errno = EFAULT;
|
|
return retv;
|
|
}
|
|
|
|
proc_t* proc = arch_getcls()->thread->proc;
|
|
|
|
// access sockfd
|
|
|
|
fd_t* fd;
|
|
|
|
int err = fd_access(&proc->fdtable, &fd, sockfd);
|
|
|
|
if(err){
|
|
retv.errno = err;
|
|
return retv;
|
|
}
|
|
|
|
if(GETTYPE(fd->mode) != TYPE_SOCKET){
|
|
retv.errno = ENOTSOCK;
|
|
goto _fail;
|
|
}
|
|
|
|
socket_t* listenersock = fd->node->objdata;
|
|
|
|
// create new fd
|
|
|
|
fd_t* connfd = NULL;
|
|
int connifd;
|
|
|
|
retv.errno = fd_alloc(&proc->fdtable, &connfd, &connifd, 0);
|
|
|
|
if(retv.errno)
|
|
goto _fail;
|
|
|
|
connfd->node = vfs_newnode("SOCKET", &kerneltmpfs, NULL);
|
|
if(!connfd->node){
|
|
retv.errno = ENOMEM;
|
|
goto _fail;
|
|
}
|
|
connfd->mode = 0777 | MAKETYPE(TYPE_SOCKET);
|
|
connfd->node->refcount = 1;
|
|
connfd->node->st.st_mode = connfd->mode;
|
|
|
|
// create socket for new fd
|
|
|
|
socket_t* connsock;
|
|
|
|
retv.errno = socket_new(&connsock, listenersock->family, listenersock->type, listenersock->protocol);
|
|
|
|
// get peer
|
|
|
|
socket_t* peer;
|
|
|
|
retv.errno = socket_accept(&peer, listenersock, addr, addrlen, fd);
|
|
|
|
if(retv.errno)
|
|
goto _fail;
|
|
|
|
|
|
retv.errno = u_memcpy(addr, &peer->addr_un, peer->addrlen);
|
|
|
|
if(retv.errno)
|
|
goto _fail;
|
|
|
|
retv.errno = u_memcpy(addrlen, &peer->addrlen, sizeof(socklen_t));
|
|
|
|
if(retv.errno)
|
|
goto _fail;
|
|
|
|
connsock->peer = peer;
|
|
connsock->state = SOCKET_STATE_CONNECTED;
|
|
|
|
spinlock_acquire(&peer->lock);
|
|
|
|
peer->peer = connsock;
|
|
|
|
peer->state = SOCKET_STATE_CONNECTED;
|
|
event_signal(&peer->acceptevent, true);
|
|
|
|
connfd->node->objdata = connsock;
|
|
|
|
|
|
spinlock_release(&peer->lock);
|
|
|
|
fd_release(fd);
|
|
fd_release(connfd);
|
|
|
|
retv.ret = connifd;
|
|
|
|
return retv;
|
|
|
|
_fail:
|
|
|
|
fd_release(fd);
|
|
|
|
if(connfd){
|
|
fd_release(connfd);
|
|
fd_free(&proc->fdtable, connifd);
|
|
}
|
|
|
|
return retv;
|
|
|
|
|
|
}
|