#ifndef __SOCKET_H__
#define __SOCKET_H__

#include <stddef.h>

// Socket Types
#define SOCK_STREAM 1
#define SOCK_DGRAM 2
#define SOCK_RAW 3
 
// IP protocols
#define IPPROTO_IP 0
#define IPPROTO_ICMP 1
#define IPPROTO_TCP 6
#define IPPROTO_UDP 17
#define IPPROTO_RAW 255
 
// IP options
#define IP_TTL 2
 
// Address families
#define AF_UNSPEC 0
#define AF_LOCAL 1
#define AF_INET4 2     // IPv4
#define AF_INET6 10    // IPv6

#define PF_UNSPEC AF_UNSPEC
#define PF_LOCAL  AF_LOCAL
#define PF_INET4  AF_INET4
#define PF_INET6  AF_INET6
 
// internal definition
#define AI_SUPPORTED 0x40F

// for system function 76
#define API_ETH (0<<16)
#define API_IPv4 (1<<16)
#define API_ICMP (2<<16)
#define API_UDP (3<<16)
#define API_TCP (4<<16)
#define API_ARP (5<<16)
#define API_PPPOE (6<<16)

// Socket flags for user calls
#define MSG_NOFLAG 0
#define MSG_PEEK 0x02
#define MSG_DONTWAIT 0x40
 
// Socket levels
#define SOL_SOCKET 0xffff

//Socket options
#define SO_BINDTODEVICE (1<<9)
#define SO_NONBLOCK (1<<31)

// Error Codes
#define ENOBUFS      1
#define EINPROGRESS  2
#define EOPNOTSUPP   4
#define EWOULDBLOCK  6
#define ENOTCONN     9
#define EALREADY     10
#define EINVALUE     11
#define EMSGSIZE     12
#define ENOMEM       18
#define EADDRINUSE   20
#define ECONNREFUSED 61
#define ECONNRESET   52
#define EISCONN      56
#define ETIMEDOUT    60
#define ECONNABORTED 53


#define PORT(X) (X<<8)
int err_code;

#pragma pack(push,1)
typedef struct{
    unsigned short sin_family;
    unsigned short sin_port; 
    unsigned int sin_addr;
    unsigned long long sin_zero;
}sockaddr; 
#pragma pack(pop)

#pragma pack(push,1)
typedef struct{
  unsigned int level;
  unsigned int optionname;
  unsigned int optlenght;
  unsigned char options;
}optstruct;
#pragma pack(pop)

static inline int socket(int domain, int type, int protocol)
{
    int socket;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(socket)
    :"a"(75), "b"(0), "c"(domain), "d"(type), "S"(protocol)
    );
    return socket;
}

static inline int close(int socket)
{
    int status;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(status)
    :"a"(75), "b"(1), "c"(socket)
    );
    return status;
}
static inline int bind(int socket, const sockaddr *addres, int addres_len)
{
    int status;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(status)
    :"a"(75), "b"(2), "c"(socket), "d"(addres), "S"(addres_len)
    );
    return status;
}

static inline int listen(int socket, int backlog)
{
    int status;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(status)
    :"a"(75), "b"(3), "c"(socket), "d"(backlog)
    );
    return status;
}

static inline int connect(int socket,const sockaddr* address, int socket_len)
{
    int status;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(status)
    :"a"(75), "b"(4), "c"(socket), "d"(address), "S"(socket_len)
    );
    return status;
}

static inline int accept(int socket, const sockaddr *address, int address_len)
{
    int new_socket;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(new_socket)
    :"a"(75), "b"(5), "c"(socket), "d"(address), "S"(address_len)
    );
    return new_socket;
}

static inline int send(int socket, const void *message, size_t msg_len, int flag)
{
    int status;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(status)
    :"a"(75), "b"(6), "c"(socket), "d"(message), "S"(msg_len), "D"(flag)
    );
    return status;
}

static inline int recv(int socket, void *buffer, size_t buff_len, int flag)
{
    int status;
    asm volatile(
    "int $0x40"
    :"=b"(err_code), "=a"(status)
    :"a"(75), "b"(7), "c"(socket), "d"(buffer), "S"(buff_len), "D"(flag)
    );
    return status;
}

static inline int setsockopt(int socket,const optstruct* opt)
{
    int status;
    asm volatile(
        "int $0x40"
        :"=b"(err_code), "=a"(status)
        :"a"(75), "b"(8), "c"(socket),"d"(opt)
    );
    return status;
}

static inline int getsockopt(int socket, optstruct* opt)
{
    int status; 
    asm volatile(
        "int $0x40"
        :"=b"(err_code), "=a"(status)
        :"a"(75), "b"(9), "c"(socket),"d"(opt)
    );
    return status;
}

static inline int socketpair(int *socket1, int *socket2)
{
    asm volatile(
        "int $0x40"
        :"=b"(*socket2), "=a"(*socket1)
        :"a"(75), "b"(10)
    ); 
    err_code=*socket2;
    return *socket1;
}
#endif