import java.io.File;
import java.io.FileInputStream;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
/**--------文件名:UDPServer.java-------------*/
public class UDPServer {
public static void main(String args[]) {
try {
System.out.println("UDP服务端启动成功,等待接收消息...");
if (args.length != 1) {
throw new IllegalArgumentException("参数个数有误");
}
int port = Integer.parseInt(args[0]); // 从命令行中获取端口号参数
// 创建一个socket,侦听这个端口。
DatagramSocket dsocket = new DatagramSocket(port);
byte[] buffer = new byte[2048]; // 保存接收到的UDP报文的字节数组
// 创建一个DatagramPacket,将收到的报文写入buffer中。
// 注意,这里指定了报文的长度,如果收到的UDP报文比2048大,
// 多余的信息被舍弃
DatagramPacket packet = new DatagramPacket(buffer, buffer.length);
// 不断循环,接收数据
while (true) {
dsocket.receive(packet); // 等待收到一个数据包
// 将收到的报文的字节数组封装成字符串。
String msg = new String(buffer, 0, packet.getLength());
// 从数据包中取得消息来源的地址
System.out.println("Receive: "
+ packet.getAddress().getHostAddress() + ": " + msg);
// 如果收到QUIT指令,则退出循环。
if (msg.equalsIgnoreCase("QUIT")) {
System.out.println("退出UDP服务");
break;
}
packet.setLength(buffer.length); // 重设数据包的长度
}
// 关闭socket
dsocket.close();
} catch (Exception e) {
System.err.println(e);
// System.err.println(usage);
}
}
}
/**--------文件名:UDPClient.java-------------*/
class UDPClient {
public static void main(String args[]) {
try {
// 检查参数个数
if (args.length < 3) {
throw new IllegalArgumentException("参数个数有误");
}
// 域名和端口
String host = args[0];
int port = Integer.parseInt(args[1]);
// 下面构造待发送报文的字节数组
byte[] message;
if (args[2].equals("-f")) {
// 如果第三个参数为 -f,则表示将文件的内容以UDP方式发送
// 获得待发送的文件对象以及文件的长度
File f = new File(args[3]);
int len = (int) f.length();
// 创建一个足够容纳文件内容的字节数组
message = new byte[len];
FileInputStream in = new FileInputStream(f);
// 将文件内容以字节的方式读到字节数组中
int bytes_read = 0, n;
do {
n = in.read(message, bytes_read, len - bytes_read);
bytes_read += n;
} while ((bytes_read < len) && (n != -1));
} else {
// 如果第三个参数不是 -f,则将后面的参数当作消息发送
String msg = args[2];
for (int i = 3; i < args.length; i++) {
msg += " " + args[i];
}
message = msg.getBytes();
}
// 根据域名获取IP地址
InetAddress address = InetAddress.getByName(host);
// 初始化一个UDP包。
// DatagramPacket的构造方法中必须使用InetAddress,而不能是IP地址或者域名
DatagramPacket packet = new DatagramPacket(message, message.length,
address, port);
// 创建一个DatagramSocket,以发送UDP包
DatagramSocket dsocket = new DatagramSocket();
dsocket.send(packet);
System.out.println("send: " + new String(message));
dsocket.close();
// 注意:如果在构造DatagramPacket时,不提供IP地址和端口号,
// 则需要调用DatagramSocket的connect方法,否则无法发送UDP包
packet = new DatagramPacket(message, message.length);
dsocket = new DatagramSocket();
dsocket.connect(address, port);
dsocket.send(packet);
System.out.println("Send: " + new String(message));
dsocket.close();
} catch (Exception e) {
System.err.println(e);
}
}
}