123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229 |
- package de.tudarmstadt.informatik.hostage.protocol.smbutils;
- import java.io.IOException;
- import java.net.DatagramPacket;
- import java.net.DatagramSocket;
- import java.net.InetAddress;
- import java.net.SocketTimeoutException;
- import java.net.UnknownHostException;
- import de.tudarmstadt.informatik.hostage.net.MyDatagramSocketFactory;
- public class NMB extends Thread {
-
- private DatagramSocket nbnsSocket;
- private DatagramSocket nbdsSocket;
- private DatagramPacket packet;
- private String ip;
- private String[] ipParts;
- private InetAddress dst;
- private int nbnsPort = 137;
- private int nbdsPort = 138;
- private String username;
- private String workgroup;
- private NBNS nbns;
- private NBDS nbds;
- private volatile boolean smbRunning;
- private Receiver receiver;
-
- public NMB(String ip, String username, String workgroup) {
- try {
- smbRunning = true;
- this.username = username;
- this.workgroup = workgroup;
- this.ip = ip;
- ipParts = ip.split("\\.");
- String newHostAddr = ipParts[0] + "." + ipParts[1] + "." + ipParts[2] + ".255";
- dst = InetAddress.getByName(newHostAddr);
- } catch (UnknownHostException e) {
- e.printStackTrace();
- }
- }
-
- private void sendPacket(NBNS nbns) {
- try {
- byte[] packetBytes = nbns.getNextPacket();
- packet = new DatagramPacket(packetBytes, packetBytes.length, dst, nbnsPort);
- nbnsSocket.send(packet);
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
-
- private void sendPacket(NBDS nbds) {
- try {
- byte[] packetBytes = nbds.getBytes();
- packet = new DatagramPacket(packetBytes, packetBytes.length, dst, nbdsPort);
- nbdsSocket.send(packet);
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
-
- private void registrate() {
- registrateUser();
- registrateGroup();
- }
-
- private void registrateUser() {
- nbns = new NBNS(new byte[]{0x50, 0x00}, NBNSType.REGISTRATION_UNIQUE, NBNSService.SERVER, username, ip);
- sendPacket(nbns);
-
- nbns.setService(NBNSService.MESSENGER);
- sendPacket(nbns);
-
- nbns.setService(NBNSService.WORKSTATION);
- sendPacket(nbns);
- }
-
- private void registrateGroup() {
- nbns.setName(workgroup);
- nbns.setType(NBNSType.REGISTRATION_GROUP);
- sendPacket(nbns);
-
- nbns.setService(NBNSService.BROWSER_ELECTION);
- sendPacket(nbns);
- }
-
- private void browserElection() {
- nbds = new NBDS(nbns.getAndIncTransactID(), nbns.getAddr(), username, workgroup, NBDSType.BROWSER);
- sendPacket(nbds);
- }
-
- private void registrateMsBrowse() {
- nbns.setName("__MSBROWSE__");
- nbns.setType(NBNSType.REGISTRATION_MSBROWSE);
- nbns.setService(NBNSService.BROWSER);
- sendPacket(nbns);
- }
-
- private void registrateLocalMaster() {
- nbns.setName(workgroup);
- nbns.setType(NBNSType.REGISTRATION_UNIQUE);
- nbns.setService(NBNSService.LOCAL_MASTER_BROWSER);
- sendPacket(nbns);
- }
-
- private void announceHost() {
- nbds = new NBDS(nbns.getAndIncTransactID(), nbns.getAddr(), username, workgroup, NBDSType.HOST_ANNOUNCEMENT_WITH_SERVICES);
- sendPacket(nbds);
- }
-
- private void queryName() {
- nbns.setType(NBNSType.NAME_QUERY);
- nbns.setService(NBNSService.LOCAL_MASTER_BROWSER);
- sendPacket(nbns);
- }
-
- private void requestAnnouncement() {
- nbds = new NBDS(nbns.getAndIncTransactID(), nbns.getAddr(), username, workgroup, NBDSType.REQUEST_ANNOUNCEMENT);
- sendPacket(nbds);
- }
-
- private void localMasterAnnouncementAll() {
- nbds = new NBDS(nbns.getAndIncTransactID(), nbns.getAddr(), username, workgroup, NBDSType.LOCAL_MASTER_ANNOUNCEMENT_ALL);
- sendPacket(nbds);
- }
-
- private void domainAnnouncement() {
- nbds = new NBDS(nbns.getAndIncTransactID(), nbns.getAddr(), username, workgroup, NBDSType.DOMAIN_ANNOUNCEMENT);
- sendPacket(nbds);
- }
-
- private void localMasterAnnouncement() {
- nbds = new NBDS(nbns.getAndIncTransactID(), nbns.getAddr(), username, workgroup, NBDSType.LOCAL_MASTER_ANNOUNCEMENT);
- sendPacket(nbds);
- }
- @Override
- public void run() {
- try {
- nbnsSocket = new MyDatagramSocketFactory().createDatagramSocket(nbnsPort);
- nbnsSocket.connect(dst, nbnsPort);
- nbdsSocket = new MyDatagramSocketFactory().createDatagramSocket(nbdsPort);
- nbdsSocket.connect(dst, nbdsPort);
- } catch (IOException e) {
- e.printStackTrace();
- }
-
- registrate();
- receiver = new Receiver();
- receiver.start();
- announceHost();
- queryName();
-
- registrate();
- queryName();
-
- registrate();
- queryName();
-
- registrate();
- queryName();
-
- browserElection();
- browserElection();
- browserElection();
- browserElection();
-
- registrateMsBrowse();
- registrateMsBrowse();
- registrateMsBrowse();
- registrateMsBrowse();
-
- registrateLocalMaster();
- registrateLocalMaster();
- registrateLocalMaster();
- registrateLocalMaster();
-
- requestAnnouncement();
- localMasterAnnouncementAll();
- domainAnnouncement();
-
- // // domain/workgroup announcement
- // // local master announcmen
-
- announceHost();
-
- nbnsSocket.close();
- nbdsSocket.close();
- }
- public synchronized void setSmbRunning(boolean smbRunning) {
- this.smbRunning = smbRunning;
- }
-
- private class Receiver extends Thread {
-
- private DatagramPacket packet;
- private byte[] buffer;
- private boolean masterAnswered = false;
-
- public Receiver() {
- buffer = new byte[2048];
- packet = new DatagramPacket(buffer, buffer.length);
- }
-
- @Override
- public void run() {
- try {
- nbnsSocket.setSoTimeout(1000);
- try {
- while (!masterAnswered) {
- nbnsSocket.receive(packet);
- if (!(packet.getAddress().toString().equals("/"+ip))) {
- masterAnswered = true;
- setSmbRunning(false);
- }
- }
- } catch (SocketTimeoutException e) {
- setSmbRunning(false);
- e.printStackTrace();
- }
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- }
-
- }
|