def setUp(self):
self.camRede = CamadaRede()
self.camTransOrigem = CamadaTransporte(1, self.camRede)
self.camTransDestino = CamadaTransporte(2, self.camRede)
self.camRede.origem = self.camTransOrigem
self.camRede.destino = self.camTransDestino
self.camAplicOrigem = CamadaAplicacao(1, self.camTransOrigem)
self.camAplicDestino = CamadaAplicacao(2, self.camTransDestino)
self.camAplicOrigem.aplicacao= self
self.camAplicDestino.aplicacao = self
self.camTransOrigem.camadaAplicacao = self.camAplicOrigem
self.camTransDestino.camadaAplicacao = self.camAplicDestino
self.camAplicDestino.start()
self.camAplicOrigem.start()
class Host(object):
def __init__(self, nome):
super(Host, self).__init__()
self.modoVerboso = False
self.nome = nome
self.ip = ''
self.ipRoteador = ''
self.ipDnsServidor = ''
self.sniffer = None
self.enlace = None
self.buff = []
self.papel = ''
self.comandos = []
self.nomeAplicacao = None
#### funcoes das camadas
self.cmdaRedes = CamadaRedes(self)
self.cmdaAplicacao = CamadaAplicacao()
self.cmdaTransporte = CamadaTransporte( self.cmdaRedes, self.cmdaAplicacao)
self.cmdaRedes.setCamadaTransporte(self.cmdaTransporte)
self.cmdaAplicacao.setCamadaTransporte(self.cmdaTransporte)
def __str__(self):
return "HOST: " + str(self.nome) + " IP(" + str(self.ip) + ")"
def __repr__(self):
return "HOST " + str(self.nome) + " IP=" + str(self.ip)
def setIp(self, args):
if not isinstance(args, basestring):
try:
self.ipDnsServidor = args[2]
self.ipRoteador = args[1]
self.ip = args[0]
self.cmdaRedes.setIp(self.ip)
except IndexError, msg:
print ("HOST(SETIP) - Hostname = " + self.nome
+ ": Argumentos Insuficientes.")
print " " + str(msg)
else:
def __init__(self, nome):
super(Host, self).__init__()
self.modoVerboso = False
self.nome = nome
self.ip = ''
self.ipRoteador = ''
self.ipDnsServidor = ''
self.sniffer = None
self.enlace = None
self.buff = []
self.papel = ''
self.comandos = []
self.nomeAplicacao = None
#### funcoes das camadas
self.cmdaRedes = CamadaRedes(self)
self.cmdaAplicacao = CamadaAplicacao()
self.cmdaTransporte = CamadaTransporte( self.cmdaRedes, self.cmdaAplicacao)
self.cmdaRedes.setCamadaTransporte(self.cmdaTransporte)
self.cmdaAplicacao.setCamadaTransporte(self.cmdaTransporte)
from interface.interface import Interface
if __name__ == '__main__':
#Criacao da camada de rede
camRede = CamadaRede()
#Criacao das camadas de transporte utilizadas no programa.
camTransOrigem = CamadaTransporte(1, camRede)
camTransDestino = CamadaTransporte(2, camRede)
#Associacao da camada de rede com as camadas de transporte.
camRede.origem = camTransOrigem
camRede.destino = camTransDestino
#Criacao das camadas de aplicacao utilizadas no programa.
camAplicOrigem = CamadaAplicacao(1, camTransOrigem)
camAplicDestino = CamadaAplicacao(2, camTransDestino)
#Associacao das camadas de transporte com as suas camadas de aplicacao
#correspondentes.
camTransOrigem.camadaAplicacao = camAplicOrigem
camTransDestino.camadaAplicacao = camAplicDestino
#Inicializacao das camadas de aplizacao
camAplicDestino.start()
camAplicOrigem.start()
#Criacao da interface grafica do programa.
interface = Interface(camAplicOrigem, camAplicDestino)
#Inicializacao da interface grafica.
class Test(unittest.TestCase):
def setUp(self):
self.camRede = CamadaRede()
self.camTransOrigem = CamadaTransporte(1, self.camRede)
self.camTransDestino = CamadaTransporte(2, self.camRede)
self.camRede.origem = self.camTransOrigem
self.camRede.destino = self.camTransDestino
self.camAplicOrigem = CamadaAplicacao(1, self.camTransOrigem)
self.camAplicDestino = CamadaAplicacao(2, self.camTransDestino)
self.camAplicOrigem.aplicacao= self
self.camAplicDestino.aplicacao = self
self.camTransOrigem.camadaAplicacao = self.camAplicOrigem
self.camTransDestino.camadaAplicacao = self.camAplicDestino
self.camAplicDestino.start()
self.camAplicOrigem.start()
def tearDown(self):
self.camRede = None
self.camTransOrigem = None
self.camTransDestino = None
self.camAplicOrigem = None
self.camAplicDestino = None
def testeConexaoOrigemDestino(self):
self.camAplicOrigem.conectar(2)
self.assertTrue(self.camAplicOrigem.cid > 0)
self.assertTrue(self.camAplicDestino.cid > 0)
def testeConexaoDestinoOrigem(self):
self.camAplicDestino.conectar(1)
self.assertTrue(self.camAplicOrigem.cid > 0)
self.assertTrue(self.camAplicDestino.cid > 0)
def testeEnviarPacoteOrigemDestino(self):
self.camAplicOrigem.conectar(2)
self.camAplicOrigem.enviarMensagem("Teste")
self.assertEquals(("Teste",2), self.msgRecebida)
def testeEnviarPacoteDestinoOrigem(self):
self.camAplicDestino.conectar(1)
self.camAplicDestino.enviarMensagem("Teste")
self.assertEquals(("Teste",1), self.msgRecebida)
def testeDesconectarOrigem(self):
self.camAplicOrigem.conectar(2)
self.assertEquals(0, self.camAplicOrigem.fecharConexao())
def testeDesconectarDestino(self):
self.camAplicDestino.conectar(1)
self.assertEquals(0, self.camAplicOrigem.fecharConexao())
def receberMensagem(self, msg, id):
self.msgRecebida = (msg, id)
