#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/skbuff.h>

#include "transport.h"

static int test_protosw_init(void)
{
	int rc;

	rc = proto_register(&test_prot, 0);/*2nd parameter 0 or 1?*/
	if (rc)
		return rc;

	inet_register_protosw(&test_stream_protosw);

	return 0;
}

static void test_protosw_exit(void)
{
	inet_unregister_protosw(&test_stream_protosw);
	proto_unregister(&test_prot);
}

static int test_add_protocol(void)
{
	if (inet_add_protocol(&test_protocol, IPPROTO_TEST) < 0)
		return -EAGAIN;

	return 0;
}

static void test_del_protocol(void)
{
	inet_del_protocol(&test_protocol, IPPROTO_TEST);
}

static int __init init_test(void)
{
    int status = -EINVAL;
    status=test_protosw_init();
    status=test_add_protocol();
    return status;
}

static void __exit exit_test(void)
{
    test_del_protocol();
    test_protosw_exit();
}

module_init(init_test);
module_exit(exit_test);

#define IPPROTO_TEST 150
#define MAX_BUFFER 100

int main()
{
    int sockfd;
    struct sockaddr_in servaddr;
    char buffer[MAX_BUFFER+1];
    
    sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TEST);
    printf("\nconnSock:%d",sockfd);
    
    bzero((void *)&servaddr,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr =inet_addr("127.0.0.1");
    servaddr.sin_port = htons(5351);
    if(bind(sockfd,(struct sockaddr *)&servaddr,sizeof(struct sockaddr))==-1)
    {
        perror("bind");
        exit(1);
    }
    close(sockfd);
}

struct proto test_prot =
{
    .name           = "TEST",
    .owner          = THIS_MODULE,
    .close          = test_close,
    .connect        = test_connect,
    .disconnect     = test_disconnect,
    .accept         = test_accept,
    .ioctl          = test_ioctl,
    .init           = test_init_sock,
    .destroy        = test_destroy_sock,
    .shutdown       = test_shutdown,
    .setsockopt     = test_setsockopt,
    .getsockopt     = test_getsockopt,
    .sendmsg        = test_sendmsg,
    .recvmsg        = test_recvmsg,
    .backlog_rcv    = test_backlog_rcv,
    .hash           = test_hash,
    .unhash         = test_unhash,
    .get_port       = test_get_port,
};