#include <WinSock2.h>
#include <Ws2tcpip.h>
#include <stdio.h>
#include <string.h>
#pragma comment(lib, "Ws2_32.lib")
#pragma warning(disable: 4996)

int main()
{
	WSADATA wsdt;

	int BufLen = 1024;
	char SendBuf[1024];

	SOCKET sock = INVALID_SOCKET;

	struct sockaddr_in servInfo;

	char ipAddr[20];
	int port;
	int size;

	int err;
	err = WSAStartup(MAKEWORD(2, 2), &wsdt);
	if (err != 0)
	{
        printf("WSAStartup failed with error: %d\n", err);
        return 1;
    }

	printf("Podaj adres do nasluchiwania :");
	scanf("%s", ipAddr);
	printf("Podaj port do nasluchiwania :");
	scanf("%d", &port);
	printf("Podaj wiadomosc do wyslania: ");
	scanf("%s", SendBuf);

	size = strlen(SendBuf);

	sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    if (sock == INVALID_SOCKET)
	{
        wprintf(L"socket function failed with error = %d\n", WSAGetLastError());
		system("pause");
		WSACleanup();
		exit(EXIT_FAILURE);
	}


	servInfo.sin_family = AF_INET;
	servInfo.sin_addr.s_addr = inet_addr(ipAddr);
	servInfo.sin_port = htons((u_short)port);

	/*err = connect(sock, (sockaddr*)&servInfo, sizeof(servInfo));
	if (err == SOCKET_ERROR)
	{
		wprintf(L"connect function failed with error: %ld\n", WSAGetLastError());
		system("pause");
		err = closesocket(sock);
		if (err == SOCKET_ERROR)
		{
			wprintf(L"closesocket function failed with error: %ld\n", WSAGetLastError());
			WSACleanup();
			exit(EXIT_FAILURE);
		}
	}*/

	err = sendto(sock, (char*)htonl(size), BufLen, 0, (SOCKADDR *) & servInfo, sizeof(servInfo));
    if (err == SOCKET_ERROR) {
        wprintf(L"sendto failed with error: %d\n", WSAGetLastError());
		system("pause");
        err = closesocket(sock);
		if (err == SOCKET_ERROR)
		{
			wprintf(L"closesocket function failed with error: %ld\n", WSAGetLastError());
			system("pause");
			WSACleanup();
			exit(EXIT_FAILURE);
		}
		WSACleanup();
		exit(EXIT_FAILURE);
    }

	err = sendto(sock, SendBuf, BufLen, 0, (SOCKADDR *) & servInfo, sizeof(servInfo));
    if (err == SOCKET_ERROR) {
        wprintf(L"sendto failed with error: %d\n", WSAGetLastError());
		system("pause");
        err = closesocket(sock);
		if (err == SOCKET_ERROR)
		{
			wprintf(L"closesocket function failed with error: %ld\n", WSAGetLastError());
			system("pause");
			WSACleanup();
			exit(EXIT_FAILURE);
		}
    }

	err = closesocket(sock);
    if (err == SOCKET_ERROR) {
        wprintf(L"closesocket function failed with error: %ld\n", WSAGetLastError());
        WSACleanup();
        exit(EXIT_FAILURE);
    }

	WSACleanup();
	system("pause");
	return 0;
}