mirror of
https://github.com/Mathewnd/Astral.git
synced 2026-10-27 12:48:22 +00:00
Matheus
128 lines
3.3 KiB
C
128 lines
3.3 KiB
C
#ifndef _SOCK_H
|
|
#define _SOCK_H
|
|
|
|
#include <kernel/net.h>
|
|
#include <kernel/vfs.h>
|
|
#include <mutex.h>
|
|
#include <kernel/poll.h>
|
|
|
|
#define SOCKET_STATE_UNBOUND 0
|
|
#define SOCKET_STATE_BOUND 1
|
|
#define SOCKET_STATE_CONNECTED 2
|
|
#define SOCKET_STATE_LISTENING 3
|
|
|
|
typedef struct {
|
|
struct socketops_t *ops;
|
|
mutex_t mutex;
|
|
int state;
|
|
int broadcast;
|
|
netdev_t *netdev;
|
|
pollheader_t pollheader;
|
|
int type;
|
|
} socket_t;
|
|
|
|
typedef struct {
|
|
union {
|
|
ipv4addr_t ipv4addr;
|
|
char path[256];
|
|
};
|
|
} sockaddr_t;
|
|
|
|
#define SOCKET_RECV_FLAGS_PEEK 0x8000000000000000l
|
|
typedef struct socketops_t {
|
|
int (*bind)(socket_t *socket, sockaddr_t *addr);
|
|
int (*send)(socket_t *socket, sockaddr_t *addr, void *buffer, size_t count, uintmax_t flags, size_t *sendcount);
|
|
int (*recv)(socket_t *socket, sockaddr_t *addr, void *buffer, size_t count, uintmax_t flags, size_t *recvcount);
|
|
int (*poll)(socket_t *socket, polldata_t *data, int events);
|
|
int (*connect)(socket_t *socket, sockaddr_t *addr, uintmax_t data);
|
|
int (*listen)(socket_t *socket, int backlog);
|
|
int (*accept)(socket_t *server, socket_t *client, sockaddr_t *addr, uintmax_t flags);
|
|
size_t (*datacount)(socket_t *socket);
|
|
void (*destroy)(socket_t *socket);
|
|
} socketops_t;
|
|
|
|
typedef struct {
|
|
vnode_t vnode;
|
|
vattr_t attr;
|
|
socket_t *socket;
|
|
} socketnode_t;
|
|
|
|
#define SOCKET_TYPE_UDP 0
|
|
#define SOCKET_TYPE_LOCAL 1
|
|
#define SOCKET_TYPE_TCP 2
|
|
#define SOCKFS_SOCKET_FROM_NODE(nodep) (((socketnode_t *)(nodep))->socket)
|
|
|
|
static inline int sock_convertaddress(sockaddr_t *sockaddr, abisockaddr_t *abisockaddr) {
|
|
switch (abisockaddr->type) {
|
|
case AF_INET:
|
|
inaddr_t *inaddr = (inaddr_t *)abisockaddr;
|
|
sockaddr->ipv4addr.addr = be_to_cpu_d(inaddr->sin_addr);
|
|
sockaddr->ipv4addr.port = be_to_cpu_w(inaddr->sin_port);
|
|
break;
|
|
case AF_LOCAL:
|
|
unaddr_t *unaddr = (unaddr_t *)abisockaddr;
|
|
strcpy(sockaddr->path, unaddr->sun_path);
|
|
break;
|
|
default:
|
|
return EINVAL;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline int sock_addrtoabiaddr(int socktype, sockaddr_t *sockaddr, abisockaddr_t *abisockaddr) {
|
|
switch (socktype) {
|
|
case SOCKET_TYPE_UDP:
|
|
case SOCKET_TYPE_TCP:
|
|
inaddr_t *inaddr = (inaddr_t *)abisockaddr;
|
|
inaddr->sin_addr = cpu_to_be_d(sockaddr->ipv4addr.addr);
|
|
inaddr->sin_port = cpu_to_be_w(sockaddr->ipv4addr.port);
|
|
break;
|
|
case SOCKET_TYPE_LOCAL:
|
|
unaddr_t *unaddr = (unaddr_t *)abisockaddr;
|
|
strcpy(unaddr->sun_path, sockaddr->path);
|
|
break;
|
|
default:
|
|
return EINVAL;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline int sock_copymsghdr(msghdr_t *khdr, msghdr_t *uhdr) {
|
|
memcpy(khdr, uhdr, sizeof(msghdr_t));
|
|
iovec_t *iovectmp = alloc(sizeof(iovec_t) * khdr->iovcount);
|
|
if (iovectmp == NULL)
|
|
return ENOMEM;
|
|
memcpy(iovectmp, khdr->iov, sizeof(iovec_t) * khdr->iovcount);
|
|
khdr->iov = iovectmp;
|
|
|
|
if (khdr->addr) {
|
|
abisockaddr_t *addrtmp = alloc(khdr->addrlen);
|
|
if (addrtmp == NULL) {
|
|
free(iovectmp);
|
|
return ENOMEM;
|
|
}
|
|
memcpy(addrtmp, khdr->addr, khdr->addrlen);
|
|
khdr->addr = addrtmp;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline void sock_freemsghdr(msghdr_t *hdr) {
|
|
free(hdr->iov);
|
|
if (hdr->addr)
|
|
free(hdr->addr);
|
|
}
|
|
|
|
void localsock_leavebinding(vnode_t *vnode);
|
|
socket_t *localsock_createsocket();
|
|
socket_t *udp_createsocket();
|
|
socket_t *tcp_createsocket();
|
|
socket_t *socket_create(int type);
|
|
int sockfs_newsocket(vnode_t **vnodep, socket_t *socket);
|
|
void sockfs_init();
|
|
|
|
#endif
|