~mateo-salta/nitroshare/nitroshare

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/* NitroShare - A simple network file sharing tool.
   Copyright (C) 2012 Nathan Osman

   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/>. */

#include <util/network.h>
#include <widgets/CBroadcastDiscoveryWidget.h>
#include "ui_CBroadcastDiscoveryWidget.h"

CBroadcastDiscoveryWidget::CBroadcastDiscoveryWidget(QWidget * parent)
    : QWidget(parent), ui(new Ui::CBroadcastDiscoveryWidget), m_listener(NULL)
{
    ui->setupUi(this);
}

CBroadcastDiscoveryWidget::~CBroadcastDiscoveryWidget()
{
    /* Free everything. */
    StopBroadcasting();

    delete ui;
}

void CBroadcastDiscoveryWidget::StartBroadcasting()
{
    ui->DiscoveryLabel->setText(tr("Broadcast discovery in progress..."));

    /* For each of the usable network interfaces, create a broadcaster
       that immediately begins listening / broadcasting. */
    QList<QNetworkInterface> interfaces = Network::FindUsableInterfaces();
    foreach(QNetworkInterface interface, interfaces)
    {
        CBasicBroadcaster * broadcaster = new CBasicBroadcaster;
        broadcaster->Init(interface);

        m_broadcasters.append(broadcaster);
    }

    /* And we also listen for any incoming pings. */
    m_listener = new CBasicListener;
    connect(m_listener, SIGNAL(Ping(QHostAddress,QByteArray)), SLOT(OnPing(QHostAddress)));

    m_listener->Init();
}

void CBroadcastDiscoveryWidget::StopBroadcasting()
{
    /* Free all of the broadcasters and the listener. */
    qDeleteAll(m_broadcasters);

    if(m_listener)
    {
        delete m_listener;
        m_listener = NULL;
    }
}

void CBroadcastDiscoveryWidget::OnPing(QHostAddress address)
{
    /* Loop through each of the network interfaces we are broadcasting
       on and check to see if this address is in its subnet. */
    foreach(CBasicBroadcaster * broadcaster, m_broadcasters)
        if(broadcaster->HasAddress(address))
        {
            ui->DiscoveryLabel->setText(tr("Found interface broadcasting to us."));
        }
}