frodo-wii/Src/NetworkWii.h

150 lines
3.0 KiB
C
Raw Normal View History

2009-02-07 12:28:29 +01:00
#include <stdio.h>
#include <stdlib.h>
#include <network.h>
static int set_sock_opts(int sock)
{
struct timeval tv;
int d = 1;
memset(&tv, 0, sizeof(tv));
#if 0
2009-02-07 12:28:29 +01:00
tv.tv_sec = 2;
net_setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO,
&tv, sizeof(struct timeval));
net_setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO,
&tv, sizeof(struct timeval));
#endif
2009-02-21 08:59:45 +01:00
return net_setsockopt(sock,SOL_SOCKET,SO_REUSEADDR, &d, sizeof(int));
2009-02-07 12:28:29 +01:00
}
2009-02-21 08:59:45 +01:00
bool Network::InitSockaddr (struct sockaddr_in *name,
2009-02-07 12:28:29 +01:00
const char *hostname, uint16_t port)
{
struct hostent *hostinfo = net_gethostbyname ((char*)hostname);
2009-02-07 12:28:29 +01:00
if (hostinfo == NULL)
{
fprintf (stderr, "Unknown host %s.\n", hostname);
return false;
}
memset(name, 0, sizeof(name));
name->sin_family = AF_INET;
name->sin_port = htons (port);
name->sin_len = 8; // sizeof(struct sockaddr_in);
name->sin_addr.s_addr = *(uint32*)hostinfo->h_addr_list[0];
2009-02-07 12:28:29 +01:00
return true;
}
2009-02-21 08:59:45 +01:00
bool Network::InitSocket(const char *remote_host, int port)
2009-02-07 12:28:29 +01:00
{
2009-02-21 08:59:45 +01:00
/* Create the socket. */
this->sock = net_socket (PF_INET, SOCK_DGRAM, 0);
if (this->sock < 0)
2009-02-07 12:28:29 +01:00
{
fprintf (stderr, "Could not init socket. Failed with %d\n", this->sock);
sleep(5);
2009-02-07 12:28:29 +01:00
return false;
}
int _true = false;
if (net_ioctl (this->sock, FIONBIO, (char *)(&_true)) < 0)
fprintf(stderr, "Could not set FIONBIO\n");
2009-02-07 12:28:29 +01:00
2009-02-21 08:59:45 +01:00
set_sock_opts(this->sock);
2009-02-07 12:28:29 +01:00
2009-02-21 08:59:45 +01:00
/* Connect to the server. */
this->InitSockaddr(&this->connection_addr, remote_host, port);
fprintf(stderr, "Bajning: %d\n",
net_bind(this->sock,(struct sockaddr *)&this->connection_addr,sizeof(struct sockaddr)));
2009-02-07 12:28:29 +01:00
return true;
}
2009-02-21 08:59:45 +01:00
bool Network::ReceiveData(void *dst, int sock, size_t sz)
2009-02-07 12:28:29 +01:00
{
2009-02-21 08:59:45 +01:00
size_t received_sz = 0;
2009-02-07 12:28:29 +01:00
2009-02-21 08:59:45 +01:00
while (received_sz < sz)
2009-02-07 12:28:29 +01:00
{
2009-02-21 08:59:45 +01:00
int v = net_read(sock, dst, sz);
2009-02-07 12:28:29 +01:00
2009-02-21 08:59:45 +01:00
if (v < 0)
return false;
received_sz += v;
2009-02-07 12:28:29 +01:00
}
2009-02-21 08:59:45 +01:00
this->traffic += received_sz;
2009-02-07 12:28:29 +01:00
2009-02-21 08:59:45 +01:00
return sz > 0;
2009-02-07 12:28:29 +01:00
}
ssize_t Network::ReceiveFrom(void *dst, int sock, size_t sz,
struct sockaddr_in *from)
{
return net_recv(sock, dst, sz, 0);
2009-02-07 12:28:29 +01:00
}
ssize_t Network::SendTo(void *src, int sock, size_t sz, struct sockaddr_in *to)
{
socklen_t to_sz = to->sin_len;
2009-02-07 12:28:29 +01:00
assert(to);
return net_sendto(sock, src, sz, 0, (struct sockaddr*)to, to_sz);
}
2009-02-21 08:59:45 +01:00
bool Network::SendData(void *src, int sock, size_t sz)
{
size_t sent_sz = 0;
2009-02-21 08:59:45 +01:00
while (sent_sz < sz)
{
int v = net_write(sock, (void*)src, sz);
if (v < 0)
return false;
sent_sz += v;
}
return true;
}
2009-02-07 12:28:29 +01:00
bool Network::Select(int sock, struct timeval *tv)
{
2009-03-28 08:59:20 +01:00
struct pollsd sds;
int v;
2009-02-07 12:28:29 +01:00
2009-03-28 08:59:20 +01:00
sds.socket = sock;
sds.events = POLLIN;
sds.revents = 0;
v = net_poll(&sds, 1, tv->tv_sec * 1000 + tv->tv_usec / 1000);
return v > 0;
2009-02-07 12:28:29 +01:00
}
void Network::CloseSocket()
{
#define SHUT_RDWR 2
net_shutdown(this->sock, 2);
2009-02-07 12:28:29 +01:00
net_close(this->sock);
}
void Network::InitNetwork()
{
char myIP[16];
/* From Snes9x-gx */
while (net_init() == -EAGAIN);
if (if_config(myIP, NULL, NULL, true) < 0)
{
fprintf(stderr, "\n\n\nError getting IP address via DHCP.\n");
sleep(2);
exit(1);
}
}
void Network::ShutdownNetwork()
{
}