SocketLib2/socketlib2_dgram.cpp at master · ferenci84/SocketLib2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
//
// Created by ferenci84 on 5/16/2020.
//

#include "socketlib2.h"
#include "socketlib2_manage_resources.h"
#include <iostream>

using namespace std;

SOCKETLIB2_API SOCKETLIB_HANDLE __stdcall open_socket_dgam_for_send(int& error){
    SOCKET sock;
    sock = socket(PF_INET, SOCK_DGRAM, 0);
    if (sock == INVALID_SOCKET) {
        error = WSAGetLastError();
        return 0;
    }
    cout << "socket opened: " << sock << endl;
    add_socket(sock);
    BOOL broadcast = TRUE;
    if (setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (char*)&broadcast, sizeof(BOOL)) != 0) {
        error = WSAGetLastError();
        int closeerr;
        close_socket(sock,closeerr);
        return 0;
    }
    return sock;
}

SOCKETLIB2_API SOCKETLIB_HANDLE __stdcall open_socket_dgam_for_receive(int& error){
    SOCKET sock;
    sock = socket(PF_INET, SOCK_DGRAM, 0);
    if (sock == INVALID_SOCKET) {
        error = WSAGetLastError();
        return 0;
    }
    cout << "socket opened: " << sock << endl;
    add_socket(sock);
    BOOL reuse = TRUE;
    if (setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,(char*)&reuse,sizeof(BOOL)) != 0) {
        error = WSAGetLastError();
        int closeerr;
        close_socket(sock,closeerr);
        return 0;
    }
    BOOL broadcast = TRUE;
    if (setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (char*)&broadcast, sizeof(BOOL)) != 0) {
        error = WSAGetLastError();
        int closeerr;
        close_socket(sock,closeerr);
        return 0;
    }
    return sock;
}

SOCKETLIB2_API int __stdcall send_dgram(SOCKETLIB_HANDLE handle, unsigned char* addr, int port, unsigned char* msg, int& error) {
    SOCKET sock = handle;
    struct sockaddr_in sockAddr;
    sockAddr.sin_family = AF_INET;
    sockAddr.sin_port = htons(port);
    cout << "send to address: " << ((char*)addr) << " port " << port << endl;
    sockAddr.sin_addr.s_addr = inet_addr((char*)addr);
    int size = strlen((char*)msg)+1;
    cout << "start sending" << endl;
    int ret = sendto(sock,(char*)msg,size,0,(struct sockaddr*) & sockAddr,sizeof(sockAddr));
    cout << "finish sending" << endl;
    if (ret == SOCKET_ERROR) {
        error = WSAGetLastError();
        return -1;
    }
    return ret;
}

SOCKETLIB2_API int __stdcall send_dgram_broadcast(SOCKETLIB_HANDLE handle, int port, unsigned char* msg, int& error) {
    SOCKET sock = handle;
    struct sockaddr_in sockAddr;
    sockAddr.sin_family = AF_INET;
    sockAddr.sin_port = htons(port);
    cout << "broadcast to address: " << " port " << port << endl;
    sockAddr.sin_addr.s_addr = INADDR_BROADCAST;
    int size = strlen((char*)msg)+1;
    cout << "start sending" << endl;
    int ret = sendto(sock,(char*)msg,size,0,(struct sockaddr*) & sockAddr,sizeof(sockAddr));
    cout << "finish sending" << endl;
    if (ret == SOCKET_ERROR) {
        error = WSAGetLastError();
        return -1;
    }
    return ret;
}

SOCKETLIB2_API int __stdcall receive_dgram_sync(SOCKETLIB_HANDLE handle, unsigned char* addr, int port, unsigned char* buffer, int bufsize, int& error) {
    SOCKET sock = handle;
    struct sockaddr_in sockAddr;
    sockAddr.sin_family = AF_INET;
    sockAddr.sin_port = htons(port);
    cout << "receive from address: " << ((char*)addr) << " port " << port << endl;
    sockAddr.sin_addr.s_addr = inet_addr((char*)addr);
    int fromlen = sizeof(sockAddr);
    cout << "start receiving" << endl;
    int ret = recvfrom(sock,(char*)buffer,bufsize,0,(struct sockaddr*) & sockAddr,&fromlen);
    cout << "finish receiving" << endl;
    if (ret == SOCKET_ERROR) {
        error = WSAGetLastError();
    }
    return ret;
}