]> git.rkrishnan.org Git - tahoe-lafs/tahoe-lafs.git/blob - src/allmydata/web/status.py
Fix mutable status (mapupdate/retrieve/publish) to use serverids, not tubids
[tahoe-lafs/tahoe-lafs.git] / src / allmydata / web / status.py
1
2 import time, pprint, itertools
3 import simplejson
4 from twisted.internet import defer
5 from nevow import rend, inevow, tags as T
6 from allmydata.util import base32, idlib
7 from allmydata.web.common import getxmlfile, get_arg, \
8      abbreviate_time, abbreviate_rate, abbreviate_size, plural, compute_rate
9 from allmydata.interfaces import IUploadStatus, IDownloadStatus, \
10      IPublishStatus, IRetrieveStatus, IServermapUpdaterStatus
11
12 class RateAndTimeMixin:
13
14     def render_time(self, ctx, data):
15         return abbreviate_time(data)
16
17     def render_rate(self, ctx, data):
18         return abbreviate_rate(data)
19
20 class UploadResultsRendererMixin(RateAndTimeMixin):
21     # this requires a method named 'upload_results'
22
23     def render_pushed_shares(self, ctx, data):
24         d = self.upload_results()
25         d.addCallback(lambda res: res.pushed_shares)
26         return d
27
28     def render_preexisting_shares(self, ctx, data):
29         d = self.upload_results()
30         d.addCallback(lambda res: res.preexisting_shares)
31         return d
32
33     def render_sharemap(self, ctx, data):
34         d = self.upload_results()
35         d.addCallback(lambda res: res.sharemap)
36         def _render(sharemap):
37             if sharemap is None:
38                 return "None"
39             l = T.ul()
40             for shnum, peerids in sorted(sharemap.items()):
41                 peerids = ', '.join([idlib.shortnodeid_b2a(i) for i in peerids])
42                 l[T.li["%d -> placed on [%s]" % (shnum, peerids)]]
43             return l
44         d.addCallback(_render)
45         return d
46
47     def render_servermap(self, ctx, data):
48         d = self.upload_results()
49         d.addCallback(lambda res: res.servermap)
50         def _render(servermap):
51             if servermap is None:
52                 return "None"
53             l = T.ul()
54             for peerid in sorted(servermap.keys()):
55                 peerid_s = idlib.shortnodeid_b2a(peerid)
56                 shares_s = ",".join(["#%d" % shnum
57                                      for shnum in servermap[peerid]])
58                 l[T.li["[%s] got share%s: %s" % (peerid_s,
59                                                  plural(servermap[peerid]),
60                                                  shares_s)]]
61             return l
62         d.addCallback(_render)
63         return d
64
65     def data_file_size(self, ctx, data):
66         d = self.upload_results()
67         d.addCallback(lambda res: res.file_size)
68         return d
69
70     def _get_time(self, name):
71         d = self.upload_results()
72         d.addCallback(lambda res: res.timings.get(name))
73         return d
74
75     def data_time_total(self, ctx, data):
76         return self._get_time("total")
77
78     def data_time_storage_index(self, ctx, data):
79         return self._get_time("storage_index")
80
81     def data_time_contacting_helper(self, ctx, data):
82         return self._get_time("contacting_helper")
83
84     def data_time_existence_check(self, ctx, data):
85         return self._get_time("existence_check")
86
87     def data_time_cumulative_fetch(self, ctx, data):
88         return self._get_time("cumulative_fetch")
89
90     def data_time_helper_total(self, ctx, data):
91         return self._get_time("helper_total")
92
93     def data_time_peer_selection(self, ctx, data):
94         return self._get_time("peer_selection")
95
96     def data_time_total_encode_and_push(self, ctx, data):
97         return self._get_time("total_encode_and_push")
98
99     def data_time_cumulative_encoding(self, ctx, data):
100         return self._get_time("cumulative_encoding")
101
102     def data_time_cumulative_sending(self, ctx, data):
103         return self._get_time("cumulative_sending")
104
105     def data_time_hashes_and_close(self, ctx, data):
106         return self._get_time("hashes_and_close")
107
108     def _get_rate(self, name):
109         d = self.upload_results()
110         def _convert(r):
111             file_size = r.file_size
112             time = r.timings.get(name)
113             return compute_rate(file_size, time)
114         d.addCallback(_convert)
115         return d
116
117     def data_rate_total(self, ctx, data):
118         return self._get_rate("total")
119
120     def data_rate_storage_index(self, ctx, data):
121         return self._get_rate("storage_index")
122
123     def data_rate_encode(self, ctx, data):
124         return self._get_rate("cumulative_encoding")
125
126     def data_rate_push(self, ctx, data):
127         return self._get_rate("cumulative_sending")
128
129     def data_rate_encode_and_push(self, ctx, data):
130         d = self.upload_results()
131         def _convert(r):
132             file_size = r.file_size
133             time1 = r.timings.get("cumulative_encoding")
134             time2 = r.timings.get("cumulative_sending")
135             if (time1 is None or time2 is None):
136                 return None
137             else:
138                 return compute_rate(file_size, time1+time2)
139         d.addCallback(_convert)
140         return d
141
142     def data_rate_ciphertext_fetch(self, ctx, data):
143         d = self.upload_results()
144         def _convert(r):
145             fetch_size = r.ciphertext_fetched
146             time = r.timings.get("cumulative_fetch")
147             return compute_rate(fetch_size, time)
148         d.addCallback(_convert)
149         return d
150
151 class UploadStatusPage(UploadResultsRendererMixin, rend.Page):
152     docFactory = getxmlfile("upload-status.xhtml")
153
154     def __init__(self, data):
155         rend.Page.__init__(self, data)
156         self.upload_status = data
157
158     def upload_results(self):
159         return defer.maybeDeferred(self.upload_status.get_results)
160
161     def render_results(self, ctx, data):
162         d = self.upload_results()
163         def _got_results(results):
164             if results:
165                 return ctx.tag
166             return ""
167         d.addCallback(_got_results)
168         return d
169
170     def render_started(self, ctx, data):
171         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
172         started_s = time.strftime(TIME_FORMAT,
173                                   time.localtime(data.get_started()))
174         return started_s
175
176     def render_si(self, ctx, data):
177         si_s = base32.b2a_or_none(data.get_storage_index())
178         if si_s is None:
179             si_s = "(None)"
180         return si_s
181
182     def render_helper(self, ctx, data):
183         return {True: "Yes",
184                 False: "No"}[data.using_helper()]
185
186     def render_total_size(self, ctx, data):
187         size = data.get_size()
188         if size is None:
189             return "(unknown)"
190         return size
191
192     def render_progress_hash(self, ctx, data):
193         progress = data.get_progress()[0]
194         # TODO: make an ascii-art bar
195         return "%.1f%%" % (100.0 * progress)
196
197     def render_progress_ciphertext(self, ctx, data):
198         progress = data.get_progress()[1]
199         # TODO: make an ascii-art bar
200         return "%.1f%%" % (100.0 * progress)
201
202     def render_progress_encode_push(self, ctx, data):
203         progress = data.get_progress()[2]
204         # TODO: make an ascii-art bar
205         return "%.1f%%" % (100.0 * progress)
206
207     def render_status(self, ctx, data):
208         return data.get_status()
209
210 class DownloadResultsRendererMixin(RateAndTimeMixin):
211     # this requires a method named 'download_results'
212
213     def render_servermap(self, ctx, data):
214         d = self.download_results()
215         d.addCallback(lambda res: res.servermap)
216         def _render(servermap):
217             if servermap is None:
218                 return "None"
219             l = T.ul()
220             for peerid in sorted(servermap.keys()):
221                 peerid_s = idlib.shortnodeid_b2a(peerid)
222                 shares_s = ",".join(["#%d" % shnum
223                                      for shnum in servermap[peerid]])
224                 l[T.li["[%s] has share%s: %s" % (peerid_s,
225                                                  plural(servermap[peerid]),
226                                                  shares_s)]]
227             return l
228         d.addCallback(_render)
229         return d
230
231     def render_servers_used(self, ctx, data):
232         d = self.download_results()
233         d.addCallback(lambda res: res.servers_used)
234         def _got(servers_used):
235             if not servers_used:
236                 return ""
237             peerids_s = ", ".join(["[%s]" % idlib.shortnodeid_b2a(peerid)
238                                    for peerid in servers_used])
239             return T.li["Servers Used: ", peerids_s]
240         d.addCallback(_got)
241         return d
242
243     def render_problems(self, ctx, data):
244         d = self.download_results()
245         d.addCallback(lambda res: res.server_problems)
246         def _got(server_problems):
247             if not server_problems:
248                 return ""
249             l = T.ul()
250             for peerid in sorted(server_problems.keys()):
251                 peerid_s = idlib.shortnodeid_b2a(peerid)
252                 l[T.li["[%s]: %s" % (peerid_s, server_problems[peerid])]]
253             return T.li["Server Problems:", l]
254         d.addCallback(_got)
255         return d
256
257     def data_file_size(self, ctx, data):
258         d = self.download_results()
259         d.addCallback(lambda res: res.file_size)
260         return d
261
262     def _get_time(self, name):
263         d = self.download_results()
264         d.addCallback(lambda res: res.timings.get(name))
265         return d
266
267     def data_time_total(self, ctx, data):
268         return self._get_time("total")
269
270     def data_time_peer_selection(self, ctx, data):
271         return self._get_time("peer_selection")
272
273     def data_time_uri_extension(self, ctx, data):
274         return self._get_time("uri_extension")
275
276     def data_time_hashtrees(self, ctx, data):
277         return self._get_time("hashtrees")
278
279     def data_time_segments(self, ctx, data):
280         return self._get_time("segments")
281
282     def data_time_cumulative_fetch(self, ctx, data):
283         return self._get_time("cumulative_fetch")
284
285     def data_time_cumulative_decode(self, ctx, data):
286         return self._get_time("cumulative_decode")
287
288     def data_time_cumulative_decrypt(self, ctx, data):
289         return self._get_time("cumulative_decrypt")
290
291     def data_time_paused(self, ctx, data):
292         return self._get_time("paused")
293
294     def _get_rate(self, name):
295         d = self.download_results()
296         def _convert(r):
297             file_size = r.file_size
298             time = r.timings.get(name)
299             return compute_rate(file_size, time)
300         d.addCallback(_convert)
301         return d
302
303     def data_rate_total(self, ctx, data):
304         return self._get_rate("total")
305
306     def data_rate_segments(self, ctx, data):
307         return self._get_rate("segments")
308
309     def data_rate_fetch(self, ctx, data):
310         return self._get_rate("cumulative_fetch")
311
312     def data_rate_decode(self, ctx, data):
313         return self._get_rate("cumulative_decode")
314
315     def data_rate_decrypt(self, ctx, data):
316         return self._get_rate("cumulative_decrypt")
317
318     def render_server_timings(self, ctx, data):
319         d = self.download_results()
320         d.addCallback(lambda res: res.timings.get("fetch_per_server"))
321         def _render(per_server):
322             if per_server is None:
323                 return ""
324             l = T.ul()
325             for peerid in sorted(per_server.keys()):
326                 peerid_s = idlib.shortnodeid_b2a(peerid)
327                 times_s = ", ".join([self.render_time(None, t)
328                                      for t in per_server[peerid]])
329                 l[T.li["[%s]: %s" % (peerid_s, times_s)]]
330             return T.li["Per-Server Segment Fetch Response Times: ", l]
331         d.addCallback(_render)
332         return d
333
334 class DownloadStatusPage(DownloadResultsRendererMixin, rend.Page):
335     docFactory = getxmlfile("download-status.xhtml")
336
337     def __init__(self, data):
338         rend.Page.__init__(self, data)
339         self.download_status = data
340
341     def child_timeline(self, ctx):
342         return DownloadStatusTimelinePage(self.download_status)
343
344     def download_results(self):
345         return defer.maybeDeferred(self.download_status.get_results)
346
347     def relative_time(self, t):
348         if t is None:
349             return t
350         if self.download_status.first_timestamp is not None:
351             return t - self.download_status.first_timestamp
352         return t
353     def short_relative_time(self, t):
354         t = self.relative_time(t)
355         if t is None:
356             return ""
357         return "+%.6fs" % t
358
359     def _find_overlap(self, events, start_key, end_key):
360         # given a list of event dicts, return a new list in which each event
361         # has an extra "row" key (an int, starting at 0), and if appropriate
362         # a "serverid" key (ascii-encoded server id), replacing the "server"
363         # key. This is a hint to our JS frontend about how to overlap the
364         # parts of the graph it is drawing.
365
366         # we must always make a copy, since we're going to be adding keys
367         # and don't want to change the original objects. If we're
368         # stringifying serverids, we'll also be changing the serverid keys.
369         new_events = []
370         rows = []
371         for ev in events:
372             ev = ev.copy()
373             if ev.has_key('server'):
374                 ev["serverid"] = base32.b2a(ev["server"].get_serverid())
375                 del ev["server"]
376             # find an empty slot in the rows
377             free_slot = None
378             for row,finished in enumerate(rows):
379                 if finished is not None:
380                     if ev[start_key] > finished:
381                         free_slot = row
382                         break
383             if free_slot is None:
384                 free_slot = len(rows)
385                 rows.append(ev[end_key])
386             else:
387                 rows[free_slot] = ev[end_key]
388             ev["row"] = free_slot
389             new_events.append(ev)
390         return new_events
391
392     def _find_overlap_requests(self, events):
393         """We compute a three-element 'row tuple' for each event: (serverid,
394         shnum, row). All elements are ints. The first is a mapping from
395         serverid to group number, the second is a mapping from shnum to
396         subgroup number. The third is a row within the subgroup.
397
398         We also return a list of lists of rowcounts, so renderers can decide
399         how much vertical space to give to each row.
400         """
401
402         serverid_to_group = {}
403         groupnum_to_rows = {} # maps groupnum to a table of rows. Each table
404                               # is a list with an element for each row number
405                               # (int starting from 0) that contains a
406                               # finish_time, indicating that the row is empty
407                               # beyond that time. If finish_time is None, it
408                               # indicate a response that has not yet
409                               # completed, so the row cannot be reused.
410         new_events = []
411         for ev in events:
412             # DownloadStatus promises to give us events in temporal order
413             ev = ev.copy()
414             ev["serverid"] = base32.b2a(ev["server"].get_serverid())
415             del ev["server"]
416             if ev["serverid"] not in serverid_to_group:
417                 groupnum = len(serverid_to_group)
418                 serverid_to_group[ev["serverid"]] = groupnum
419             groupnum = serverid_to_group[ev["serverid"]]
420             if groupnum not in groupnum_to_rows:
421                 groupnum_to_rows[groupnum] = []
422             rows = groupnum_to_rows[groupnum]
423             # find an empty slot in the rows
424             free_slot = None
425             for row,finished in enumerate(rows):
426                 if finished is not None:
427                     if ev["start_time"] > finished:
428                         free_slot = row
429                         break
430             if free_slot is None:
431                 free_slot = len(rows)
432                 rows.append(ev["finish_time"])
433             else:
434                 rows[free_slot] = ev["finish_time"]
435             ev["row"] = (groupnum, free_slot)
436             new_events.append(ev)
437         # maybe also return serverid_to_group, groupnum_to_rows, and some
438         # indication of the highest finish_time
439         #
440         # actually, return the highest rownum for each groupnum
441         highest_rownums = [len(groupnum_to_rows[groupnum])
442                            for groupnum in range(len(serverid_to_group))]
443         return new_events, highest_rownums
444
445     def child_event_json(self, ctx):
446         inevow.IRequest(ctx).setHeader("content-type", "text/plain")
447         data = { } # this will be returned to the GET
448         ds = self.download_status
449
450         data["misc"] = self._find_overlap(ds.misc_events,
451                                           "start_time", "finish_time")
452         data["read"] = self._find_overlap(ds.read_events,
453                                           "start_time", "finish_time")
454         data["segment"] = self._find_overlap(ds.segment_events,
455                                              "start_time", "finish_time")
456         # TODO: overlap on DYHB isn't very useful, and usually gets in the
457         # way. So don't do it.
458         data["dyhb"] = self._find_overlap(ds.dyhb_requests,
459                                           "start_time", "finish_time")
460         data["block"],data["block_rownums"] = self._find_overlap_requests(ds.block_requests)
461
462         servernums = {}
463         serverid_strings = {}
464         for d_ev in data["dyhb"]:
465             if d_ev["serverid"] not in servernums:
466                 servernum = len(servernums)
467                 servernums[d_ev["serverid"]] = servernum
468                 #title= "%s: %s" % ( ",".join([str(shnum) for shnum in shnums]))
469                 serverid_strings[servernum] = d_ev["serverid"][:4]
470         data["server_info"] = dict([(serverid, {"num": servernums[serverid],
471                                                 "color": self.color(base32.a2b(serverid)),
472                                                 "short": serverid_strings[servernums[serverid]],
473                                                 })
474                                    for serverid in servernums.keys()])
475         data["num_serverids"] = len(serverid_strings)
476         # we'd prefer the keys of serverids[] to be ints, but this is JSON,
477         # so they get converted to strings. Stupid javascript.
478         data["serverids"] = serverid_strings
479         data["bounds"] = {"min": ds.first_timestamp, "max": ds.last_timestamp}
480         return simplejson.dumps(data, indent=1) + "\n"
481
482     def render_timeline_link(self, ctx, data):
483         from nevow import url
484         return T.a(href=url.URL.fromContext(ctx).child("timeline"))["timeline"]
485
486     def _rate_and_time(self, bytes, seconds):
487         time_s = self.render_time(None, seconds)
488         if seconds != 0:
489             rate = self.render_rate(None, 1.0 * bytes / seconds)
490             return T.span(title=rate)[time_s]
491         return T.span[time_s]
492
493     def render_events(self, ctx, data):
494         if not self.download_status.storage_index:
495             return
496         srt = self.short_relative_time
497         l = T.div()
498
499         t = T.table(align="left", class_="status-download-events")
500         t[T.tr[T.th["serverid"], T.th["sent"], T.th["received"],
501                T.th["shnums"], T.th["RTT"]]]
502         for d_ev in self.download_status.dyhb_requests:
503             server = d_ev["server"]
504             sent = d_ev["start_time"]
505             shnums = d_ev["response_shnums"]
506             received = d_ev["finish_time"]
507             rtt = None
508             if received is not None:
509                 rtt = received - sent
510             if not shnums:
511                 shnums = ["-"]
512             t[T.tr(style="background: %s" % self.color(server.get_serverid()))[
513                 [T.td[server.get_name()], T.td[srt(sent)], T.td[srt(received)],
514                  T.td[",".join([str(shnum) for shnum in shnums])],
515                  T.td[self.render_time(None, rtt)],
516                  ]]]
517
518         l[T.h2["DYHB Requests:"], t]
519         l[T.br(clear="all")]
520
521         t = T.table(align="left",class_="status-download-events")
522         t[T.tr[T.th["range"], T.th["start"], T.th["finish"], T.th["got"],
523                T.th["time"], T.th["decrypttime"], T.th["pausedtime"],
524                T.th["speed"]]]
525         for r_ev in self.download_status.read_events:
526             start = r_ev["start"]
527             length = r_ev["length"]
528             bytes = r_ev["bytes_returned"]
529             decrypt_time = ""
530             if bytes:
531                 decrypt_time = self._rate_and_time(bytes, r_ev["decrypt_time"])
532             speed, rtt = "",""
533             if r_ev["finish_time"] is not None:
534                 rtt = r_ev["finish_time"] - r_ev["start_time"] - r_ev["paused_time"]
535                 speed = self.render_rate(None, compute_rate(bytes, rtt))
536                 rtt = self.render_time(None, rtt)
537             paused = self.render_time(None, r_ev["paused_time"])
538
539             t[T.tr[T.td["[%d:+%d]" % (start, length)],
540                    T.td[srt(r_ev["start_time"])], T.td[srt(r_ev["finish_time"])],
541                    T.td[bytes], T.td[rtt],
542                    T.td[decrypt_time], T.td[paused],
543                    T.td[speed],
544                    ]]
545
546         l[T.h2["Read Events:"], t]
547         l[T.br(clear="all")]
548
549         t = T.table(align="left",class_="status-download-events")
550         t[T.tr[T.th["segnum"], T.th["start"], T.th["active"], T.th["finish"],
551                T.th["range"],
552                T.th["decodetime"], T.th["segtime"], T.th["speed"]]]
553         for s_ev in self.download_status.segment_events:
554             range_s = "-"
555             segtime_s = "-"
556             speed = "-"
557             decode_time = "-"
558             if s_ev["finish_time"] is not None:
559                 if s_ev["success"]:
560                     segtime = s_ev["finish_time"] - s_ev["active_time"]
561                     segtime_s = self.render_time(None, segtime)
562                     seglen = s_ev["segment_length"]
563                     range_s = "[%d:+%d]" % (s_ev["segment_start"], seglen)
564                     speed = self.render_rate(None, compute_rate(seglen, segtime))
565                     decode_time = self._rate_and_time(seglen, s_ev["decode_time"])
566                 else:
567                     # error
568                     range_s = "error"
569             else:
570                 # not finished yet
571                 pass
572
573             t[T.tr[T.td["seg%d" % s_ev["segment_number"]],
574                    T.td[srt(s_ev["start_time"])],
575                    T.td[srt(s_ev["active_time"])],
576                    T.td[srt(s_ev["finish_time"])],
577                    T.td[range_s],
578                    T.td[decode_time],
579                    T.td[segtime_s], T.td[speed]]]
580
581         l[T.h2["Segment Events:"], t]
582         l[T.br(clear="all")]
583         t = T.table(align="left",class_="status-download-events")
584         t[T.tr[T.th["serverid"], T.th["shnum"], T.th["range"],
585                T.th["txtime"], T.th["rxtime"],
586                T.th["received"], T.th["RTT"]]]
587         for r_ev in self.download_status.block_requests:
588             server = r_ev["server"]
589             rtt = None
590             if r_ev["finish_time"] is not None:
591                 rtt = r_ev["finish_time"] - r_ev["start_time"]
592             color = self.color(server.get_serverid())
593             t[T.tr(style="background: %s" % color)[
594                 T.td[server.get_name()], T.td[r_ev["shnum"]],
595                 T.td["[%d:+%d]" % (r_ev["start"], r_ev["length"])],
596                 T.td[srt(r_ev["start_time"])], T.td[srt(r_ev["finish_time"])],
597                 T.td[r_ev["response_length"] or ""],
598                 T.td[self.render_time(None, rtt)],
599                 ]]
600
601         l[T.h2["Requests:"], t]
602         l[T.br(clear="all")]
603
604         return l
605
606     def color(self, peerid):
607         def m(c):
608             return min(ord(c) / 2 + 0x80, 0xff)
609         return "#%02x%02x%02x" % (m(peerid[0]), m(peerid[1]), m(peerid[2]))
610
611     def render_results(self, ctx, data):
612         d = self.download_results()
613         def _got_results(results):
614             if results:
615                 return ctx.tag
616             return ""
617         d.addCallback(_got_results)
618         return d
619
620     def render_started(self, ctx, data):
621         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
622         started_s = time.strftime(TIME_FORMAT,
623                                   time.localtime(data.get_started()))
624         return started_s + " (%s)" % data.get_started()
625
626     def render_si(self, ctx, data):
627         si_s = base32.b2a_or_none(data.get_storage_index())
628         if si_s is None:
629             si_s = "(None)"
630         return si_s
631
632     def render_helper(self, ctx, data):
633         return {True: "Yes",
634                 False: "No"}[data.using_helper()]
635
636     def render_total_size(self, ctx, data):
637         size = data.get_size()
638         if size is None:
639             return "(unknown)"
640         return size
641
642     def render_progress(self, ctx, data):
643         progress = data.get_progress()
644         # TODO: make an ascii-art bar
645         return "%.1f%%" % (100.0 * progress)
646
647     def render_status(self, ctx, data):
648         return data.get_status()
649
650 class DownloadStatusTimelinePage(rend.Page):
651     docFactory = getxmlfile("download-status-timeline.xhtml")
652
653     def render_started(self, ctx, data):
654         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
655         started_s = time.strftime(TIME_FORMAT,
656                                   time.localtime(data.get_started()))
657         return started_s + " (%s)" % data.get_started()
658
659     def render_si(self, ctx, data):
660         si_s = base32.b2a_or_none(data.get_storage_index())
661         if si_s is None:
662             si_s = "(None)"
663         return si_s
664
665     def render_helper(self, ctx, data):
666         return {True: "Yes",
667                 False: "No"}[data.using_helper()]
668
669     def render_total_size(self, ctx, data):
670         size = data.get_size()
671         if size is None:
672             return "(unknown)"
673         return size
674
675     def render_progress(self, ctx, data):
676         progress = data.get_progress()
677         # TODO: make an ascii-art bar
678         return "%.1f%%" % (100.0 * progress)
679
680     def render_status(self, ctx, data):
681         return data.get_status()
682
683 class RetrieveStatusPage(rend.Page, RateAndTimeMixin):
684     docFactory = getxmlfile("retrieve-status.xhtml")
685
686     def __init__(self, data):
687         rend.Page.__init__(self, data)
688         self.retrieve_status = data
689
690     def render_started(self, ctx, data):
691         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
692         started_s = time.strftime(TIME_FORMAT,
693                                   time.localtime(data.get_started()))
694         return started_s
695
696     def render_si(self, ctx, data):
697         si_s = base32.b2a_or_none(data.get_storage_index())
698         if si_s is None:
699             si_s = "(None)"
700         return si_s
701
702     def render_helper(self, ctx, data):
703         return {True: "Yes",
704                 False: "No"}[data.using_helper()]
705
706     def render_current_size(self, ctx, data):
707         size = data.get_size()
708         if size is None:
709             size = "(unknown)"
710         return size
711
712     def render_progress(self, ctx, data):
713         progress = data.get_progress()
714         # TODO: make an ascii-art bar
715         return "%.1f%%" % (100.0 * progress)
716
717     def render_status(self, ctx, data):
718         return data.get_status()
719
720     def render_encoding(self, ctx, data):
721         k, n = data.get_encoding()
722         return ctx.tag["Encoding: %s of %s" % (k, n)]
723
724     def render_problems(self, ctx, data):
725         problems = data.get_problems()
726         if not problems:
727             return ""
728         l = T.ul()
729         for peerid in sorted(problems.keys()):
730             peerid_s = idlib.shortnodeid_b2a(peerid)
731             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
732         return ctx.tag["Server Problems:", l]
733
734     def _get_rate(self, data, name):
735         file_size = self.retrieve_status.get_size()
736         time = self.retrieve_status.timings.get(name)
737         return compute_rate(file_size, time)
738
739     def data_time_total(self, ctx, data):
740         return self.retrieve_status.timings.get("total")
741     def data_rate_total(self, ctx, data):
742         return self._get_rate(data, "total")
743
744     def data_time_fetch(self, ctx, data):
745         return self.retrieve_status.timings.get("fetch")
746     def data_rate_fetch(self, ctx, data):
747         return self._get_rate(data, "fetch")
748
749     def data_time_decode(self, ctx, data):
750         return self.retrieve_status.timings.get("decode")
751     def data_rate_decode(self, ctx, data):
752         return self._get_rate(data, "decode")
753
754     def data_time_decrypt(self, ctx, data):
755         return self.retrieve_status.timings.get("decrypt")
756     def data_rate_decrypt(self, ctx, data):
757         return self._get_rate(data, "decrypt")
758
759     def render_server_timings(self, ctx, data):
760         per_server = self.retrieve_status.timings.get("fetch_per_server")
761         if not per_server:
762             return ""
763         l = T.ul()
764         for server in sorted(per_server.keys(), key=lambda s: s.get_name()):
765             times_s = ", ".join([self.render_time(None, t)
766                                  for t in per_server[server]])
767             l[T.li["[%s]: %s" % (server.get_name(), times_s)]]
768         return T.li["Per-Server Fetch Response Times: ", l]
769
770
771 class PublishStatusPage(rend.Page, RateAndTimeMixin):
772     docFactory = getxmlfile("publish-status.xhtml")
773
774     def __init__(self, data):
775         rend.Page.__init__(self, data)
776         self.publish_status = data
777
778     def render_started(self, ctx, data):
779         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
780         started_s = time.strftime(TIME_FORMAT,
781                                   time.localtime(data.get_started()))
782         return started_s
783
784     def render_si(self, ctx, data):
785         si_s = base32.b2a_or_none(data.get_storage_index())
786         if si_s is None:
787             si_s = "(None)"
788         return si_s
789
790     def render_helper(self, ctx, data):
791         return {True: "Yes",
792                 False: "No"}[data.using_helper()]
793
794     def render_current_size(self, ctx, data):
795         size = data.get_size()
796         if size is None:
797             size = "(unknown)"
798         return size
799
800     def render_progress(self, ctx, data):
801         progress = data.get_progress()
802         # TODO: make an ascii-art bar
803         return "%.1f%%" % (100.0 * progress)
804
805     def render_status(self, ctx, data):
806         return data.get_status()
807
808     def render_encoding(self, ctx, data):
809         k, n = data.get_encoding()
810         return ctx.tag["Encoding: %s of %s" % (k, n)]
811
812     def render_sharemap(self, ctx, data):
813         servermap = data.get_servermap()
814         if servermap is None:
815             return ctx.tag["None"]
816         l = T.ul()
817         sharemap = servermap.make_sharemap()
818         for shnum in sorted(sharemap.keys()):
819             l[T.li["%d -> Placed on " % shnum,
820                    ", ".join(["[%s]" % server.get_name()
821                               for server in sharemap[shnum]])]]
822         return ctx.tag["Sharemap:", l]
823
824     def render_problems(self, ctx, data):
825         problems = data.get_problems()
826         if not problems:
827             return ""
828         l = T.ul()
829         # XXX: is this exercised? I don't think PublishStatus.problems is
830         # ever populated
831         for peerid in sorted(problems.keys()):
832             peerid_s = idlib.shortnodeid_b2a(peerid)
833             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
834         return ctx.tag["Server Problems:", l]
835
836     def _get_rate(self, data, name):
837         file_size = self.publish_status.get_size()
838         time = self.publish_status.timings.get(name)
839         return compute_rate(file_size, time)
840
841     def data_time_total(self, ctx, data):
842         return self.publish_status.timings.get("total")
843     def data_rate_total(self, ctx, data):
844         return self._get_rate(data, "total")
845
846     def data_time_setup(self, ctx, data):
847         return self.publish_status.timings.get("setup")
848
849     def data_time_encrypt(self, ctx, data):
850         return self.publish_status.timings.get("encrypt")
851     def data_rate_encrypt(self, ctx, data):
852         return self._get_rate(data, "encrypt")
853
854     def data_time_encode(self, ctx, data):
855         return self.publish_status.timings.get("encode")
856     def data_rate_encode(self, ctx, data):
857         return self._get_rate(data, "encode")
858
859     def data_time_pack(self, ctx, data):
860         return self.publish_status.timings.get("pack")
861     def data_rate_pack(self, ctx, data):
862         return self._get_rate(data, "pack")
863     def data_time_sign(self, ctx, data):
864         return self.publish_status.timings.get("sign")
865
866     def data_time_push(self, ctx, data):
867         return self.publish_status.timings.get("push")
868     def data_rate_push(self, ctx, data):
869         return self._get_rate(data, "push")
870
871     def render_server_timings(self, ctx, data):
872         per_server = self.publish_status.timings.get("send_per_server")
873         if not per_server:
874             return ""
875         l = T.ul()
876         for server in sorted(per_server.keys(), key=lambda s: s.get_name()):
877             times_s = ", ".join([self.render_time(None, t)
878                                  for t in per_server[server]])
879             l[T.li["[%s]: %s" % (server.get_name(), times_s)]]
880         return T.li["Per-Server Response Times: ", l]
881
882 class MapupdateStatusPage(rend.Page, RateAndTimeMixin):
883     docFactory = getxmlfile("map-update-status.xhtml")
884
885     def __init__(self, data):
886         rend.Page.__init__(self, data)
887         self.update_status = data
888
889     def render_started(self, ctx, data):
890         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
891         started_s = time.strftime(TIME_FORMAT,
892                                   time.localtime(data.get_started()))
893         return started_s
894
895     def render_finished(self, ctx, data):
896         when = data.get_finished()
897         if not when:
898             return "not yet"
899         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
900         started_s = time.strftime(TIME_FORMAT,
901                                   time.localtime(data.get_finished()))
902         return started_s
903
904     def render_si(self, ctx, data):
905         si_s = base32.b2a_or_none(data.get_storage_index())
906         if si_s is None:
907             si_s = "(None)"
908         return si_s
909
910     def render_helper(self, ctx, data):
911         return {True: "Yes",
912                 False: "No"}[data.using_helper()]
913
914     def render_progress(self, ctx, data):
915         progress = data.get_progress()
916         # TODO: make an ascii-art bar
917         return "%.1f%%" % (100.0 * progress)
918
919     def render_status(self, ctx, data):
920         return data.get_status()
921
922     def render_problems(self, ctx, data):
923         problems = data.problems
924         if not problems:
925             return ""
926         l = T.ul()
927         for peerid in sorted(problems.keys()):
928             peerid_s = idlib.shortnodeid_b2a(peerid)
929             l[T.li["[%s]: %s" % (peerid_s, problems[peerid])]]
930         return ctx.tag["Server Problems:", l]
931
932     def render_privkey_from(self, ctx, data):
933         server = data.get_privkey_from()
934         if server:
935             return ctx.tag["Got privkey from: [%s]" % server.get_name()]
936         else:
937             return ""
938
939     def data_time_total(self, ctx, data):
940         return self.update_status.timings.get("total")
941
942     def data_time_initial_queries(self, ctx, data):
943         return self.update_status.timings.get("initial_queries")
944
945     def data_time_cumulative_verify(self, ctx, data):
946         return self.update_status.timings.get("cumulative_verify")
947
948     def render_server_timings(self, ctx, data):
949         per_server = self.update_status.timings.get("per_server")
950         if not per_server:
951             return ""
952         l = T.ul()
953         for server in sorted(per_server.keys(), key=lambda s: s.get_name()):
954             times = []
955             for op,started,t in per_server[server]:
956                 #times.append("%s/%.4fs/%s/%s" % (op,
957                 #                              started,
958                 #                              self.render_time(None, started - self.update_status.get_started()),
959                 #                              self.render_time(None,t)))
960                 if op == "query":
961                     times.append( self.render_time(None, t) )
962                 elif op == "late":
963                     times.append( "late(" + self.render_time(None, t) + ")" )
964                 else:
965                     times.append( "privkey(" + self.render_time(None, t) + ")" )
966             times_s = ", ".join(times)
967             l[T.li["[%s]: %s" % (server.get_name(), times_s)]]
968         return T.li["Per-Server Response Times: ", l]
969
970     def render_timing_chart(self, ctx, data):
971         imageurl = self._timing_chart()
972         return ctx.tag[imageurl]
973
974     def _timing_chart(self):
975         started = self.update_status.get_started()
976         total = self.update_status.timings.get("total")
977         per_server = self.update_status.timings.get("per_server")
978         # We'd like to use an https: URL here, but the site has a domain/cert mismatch.
979         base = "http://chart.apis.google.com/chart?"
980         pieces = ["cht=bhs"]
981         pieces.append("chco=ffffff,4d89f9,c6d9fd") # colors
982         data0 = []
983         data1 = []
984         data2 = []
985         nb_nodes = 0
986         graph_botom_margin= 21
987         graph_top_margin = 5
988         server_names = []
989         top_abs = started
990         # we sort the queries by the time at which we sent the first request
991         sorttable = [ (times[0][1], server)
992                       for server, times in per_server.items() ]
993         sorttable.sort()
994         servers = [t[1] for t in sorttable]
995
996         for server in servers:
997             nb_nodes += 1
998             times = per_server[server]
999             name = server.get_name()
1000             server_names.append(name)
1001             # for servermap updates, there are either one or two queries per
1002             # peer. The second (if present) is to get the privkey.
1003             op,q_started,q_elapsed = times[0]
1004             data0.append("%.3f" % (q_started-started))
1005             data1.append("%.3f" % q_elapsed)
1006             top_abs = max(top_abs, q_started+q_elapsed)
1007             if len(times) > 1:
1008                 op,p_started,p_elapsed = times[0]
1009                 data2.append("%.3f" % p_elapsed)
1010                 top_abs = max(top_abs, p_started+p_elapsed)
1011             else:
1012                 data2.append("0.0")
1013         finished = self.update_status.get_finished()
1014         if finished:
1015             top_abs = max(top_abs, finished)
1016         top_rel = top_abs - started
1017         chs ="chs=400x%d" % ( (nb_nodes*28) + graph_top_margin + graph_botom_margin )
1018         chd = "chd=t:" + "|".join([",".join(data0),
1019                                    ",".join(data1),
1020                                    ",".join(data2)])
1021         pieces.append(chd)
1022         pieces.append(chs)
1023         chds = "chds=0,%0.3f" % top_rel
1024         pieces.append(chds)
1025         pieces.append("chxt=x,y")
1026         pieces.append("chxr=0,0.0,%0.3f" % top_rel)
1027         pieces.append("chxl=1:|" + "|".join(reversed(server_names)))
1028         # use up to 10 grid lines, at decimal multiples.
1029         # mathutil.next_power_of_k doesn't handle numbers smaller than one,
1030         # unfortunately.
1031         #pieces.append("chg="
1032
1033         if total is not None:
1034             finished_f = 1.0 * total / top_rel
1035             pieces.append("chm=r,FF0000,0,%0.3f,%0.3f" % (finished_f,
1036                                                           finished_f+0.01))
1037         url = base + "&".join(pieces)
1038         return T.img(src=url,border="1",align="right", float="right")
1039
1040
1041 class Status(rend.Page):
1042     docFactory = getxmlfile("status.xhtml")
1043     addSlash = True
1044
1045     def __init__(self, history):
1046         rend.Page.__init__(self, history)
1047         self.history = history
1048
1049     def renderHTTP(self, ctx):
1050         req = inevow.IRequest(ctx)
1051         t = get_arg(req, "t")
1052         if t == "json":
1053             return self.json(req)
1054         return rend.Page.renderHTTP(self, ctx)
1055
1056     def json(self, req):
1057         req.setHeader("content-type", "text/plain")
1058         data = {}
1059         data["active"] = active = []
1060         for s in self._get_active_operations():
1061             si_s = base32.b2a_or_none(s.get_storage_index())
1062             size = s.get_size()
1063             status = s.get_status()
1064             if IUploadStatus.providedBy(s):
1065                 h,c,e = s.get_progress()
1066                 active.append({"type": "upload",
1067                                "storage-index-string": si_s,
1068                                "total-size": size,
1069                                "status": status,
1070                                "progress-hash": h,
1071                                "progress-ciphertext": c,
1072                                "progress-encode-push": e,
1073                                })
1074             elif IDownloadStatus.providedBy(s):
1075                 active.append({"type": "download",
1076                                "storage-index-string": si_s,
1077                                "total-size": size,
1078                                "status": status,
1079                                "progress": s.get_progress(),
1080                                })
1081
1082         return simplejson.dumps(data, indent=1) + "\n"
1083
1084     def _get_all_statuses(self):
1085         h = self.history
1086         return itertools.chain(h.list_all_upload_statuses(),
1087                                h.list_all_download_statuses(),
1088                                h.list_all_mapupdate_statuses(),
1089                                h.list_all_publish_statuses(),
1090                                h.list_all_retrieve_statuses(),
1091                                h.list_all_helper_statuses(),
1092                                )
1093
1094     def data_active_operations(self, ctx, data):
1095         return self._get_active_operations()
1096
1097     def _get_active_operations(self):
1098         active = [s
1099                   for s in self._get_all_statuses()
1100                   if s.get_active()]
1101         return active
1102
1103     def data_recent_operations(self, ctx, data):
1104         return self._get_recent_operations()
1105
1106     def _get_recent_operations(self):
1107         recent = [s
1108                   for s in self._get_all_statuses()
1109                   if not s.get_active()]
1110         recent.sort(lambda a,b: cmp(a.get_started(), b.get_started()))
1111         recent.reverse()
1112         return recent
1113
1114     def render_row(self, ctx, data):
1115         s = data
1116
1117         TIME_FORMAT = "%H:%M:%S %d-%b-%Y"
1118         started_s = time.strftime(TIME_FORMAT,
1119                                   time.localtime(s.get_started()))
1120         ctx.fillSlots("started", started_s)
1121
1122         si_s = base32.b2a_or_none(s.get_storage_index())
1123         if si_s is None:
1124             si_s = "(None)"
1125         ctx.fillSlots("si", si_s)
1126         ctx.fillSlots("helper", {True: "Yes",
1127                                  False: "No"}[s.using_helper()])
1128
1129         size = s.get_size()
1130         if size is None:
1131             size = "(unknown)"
1132         elif isinstance(size, (int, long, float)):
1133             size = abbreviate_size(size)
1134         ctx.fillSlots("total_size", size)
1135
1136         progress = data.get_progress()
1137         if IUploadStatus.providedBy(data):
1138             link = "up-%d" % data.get_counter()
1139             ctx.fillSlots("type", "upload")
1140             # TODO: make an ascii-art bar
1141             (chk, ciphertext, encandpush) = progress
1142             progress_s = ("hash: %.1f%%, ciphertext: %.1f%%, encode: %.1f%%" %
1143                           ( (100.0 * chk),
1144                             (100.0 * ciphertext),
1145                             (100.0 * encandpush) ))
1146             ctx.fillSlots("progress", progress_s)
1147         elif IDownloadStatus.providedBy(data):
1148             link = "down-%d" % data.get_counter()
1149             ctx.fillSlots("type", "download")
1150             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1151         elif IPublishStatus.providedBy(data):
1152             link = "publish-%d" % data.get_counter()
1153             ctx.fillSlots("type", "publish")
1154             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1155         elif IRetrieveStatus.providedBy(data):
1156             ctx.fillSlots("type", "retrieve")
1157             link = "retrieve-%d" % data.get_counter()
1158             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1159         else:
1160             assert IServermapUpdaterStatus.providedBy(data)
1161             ctx.fillSlots("type", "mapupdate %s" % data.get_mode())
1162             link = "mapupdate-%d" % data.get_counter()
1163             ctx.fillSlots("progress", "%.1f%%" % (100.0 * progress))
1164         ctx.fillSlots("status", T.a(href=link)[s.get_status()])
1165         return ctx.tag
1166
1167     def childFactory(self, ctx, name):
1168         h = self.history
1169         stype,count_s = name.split("-")
1170         count = int(count_s)
1171         if stype == "up":
1172             for s in itertools.chain(h.list_all_upload_statuses(),
1173                                      h.list_all_helper_statuses()):
1174                 # immutable-upload helpers use the same status object as a
1175                 # regular immutable-upload
1176                 if s.get_counter() == count:
1177                     return UploadStatusPage(s)
1178         if stype == "down":
1179             for s in h.list_all_download_statuses():
1180                 if s.get_counter() == count:
1181                     return DownloadStatusPage(s)
1182         if stype == "mapupdate":
1183             for s in h.list_all_mapupdate_statuses():
1184                 if s.get_counter() == count:
1185                     return MapupdateStatusPage(s)
1186         if stype == "publish":
1187             for s in h.list_all_publish_statuses():
1188                 if s.get_counter() == count:
1189                     return PublishStatusPage(s)
1190         if stype == "retrieve":
1191             for s in h.list_all_retrieve_statuses():
1192                 if s.get_counter() == count:
1193                     return RetrieveStatusPage(s)
1194
1195
1196 class HelperStatus(rend.Page):
1197     docFactory = getxmlfile("helper.xhtml")
1198
1199     def __init__(self, helper):
1200         rend.Page.__init__(self, helper)
1201         self.helper = helper
1202
1203     def renderHTTP(self, ctx):
1204         req = inevow.IRequest(ctx)
1205         t = get_arg(req, "t")
1206         if t == "json":
1207             return self.render_JSON(req)
1208         return rend.Page.renderHTTP(self, ctx)
1209
1210     def data_helper_stats(self, ctx, data):
1211         return self.helper.get_stats()
1212
1213     def render_JSON(self, req):
1214         req.setHeader("content-type", "text/plain")
1215         if self.helper:
1216             stats = self.helper.get_stats()
1217             return simplejson.dumps(stats, indent=1) + "\n"
1218         return simplejson.dumps({}) + "\n"
1219
1220     def render_active_uploads(self, ctx, data):
1221         return data["chk_upload_helper.active_uploads"]
1222
1223     def render_incoming(self, ctx, data):
1224         return "%d bytes in %d files" % (data["chk_upload_helper.incoming_size"],
1225                                          data["chk_upload_helper.incoming_count"])
1226
1227     def render_encoding(self, ctx, data):
1228         return "%d bytes in %d files" % (data["chk_upload_helper.encoding_size"],
1229                                          data["chk_upload_helper.encoding_count"])
1230
1231     def render_upload_requests(self, ctx, data):
1232         return str(data["chk_upload_helper.upload_requests"])
1233
1234     def render_upload_already_present(self, ctx, data):
1235         return str(data["chk_upload_helper.upload_already_present"])
1236
1237     def render_upload_need_upload(self, ctx, data):
1238         return str(data["chk_upload_helper.upload_need_upload"])
1239
1240     def render_upload_bytes_fetched(self, ctx, data):
1241         return str(data["chk_upload_helper.fetched_bytes"])
1242
1243     def render_upload_bytes_encoded(self, ctx, data):
1244         return str(data["chk_upload_helper.encoded_bytes"])
1245
1246
1247 class Statistics(rend.Page):
1248     docFactory = getxmlfile("statistics.xhtml")
1249
1250     def __init__(self, provider):
1251         rend.Page.__init__(self, provider)
1252         self.provider = provider
1253
1254     def renderHTTP(self, ctx):
1255         req = inevow.IRequest(ctx)
1256         t = get_arg(req, "t")
1257         if t == "json":
1258             stats = self.provider.get_stats()
1259             req.setHeader("content-type", "text/plain")
1260             return simplejson.dumps(stats, indent=1) + "\n"
1261         return rend.Page.renderHTTP(self, ctx)
1262
1263     def data_get_stats(self, ctx, data):
1264         return self.provider.get_stats()
1265
1266     def render_load_average(self, ctx, data):
1267         return str(data["stats"].get("load_monitor.avg_load"))
1268
1269     def render_peak_load(self, ctx, data):
1270         return str(data["stats"].get("load_monitor.max_load"))
1271
1272     def render_uploads(self, ctx, data):
1273         files = data["counters"].get("uploader.files_uploaded", 0)
1274         bytes = data["counters"].get("uploader.bytes_uploaded", 0)
1275         return ("%s files / %s bytes (%s)" %
1276                 (files, bytes, abbreviate_size(bytes)))
1277
1278     def render_downloads(self, ctx, data):
1279         files = data["counters"].get("downloader.files_downloaded", 0)
1280         bytes = data["counters"].get("downloader.bytes_downloaded", 0)
1281         return ("%s files / %s bytes (%s)" %
1282                 (files, bytes, abbreviate_size(bytes)))
1283
1284     def render_publishes(self, ctx, data):
1285         files = data["counters"].get("mutable.files_published", 0)
1286         bytes = data["counters"].get("mutable.bytes_published", 0)
1287         return "%s files / %s bytes (%s)" % (files, bytes,
1288                                              abbreviate_size(bytes))
1289
1290     def render_retrieves(self, ctx, data):
1291         files = data["counters"].get("mutable.files_retrieved", 0)
1292         bytes = data["counters"].get("mutable.bytes_retrieved", 0)
1293         return "%s files / %s bytes (%s)" % (files, bytes,
1294                                              abbreviate_size(bytes))
1295
1296     def render_drop_monitored(self, ctx, data):
1297         dirs = data["counters"].get("drop_upload.dirs_monitored", 0)
1298         return "%s directories" % (dirs,)
1299
1300     def render_drop_uploads(self, ctx, data):
1301         # TODO: bytes uploaded
1302         files = data["counters"].get("drop_upload.files_uploaded", 0)
1303         return "%s files" % (files,)
1304
1305     def render_drop_queued(self, ctx, data):
1306         files = data["counters"].get("drop_upload.files_queued", 0)
1307         return "%s files" % (files,)
1308
1309     def render_drop_failed(self, ctx, data):
1310         files = data["counters"].get("drop_upload.files_failed", 0)
1311         return "%s files" % (files,)
1312
1313     def render_raw(self, ctx, data):
1314         raw = pprint.pformat(data)
1315         return ctx.tag[raw]