/*
* Copyright 2015 the original author or authors.
* @https://github.com/scouter-project/scouter
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package scouter.agent.netio.data.net;
import scouter.agent.Configure;
import scouter.agent.Logger;
import scouter.io.DataInputX;
import scouter.io.DataOutputX;
import scouter.lang.conf.ConfObserver;
import scouter.net.NetCafe;
import scouter.util.*;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.util.List;
public class DataUdpAgent {
private static DataUdpAgent inst;
InetAddress server_host;
int server_port;
String local_udp_addr;
int local_udp_port;
private DatagramSocket datagram;
private DataUdpAgent() {
setTarget();
openDatagramSocket();
ConfObserver.add(DataUdpAgent.class.getName(), new Runnable() {
public void run() {
setTarget();
openDatagramSocket();
}
});
}
private void setTarget() {
Configure conf = Configure.getInstance();
String host = conf.net_collector_ip;
int port = conf.net_collector_udp_port;
try {
server_host = InetAddress.getByName(host);
server_port = port;
} catch (Exception e) {
e.printStackTrace();
}
}
protected void close(DatagramSocket d) {
if (d != null) {
try {
d.close();
} catch (Exception e) {
}
}
}
private void openDatagramSocket() {
try {
Configure conf = Configure.getInstance();
String host = conf.net_local_udp_ip;
int port = conf.net_local_udp_port;
if (datagram == null || CompareUtil.equals(host, local_udp_addr) == false || local_udp_port != port) {
close(datagram);
local_udp_addr = host;
local_udp_port = port;
if (host != null) {
datagram = new DatagramSocket(port, InetAddress.getByName(host));
Logger.println("A118", "Agent UDP local.addr=" + host + " local.port=" + port);
} else {
datagram = new DatagramSocket(port);
Logger.println("A119", "Agent UDP local.port=" + port);
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
public static synchronized DataUdpAgent getInstance() {
if (inst == null) {
inst = new DataUdpAgent();
}
return inst;
}
private Configure conf = Configure.getInstance();
public boolean write(byte[] p) {
try {
if (server_host == null)
return false;
if (p.length > conf.net_udp_packet_max_bytes) {
return writeMTU(p, conf.net_udp_packet_max_bytes);
}
DataOutputX out = new DataOutputX();
out.write(NetCafe.CAFE);
out.write(p);
byte[] buff = out.toByteArray();
DatagramPacket packet = new DatagramPacket(buff, buff.length);
packet.setAddress(server_host);
packet.setPort(server_port);
datagram.send(packet);
return true;
} catch (IOException e) {
Logger.println("A120", "UDP", e);
return false;
}
}
private boolean writeMTU(byte[] data, int packetSize) {
try {
if (server_host == null)
return false;
long pkid = KeyGen.next();
int total = data.length / packetSize;
int remainder = data.length % packetSize;
if (remainder > 0)
total++;
int num = 0;
for (num = 0; num < data.length / packetSize; num++) {
writeMTU(pkid, total, num, packetSize, DataInputX.get(data, num * packetSize, packetSize));
}
if (remainder > 0) {
writeMTU(pkid, total, num, remainder, DataInputX.get(data, data.length - remainder, remainder));
}
return true;
} catch (IOException e) {
Logger.println("A121","UDP", e);
return false;
}
}
private void writeMTU(long pkid, int total, int num, int packetSize, byte[] data) throws IOException {
DataOutputX out = new DataOutputX();
out.write(NetCafe.CAFE_MTU);
out.writeInt(conf.getObjHash());
out.writeLong(pkid);
out.writeShort(total);
out.writeShort(num);
out.writeBlob(data);
byte[] buff = out.toByteArray();
DatagramPacket packet = new DatagramPacket(buff, buff.length);
packet.setAddress(server_host);
packet.setPort(server_port);
datagram.send(packet);
}
public void close() {
if (datagram != null)
datagram.close();
datagram = null;
}
public boolean write(List<byte[]> p) {
try {
if (server_host == null)
return false;
DataOutputX buffer = new DataOutputX();
int bufferCount = 0;
for (int i = 0; i < p.size(); i++) {
byte[] b = p.get(i);
if (b.length > conf.net_udp_packet_max_bytes) {
writeMTU(b, conf.net_udp_packet_max_bytes);
} else if (b.length + buffer.getWriteSize() > conf.net_udp_packet_max_bytes) {
sendList(bufferCount, buffer.toByteArray());
buffer = new DataOutputX();
bufferCount = 1;
buffer.write(b);
} else {
bufferCount++;
buffer.write(b);
}
}
if (buffer.getWriteSize() > 0) {
sendList(bufferCount, buffer.toByteArray());
}
return true;
} catch (IOException e) {
Logger.println("A123", "UDP", e);
return false;
}
}
private void sendList(int bufferCount, byte[] buffer) throws IOException {
DataOutputX outter = new DataOutputX();
outter.write(NetCafe.CAFE_N);
outter.writeShort(bufferCount);
outter.write(buffer);
byte[] buff = outter.toByteArray();
DatagramPacket packet = new DatagramPacket(buff, buff.length);
packet.setAddress(server_host);
packet.setPort(server_port);
datagram.send(packet);
}
public boolean debugWrite(String ip, int port, int length) {
try {
DataOutputX out = new DataOutputX();
out.write("TEST".getBytes());
if (length > 4) {
out.write(new byte[length - 4]);
}
byte[] buff = out.toByteArray();
DatagramPacket packet = new DatagramPacket(buff, buff.length);
packet.setAddress(InetAddress.getByName(ip));
packet.setPort(port);
datagram.send(packet);
Logger.println("A124", "Sent " + length + " bytes to " + ip + ":" + port);
return true;
} catch (IOException e) {
Logger.println("A125", "UDP " + e.toString());
return false;
}
}
public static void main(String[] args) {
ShellArg param = new ShellArg(args);
String host = param.get("-h");
int port = CastUtil.cint(param.get("-p"));
int length = CastUtil.cint(param.get("-l"));
if (length == 0) {
System.out.println("Incorrect args\nex) -h 127.0.0.1 -p 6100 -l 32767");
return;
}
for (int i = 0; i < 100; i++) {
DataUdpAgent.getInstance().debugWrite(host, port, length);
ThreadUtil.sleep(2000);
}
}
}