#include <net/if.h>		/* ifreq */
#include <netinet/if_ether.h>	/* ETH_P_IP, ETH_P_ALL */
#include <netpacket/packet.h>	/* sockaddr_ll, PACKET_OTHERHOST */

#include "wrapper.h"

void send_there(struct in_addr ip4_addr, unsigned char ttl, unsigned int type, unsigned char *payload, unsigned int paylen) {
	struct sockaddr_in sock_addr;

	int sock;

	/* prepare data for the socket */
	sock_addr.sin_family = AF_INET;
	sock_addr.sin_port   = 0;
	sock_addr.sin_addr   = ip4_addr;

	/* initialize the socket */
	if ((sock = socket(AF_INET, SOCK_RAW, type)) == -1) {
		fprintf(stderr, "Couldn't open RAW socket.\n");
		perror("socket()");
		exit(EXIT_FAILURE);
	}

	/* applies settings of the socket */
	if (setsockopt(sock, IPPROTO_IP, IP_TTL, (const char *) &ttl, sizeof(ttl)) == -1) {
		fprintf(stderr, "Couldn't apply settings of the socket.\n");
		perror("setsockopt()");
		exit(EXIT_FAILURE);
	}

	/* send the packet */
	if (sendto(sock, (char *) payload, paylen, 0, (struct sockaddr *) &sock_addr, sizeof(struct sockaddr)) != paylen) {
		fprintf(stderr, "      Error: Couldn't send an IPv4 packet.\n");
		perror("sendto()");
		exit(EXIT_FAILURE);
	}

	/* close the socket */
	close(sock);
}

void send_raw_ipv4(struct in_addr ip4_addr, unsigned char *packet, int packet_size)
{
	struct sockaddr_in	socket_address;

	unsigned char		on = 1;
	int			sock;

	/* prepare data for RAW socket */
	socket_address.sin_family = AF_INET;
	socket_address.sin_port   = htons(0);
	memcpy(&socket_address.sin_addr.s_addr, &ip4_addr, sizeof(struct in_addr));

	/* initialize raw socket */
	if ((sock = socket(AF_INET, SOCK_RAW, IPPROTO_RAW)) == -1) {
		fprintf(stderr, "Couldn't open RAW socket.\n");
		perror("socket()");
		exit(EXIT_FAILURE);
	}

	/* we will provide our own IPv4 header */
	if (setsockopt(sock, IPPROTO_IP, IP_HDRINCL, &on, sizeof(on)) == -1) {
		fprintf(stderr, "Couldn't apply the socket settings.\n");
		perror("setsockopt()");
		exit(EXIT_FAILURE);
	}

	/* send the packet */
	if (sendto(sock, packet, packet_size, 0, (struct sockaddr *) &socket_address, sizeof(struct sockaddr)) != packet_size) {
		fprintf(stderr, "Couldn't send a RAW packet.\n");
		perror("sendto()");
		exit(EXIT_FAILURE);
	}

	/* close the socket */
	close(sock);
}

void send_raw(unsigned char *packet, int packet_size)
{
	struct sockaddr_ll	socket_address;
	struct ifreq		ifr;

	int sock;

	/* prepare data for RAW socket */
	socket_address.sll_family	= PF_PACKET;		/* RAW communication */
	socket_address.sll_protocol	= htons(ETH_P_IP);	/* protocol above the ethernet layer */
	socket_address.sll_ifindex	= dev_index;		/* set index of the network device */
	socket_address.sll_pkttype	= PACKET_OTHERHOST;	/* target host is another host */

	/* initialize with zeros */
	memset(&ifr, 0, sizeof(struct ifreq));

	/* set device */
	strncpy(ifr.ifr_name, dev, strlen(dev));

	/* initialize raw socket */
	if ((sock = socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL))) == -1) {
		fprintf(stderr, "Couldn't open RAW socket.\n");
		perror("socket()");
		exit(EXIT_FAILURE);
	}

	/* bind the socket to the interface */
	if (setsockopt(sock, SOL_SOCKET, SO_BINDTODEVICE, &ifr, sizeof(struct ifreq)) == -1){
		fprintf(stderr, "Couldn't bind the socket to the interface.\n");
		perror("setsockopt()");
		exit(EXIT_FAILURE);
	}

	/* send the packet */
	if (sendto(sock, packet, packet_size, 0, (struct sockaddr *) &socket_address, sizeof(struct sockaddr_ll)) != packet_size) {
		fprintf(stderr, "Couldn't send a RAW packet.\n");
		perror("sendto()");
		exit(EXIT_FAILURE);
	}

	/* close the socket */
	close(sock);
}