aboutsummaryrefslogtreecommitdiff
path: root/src/org/gnunet/peerinfo/PeerInfo.java
blob: 2cc04d606f6ce50d1b95faa89660249dbab28d7d (plain) (blame)
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
 This file is part of GNUnet.
 (C) 2011, 2012 Christian Grothoff (and other contributing authors)

 GNUnet 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, or (at your
 option) any later version.

 GNUnet 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 GNUnet; see the file COPYING.  If not, write to the
 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 Boston, MA 02111-1307, USA.
 */

package org.gnunet.peerinfo;


import org.gnunet.hello.HelloMessage;
import org.gnunet.requests.Request;
import org.gnunet.requests.RequestQueue;
import org.gnunet.util.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Interface to the service that maintains all known hosts.
 *
 * @author Florian Dold
 */
public class PeerInfo {
    private static final Logger logger = LoggerFactory
            .getLogger(AbsoluteTime.class);


    private final RequestQueue requests;
    private PeerIterateRequest activeRequest;


    private class PeerIterateRequest extends Request {
        public PeerIdentity peer;
        public PeerProcessor peerProcessor;
        @Override
        public void transmit(Connection.MessageSink sink) {
            if (peer == null) {
                ListAllPeersMessage lapm = new ListAllPeersMessage();
                sink.send(lapm);
            } else {
                ListPeerMessage lpm = new ListPeerMessage();
                lpm.peer = peer;
                sink.send(lpm);
            }
            activeRequest = this;
            requests.clog();
        }
    }

    private class PeerInfoMessageReceiver extends RunaboutMessageReceiver {
        public void visit(InfoEnd infoEnd) {
            activeRequest.peerProcessor.onEnd();
            activeRequest = null;
            requests.unclog();
        }
        public void visit(InfoMessage infoMessage) {
            if (activeRequest == null) {
                logger.warn("PEERINFO sent info message, but no request is active");
            }
            activeRequest.peerProcessor.onPeer(infoMessage.peerIdentity, infoMessage.hello);
        }

        @Override
        public void handleError() {
        }
    }


    public PeerInfo(Configuration cfg) {
        Client client = new Client("peerinfo", cfg);
        requests = new RequestQueue(client, new PeerInfoMessageReceiver());
    }

    public Cancelable iterate(RelativeTime timeout, PeerProcessor iterator) {
        return iterate(timeout, null, iterator);
    }

    public Cancelable iterate(RelativeTime timeout, PeerIdentity peer, PeerProcessor iterator) {
        PeerIterateRequest peerIterateRequest = new PeerIterateRequest();
        peerIterateRequest.peer = peer;
        peerIterateRequest.peerProcessor = iterator;
        peerIterateRequest.setDeadline(timeout.toAbsolute());
        return requests.add(peerIterateRequest);
    }

    public void disconnect() {
        requests.destroy();
    }

    public static void main(String... args) {
        new Program(args) {

            @Override
            public void run() {
                final PeerInfo peerInfo = new PeerInfo(getConfiguration());
                peerInfo.iterate(RelativeTime.FOREVER, new PeerProcessor() {
                    @Override
                    public void onPeer(PeerIdentity peerIdentity, HelloMessage hello) {
                        System.out.println("got peer");
                    }

                    @Override
                    public void onEnd() {
                        System.out.println("got peer end");
                        peerInfo.disconnect();
                    }
                });
            }
        }.start();

    }
}