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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
|
/*
This file is part of GNUnet.
Copyright (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., 51 Franklin Street, Fifth Floor,
Boston, MA 02110-1301, USA.
*/
package org.gnunet.peerinfo;
import org.gnunet.hello.HelloMessage;
import org.gnunet.mq.Envelope;
import org.gnunet.peerinfo.messages.InfoEnd;
import org.gnunet.peerinfo.messages.InfoMessage;
import org.gnunet.peerinfo.messages.ListAllPeersMessage;
import org.gnunet.peerinfo.messages.ListPeerMessage;
import org.gnunet.requests.Request;
import org.gnunet.requests.SequentialRequestContainer;
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(PeerInfo.class);
/**
* Client that connects to the peerinfo service.
*/
private final Client client;
/**
* All currently active iterate requests.
*/
private SequentialRequestContainer<PeerIterateRequest> iterateRequests;
private class PeerIterateRequest extends Request {
public PeerIdentity peer;
public PeerProcessor peerProcessor;
public boolean friendOnly;
public boolean canceled;
public PeerIterateRequest(PeerIdentity peer, boolean friendOnly, PeerProcessor peerProcessor) {
this.peer = peer;
this.peerProcessor = peerProcessor;
this.friendOnly = friendOnly;
}
@Override
public Envelope assembleRequest() {
if (peer == null) {
ListAllPeersMessage m = new ListAllPeersMessage();
m.includeFriendOnly = friendOnly ? 1 : 0;
return new Envelope(m);
} else {
ListPeerMessage m = new ListPeerMessage();
m.peer = peer;
return new Envelope(m);
}
}
public void onCancel() {
canceled = true;
}
}
private class PeerInfoMessageReceiver extends RunaboutMessageReceiver {
public void visit(InfoEnd infoEnd) {
PeerIterateRequest r = iterateRequests.getAndRetireRequest();
if (!r.canceled)
r.peerProcessor.onEnd();
}
public void visit(InfoMessage infoMessage) {
PeerIterateRequest r = iterateRequests.getRequest();
if (!r.canceled)
r.peerProcessor.onPeer(infoMessage.peerIdentity, infoMessage.hello);
}
@Override
public void handleError() {
client.reconnect();
iterateRequests.restart();
}
}
/**
* Create a connection to the peerinfo service.
*
* @param cfg configuration to use
*/
public PeerInfo(Configuration cfg) {
client = new Client("peerinfo", cfg);
client.installReceiver(new PeerInfoMessageReceiver());
iterateRequests = new SequentialRequestContainer<PeerIterateRequest>(client);
}
/**
* Iterates over the HELLOs of all peers.
*
* @param timeout
* @param processor
* @return a handle to cancel the request
*/
public Cancelable iterate(RelativeTime timeout, boolean friendOnly, PeerProcessor processor) {
return iterate(timeout, null, friendOnly, processor);
}
/**
* Iterates over the HELLOs of the given peer.
* Can be called with peer=null to iterate over all peers.
* @param timeout
* @param peer
* @param processor
* @return a handle to cancel the request
*/
public Cancelable iterate(RelativeTime timeout, PeerIdentity peer, boolean friendOnly, PeerProcessor processor) {
PeerIterateRequest r = new PeerIterateRequest(peer, friendOnly, processor);
return iterateRequests.addRequest(r);
}
/**
* Disconnect from the peerinfo service.
*/
public void disconnect() {
client.disconnect();
}
public static void main(String... args) {
new Program() {
@Override
public void run() {
final PeerInfo peerInfo = new PeerInfo(getConfiguration());
peerInfo.iterate(RelativeTime.FOREVER, false, new PeerProcessor() {
@Override
public void onPeer(PeerIdentity peerIdentity, HelloMessage hello) {
System.out.println("peer " + peerIdentity.toString());
}
@Override
public void onEnd() {
System.out.println("got peer end");
peerInfo.disconnect();
}
});
}
}.start(args);
}
}
|
