update: migrating MPTP to TCP
This commit is contained in:
@ -2,8 +2,10 @@
|
||||
#include "../../include/mptp.h"
|
||||
#include "../../include/util.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <stdbool.h>
|
||||
#include <string.h>
|
||||
#include <sys/socket.h>
|
||||
|
||||
int lm_mptp_server_listen(char *addr, uint16_t port) {
|
||||
struct sockaddr saddr;
|
||||
@ -23,6 +25,22 @@ int lm_mptp_server_listen(char *addr, uint16_t port) {
|
||||
return sock;
|
||||
}
|
||||
|
||||
int lm_mptp_server_accept(int sock, struct sockaddr *addr){
|
||||
socklen_t sl = sizeof(struct sockaddr);
|
||||
int s = -1;
|
||||
|
||||
if((s = accept(sock, addr, &sl) < 0)){
|
||||
lm_error_set(LM_ERR_MPTPAcceptFail, strerror(errno));
|
||||
s = -1;
|
||||
}
|
||||
|
||||
return s;
|
||||
}
|
||||
|
||||
void lm_mptp_server_close(int sock){
|
||||
close(sock); // you didn't see that comming, did you?
|
||||
}
|
||||
|
||||
bool lm_mptp_server_verify(lm_mptp_t *packet) {
|
||||
if (!lm_mptp_verify(packet))
|
||||
return false;
|
||||
@ -45,20 +63,19 @@ bool lm_mptp_server_verify(lm_mptp_t *packet) {
|
||||
return true;
|
||||
}
|
||||
|
||||
bool lm_mptp_server_recv(int sock, lm_mptp_t *packet, struct sockaddr *addr) {
|
||||
if (NULL == packet || NULL == addr) {
|
||||
bool lm_mptp_server_recv(int sock, lm_mptp_t *packet) {
|
||||
if (NULL == packet) {
|
||||
lm_error_set(LM_ERR_ArgNULL);
|
||||
return false;
|
||||
}
|
||||
|
||||
char buffer[sizeof(packet->header) + MPTP_HOST_MAX + MPTP_DATA_MAX];
|
||||
socklen_t socklen = sizeof(struct sockaddr);
|
||||
ssize_t total = sizeof(buffer), used = 0;
|
||||
|
||||
bzero(buffer, sizeof(buffer));
|
||||
bzero(packet, sizeof(lm_mptp_t));
|
||||
|
||||
if (recvfrom(sock, buffer, sizeof(buffer), 0, addr, &socklen) <= 0) {
|
||||
if (recv(sock, buffer, sizeof(buffer), 0) <= 0) {
|
||||
lm_error_set(LM_ERR_MPTPRecvFail);
|
||||
return false;
|
||||
}
|
||||
@ -77,7 +94,7 @@ bool lm_mptp_server_recv(int sock, lm_mptp_t *packet, struct sockaddr *addr) {
|
||||
return true;
|
||||
}
|
||||
|
||||
bool lm_mptp_server_send(int sock, lm_mptp_t *packet, struct sockaddr *addr) {
|
||||
bool lm_mptp_server_send(int sock, lm_mptp_t *packet) {
|
||||
if (NULL == packet) {
|
||||
lm_error_set(LM_ERR_ArgNULL);
|
||||
return false;
|
||||
@ -98,7 +115,6 @@ bool lm_mptp_server_send(int sock, lm_mptp_t *packet, struct sockaddr *addr) {
|
||||
return false;
|
||||
}
|
||||
|
||||
socklen_t addrlen = sizeof(struct sockaddr);
|
||||
char buffer[sizeof(packet->header) + packet->header.host_size + packet->header.data_size];
|
||||
ssize_t total = sizeof(buffer), used = 0;
|
||||
|
||||
@ -108,7 +124,7 @@ bool lm_mptp_server_send(int sock, lm_mptp_t *packet, struct sockaddr *addr) {
|
||||
copy_to_buffer(buffer, packet->host, packet->header.host_size, &total, &used);
|
||||
copy_to_buffer(buffer, packet->data, packet->header.data_size, &total, &used);
|
||||
|
||||
if (sendto(sock, buffer, sizeof(buffer), 0, addr, addrlen) < 0) {
|
||||
if (send(sock, buffer, sizeof(buffer), 0) < 0) {
|
||||
lm_error_set(LM_ERR_MPTPSendFail);
|
||||
return false;
|
||||
}
|
||||
|
Reference in New Issue
Block a user