added naive OneService implementation.

Does nothing but send packets to master nodes and listen for packets back.
This commit is contained in:
Grant Limberg 2015-05-02 18:24:52 -07:00
parent ad6ec22857
commit 06ab338ecc

View File

@ -0,0 +1,198 @@
/*
* ZeroTier One - Network Virtualization Everywhere
* Copyright (C) 2011-2015 ZeroTier, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* --
*
* ZeroTier may be used and distributed under the terms of the GPLv3, which
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
*
* If you would like to embed ZeroTier into a commercial application or
* redistribute it in a modified binary form, please contact ZeroTier Networks
* LLC. Start here: http://www.zerotier.com/
*/
package com.zerotier.one;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.SocketException;
import com.zerotier.sdk.Event;
import com.zerotier.sdk.EventListener;
import com.zerotier.sdk.Node;
import com.zerotier.sdk.PacketSender;
import com.zerotier.sdk.ResultCode;
import com.zerotier.sdk.Version;
import com.zerotier.sdk.VirtualNetworkConfig;
import com.zerotier.sdk.VirtualNetworkConfigListener;
import com.zerotier.sdk.VirtualNetworkConfigOperation;
import com.zerotier.sdk.VirtualNetworkFrameListener;
public class OneService extends Thread implements Runnable, PacketSender,
EventListener, VirtualNetworkConfigListener,
VirtualNetworkFrameListener {
private Node _node;
private int _port;
private DatagramSocket _udpSocket;
private ServerSocket _tcpSocket;
private DataStore _ds;
private long _nextBackgroundTaskDeadline = 0;
private final Thread _udpReceiveThread = new Thread() {
@Override
public void run() {
try {
long[] bgtask = new long[1];
byte[] buffer = new byte[16384];
while(true) {
bgtask[0] = 0;
DatagramPacket p = new DatagramPacket(buffer, buffer.length);
_udpSocket.receive(p);
if(p.getLength() > 0)
{
System.out.println("Got Data From: " + p.getAddress().toString() +":" + p.getPort());
_node.processWirePacket(System.currentTimeMillis(), new InetSocketAddress(p.getAddress(), p.getPort()), 0, p.getData(), bgtask);
_nextBackgroundTaskDeadline = bgtask[0];
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
};
public OneService(DataStoreFileProvider prov, int port) {
this._port = port;
this._ds = new DataStore(prov);
try {
_udpSocket = new DatagramSocket(_port);
_tcpSocket = new ServerSocket();
_tcpSocket.bind(new InetSocketAddress("127.0.0.1", _port));
} catch (SocketException e) {
e.printStackTrace();
return;
} catch (IOException e) {
e.printStackTrace();
return;
}
_udpReceiveThread.start();
_node = new Node(
System.currentTimeMillis(),
_ds,
_ds,
this,
this,
this,
this);
}
@Override
public void run() {
while(true) {
try {
long dl = _nextBackgroundTaskDeadline;
long now = System.currentTimeMillis();
if (dl <= now) {
long[] returnDeadline = {0};
ResultCode rc = _node.processBackgroundTasks(now, returnDeadline);
_nextBackgroundTaskDeadline = returnDeadline[0];
if(rc != ResultCode.RESULT_OK) {
System.out.println(rc.toString());
}
}
long delay = (dl > now) ? (dl - now) : 100;
Thread.sleep(delay);
} catch (Exception ex) {
System.out.println("Exception in run loop: " + ex.getMessage());
ex.printStackTrace();
}
}
}
@Override
public int onSendPacketRequested(InetSocketAddress addr,
int linkDesperation, byte[] packetData) {
System.out.println("onSendPacketRequested to: " + addr.getHostString() +":"+ addr.getPort() + " ");
if(_udpSocket == null)
return -1;
try {
DatagramPacket p = new DatagramPacket(packetData, packetData.length, addr);
_udpSocket.send(p);
System.out.println("Sent");
} catch (Exception e) {
System.out.println("Error sending datagram: " + e.getMessage());
return -1;
}
return 0;
}
@Override
public void onVirtualNetworkFrame(long nwid, long srcMac, long destMac,
long etherType, long vlanId, byte[] frameData) {
// TODO Auto-generated method stub
}
@Override
public int onNetworkConfigurationUpdated(long nwid,
VirtualNetworkConfigOperation op, VirtualNetworkConfig config) {
// TODO Auto-generated method stub
return 0;
}
@Override
public void onEvent(Event event) {
// TODO Auto-generated method stub
}
@Override
public void onNetworkError(Event event, InetSocketAddress source) {
// TODO Auto-generated method stub
}
@Override
public void onOutOfDate(Version newVersion) {
// TODO Auto-generated method stub
}
@Override
public void onTrace(String message) {
// TODO Auto-generated method stub
}
}