* Copyright 2006, Haiku, Inc. All Rights Reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
* Oliver Tappe, zooey@hirschkaefer.de
* Axel Dörfler, axeld@pinc-software.de
*/
#include <arpa/inet.h>
#include <ctype.h>
#include <errno.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#define MAXLEN 65535
#ifndef HAIKU_TARGET_PLATFORM_HAIKU
typedef int socklen_t;
#endif
extern const char* __progname;
static void
udp_echo_client(int sockFD, const struct sockaddr_in* serverAddr)
{
char buf[MAXLEN];
unsigned int len;
long status;
while (fgets(buf, MAXLEN, stdin) != NULL) {
len = strlen(buf);
if (len > 0)
len--;
printf("trying to send %u bytes...\n", len);
status = sendto(sockFD, buf, len, 0,
(struct sockaddr*)serverAddr, sizeof(struct sockaddr_in));
if (status < 0) {
printf("sendto(): %x (%s)\n", errno, strerror(errno));
exit(5);
}
len = 0;
status = recvfrom(sockFD, buf, MAXLEN-1, 0, NULL, NULL);
if (status < 0) {
printf("recvfrom(): %x (%s)\n", errno, strerror(errno));
exit(5);
}
buf[status] = 0;
printf("-> %s\n", buf);
}
}
static void
udp_broadcast(int sockFD, const struct sockaddr_in* serverAddr)
{
char buf[MAXLEN];
int option = 1;
int len;
int status;
strcpy(buf, "broadcast");
len = strlen(buf);
setsockopt(sockFD, SOL_SOCKET, SO_BROADCAST, &option, sizeof(option));
status = sendto(sockFD, buf, len, 0,
(struct sockaddr*)serverAddr, sizeof(struct sockaddr_in));
if (status < 0) {
printf("sendto(): %s\n", strerror(errno));
exit(5);
}
status = recvfrom(sockFD, buf, MAXLEN-1, 0, NULL, NULL);
if (status < 0) {
printf("recvfrom(): %x (%s)\n", errno, strerror(errno));
exit(5);
}
buf[status] = 0;
printf("-> %s\n", buf);
}
static void
udp_echo_server(int sockFD)
{
char buf[MAXLEN];
long status;
socklen_t len;
long i;
struct sockaddr_in clientAddr;
while (1) {
len = sizeof(clientAddr);
status = recvfrom(sockFD, buf, MAXLEN-1, 0, (struct sockaddr*)&clientAddr, &len);
if (status < 0) {
printf("recvfrom(): %x (%s)\n", errno, strerror(errno));
exit(5);
}
buf[status] = 0;
printf("got <%s> from client(%08x:%u)\n", buf, clientAddr.sin_addr.s_addr, clientAddr.sin_port);
for (i = 0; i < status; ++i) {
if (islower(buf[i]))
buf[i] = toupper(buf[i]);
else if (isupper(buf[i]))
buf[i] = tolower(buf[i]);
}
printf("sending <%s>\n", buf);
status = sendto(sockFD, buf, status, 0,
(struct sockaddr*)&clientAddr, sizeof(clientAddr));
if (status < 0) {
printf("sendto(): %x (%s)\n", errno, strerror(errno));
exit(5);
}
}
}
int
main(int argc, char** argv)
{
unsigned long status;
int sockFD;
struct sockaddr_in serverAddr, localAddr;
enum {
CLIENT_MODE,
BROADCAST_MODE,
SERVER_MODE,
} mode = 0;
unsigned short bindPort = 0;
const char* bindAddr = NULL;
if (argc < 2) {
printf("usage: %s client <IP-address> <port> [local-port]\n", __progname);
printf("or %s broadcast <port> <local-port>\n", __progname);
printf("or %s server <local-port>\n", __progname);
exit(5);
}
if (!strcmp(argv[1], "client")) {
mode = CLIENT_MODE;
if (argc < 4) {
printf("usage: %s client <IP-address> <port> [local-port]\n", __progname);
exit(5);
}
memset(&serverAddr, 0, sizeof(struct sockaddr_in));
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(atoi(argv[3]));
serverAddr.sin_addr.s_addr = inet_addr(argv[2]);
if (argc > 4)
bindPort = atoi(argv[4]);
printf("client connected to server(%08x:%u)\n", serverAddr.sin_addr.s_addr,
ntohs(serverAddr.sin_port));
} else if (!strcmp(argv[1], "broadcast")) {
mode = BROADCAST_MODE;
if (argc < 3) {
printf("usage: %s broadcast <port> [local-addr] [broadcast-addr] [local-port]\n", __progname);
exit(5);
}
if (argc > 3)
bindAddr = argv[3];
memset(&serverAddr, 0, sizeof(struct sockaddr_in));
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(atoi(argv[2]));
if (argc > 4)
serverAddr.sin_addr.s_addr = inet_addr(argv[4]);
else
serverAddr.sin_addr.s_addr = INADDR_BROADCAST;
if (argc > 5)
bindPort = atoi(argv[5]);
} else if (!strcmp(argv[1], "server")) {
mode = SERVER_MODE;
if (argc < 3) {
printf("usage: %s server <local-port>\n", argv[0]);
exit(5);
}
bindPort = atoi(argv[2]);
}
sockFD = socket(AF_INET, SOCK_DGRAM, 0);
if (bindAddr != NULL || bindPort > 0) {
memset(&localAddr, 0, sizeof(struct sockaddr_in));
localAddr.sin_family = AF_INET;
if (bindAddr != NULL) {
localAddr.sin_addr.s_addr = inet_addr(bindAddr);
printf("binding to addr %s\n", bindAddr);
}
if (bindPort > 0) {
localAddr.sin_port = htons(bindPort);
printf("binding to port %u\n", bindPort);
}
status = bind(sockFD, (struct sockaddr *)&localAddr, sizeof(localAddr));
if (status < 0) {
printf("bind(): %x (%s)\n", errno, strerror(errno));
exit(5);
}
}
switch (mode) {
case CLIENT_MODE:
udp_echo_client(sockFD, &serverAddr);
break;
case BROADCAST_MODE:
udp_broadcast(sockFD, &serverAddr);
break;
case SERVER_MODE:
udp_echo_server(sockFD);
break;
}
return 0;
}
