socket.h

 
#ifndef KERNEL_SOCKET_H
#define KERNEL_SOCKET_H
 
#include <kernel/net.h>
#include <kernel/spinlock.h>
#include <kernel/vfs.h>
 
#include <libalgo/queue.h>
#include <utils/container_of.h>
 
struct packet;
 
enum socket_state {
    SOCKET_DISCONNECTED, 
    SOCKET_CONNECTED,    
};
 
struct socket {
    struct file *file;                   
    const struct socket_protocol *proto; 
    enum socket_state state;             
    spinlock_t lock;    
    void *data;         
    queue_t rx_packets; 
    spinlock_t rx_lock; 
};
 
static inline bool socket_mode_is_connection(enum socket_type socket_type)
{
    return socket_type == SOCK_STREAM;
}
 
static inline void socket_lock(struct socket *socket)
{
    spinlock_acquire(&socket->lock);
}
 
static inline void socket_unlock(struct socket *socket)
{
    spinlock_release(&socket->lock);
}
 
struct socket_node {
    struct socket socket; 
    struct vnode vnode;   
};
 
static inline struct socket *socket_from_vnode(struct vnode *vnode)
{
    struct socket_node *socket_node = container_of(vnode, struct socket_node,
                                                   vnode);
    return &socket_node->socket;
}
 
static inline struct vnode *socket_node(struct socket *socket)
{
    struct socket_node *socket_node = container_of(socket, struct socket_node,
                                                   socket);
    return &socket_node->vnode;
}
 
struct socket *socket_alloc(void);
 
error_t socket_init(struct socket *socket, int domain, int type, int proto);
 
error_t socket_enqueue_packet(struct socket *socket, struct packet *packet);
 
struct packet *socket_dequeue_packet(struct socket *socket);
 
struct socket_domain {
    enum communication_domain domain; 
    node_t this;                      
    error_t (*socket_init)(struct socket *, int type, int proto);
};
 
error_t socket_domain_register(struct socket_domain *);
 
struct socket_protocol_ops {
    error_t (*init)(struct socket *);
    error_t (*bind)(struct socket *, struct sockaddr *addr, socklen_t addrlen);
    error_t (*connect)(struct socket *, struct sockaddr *addr,
                       socklen_t addrlen);
    ssize_t (*sendmsg)(struct socket *, const struct msghdr *, int flags);
    ssize_t (*recvmsg)(struct socket *, struct msghdr *, int flags);
};
 
struct socket_protocol {
    int proto;                             
    enum socket_type type;                 
    const struct socket_protocol_ops *ops; 
};
 
#endif /* KERNEL_SOCKET_H */