mercurial-scm/hg-stable: hgext/zeroconf/Zeroconf.py annotate

annotate hgext/zeroconf/Zeroconf.py @ 28296:a73394e7b47c

zeroconf: use absolute_import

author	timeless <timeless@mozdev.org>
date	Tue, 01 Mar 2016 08:42:46 +0000
parents	01815c159856
children	f6d07520c665

rev	line source
28296 a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	1 from __future__ import absolute_import, print_function
28295 01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	2
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	3 """ Multicast DNS Service Discovery for Python, v0.12
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	4 Copyright (C) 2003, Paul Scott-Murphy
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	5
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	6 This module provides a framework for the use of DNS Service Discovery
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	7 using IP multicast. It has been tested against the JRendezvous
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	8 implementation from <a href="http://strangeberry.com">StrangeBerry</a>,
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	9 and against the mDNSResponder from Mac OS X 10.3.8.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	10
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	11 This library is free software; you can redistribute it and/or
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	12 modify it under the terms of the GNU Lesser General Public
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	13 License as published by the Free Software Foundation; either
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	14 version 2.1 of the License, or (at your option) any later version.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	15
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	16 This library is distributed in the hope that it will be useful,
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	17 but WITHOUT ANY WARRANTY; without even the implied warranty of
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	19 Lesser General Public License for more details.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	20
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	21 You should have received a copy of the GNU Lesser General Public
15782 7de7630053cb Remove FSF mailing address from GPL headers Martin Geisler <mg@aragost.com> parents: 14494 diff changeset	22 License along with this library; if not, see
7de7630053cb Remove FSF mailing address from GPL headers Martin Geisler <mg@aragost.com> parents: 14494 diff changeset	23 <http://www.gnu.org/licenses/>.
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	24
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	25 """
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	26
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	27 """0.12 update - allow selection of binding interface
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	28 typo fix - Thanks A. M. Kuchlingi
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	29 removed all use of word 'Rendezvous' - this is an API change"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	30
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	31 """0.11 update - correction to comments for addListener method
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	32 support for new record types seen from OS X
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	33 - IPv6 address
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	34 - hostinfo
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	35 ignore unknown DNS record types
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	36 fixes to name decoding
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	37 works alongside other processes using port 5353 (e.g. on Mac OS X)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	38 tested against Mac OS X 10.3.2's mDNSResponder
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	39 corrections to removal of list entries for service browser"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	40
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	41 """0.10 update - Jonathon Paisley contributed these corrections:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	42 always multicast replies, even when query is unicast
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	43 correct a pointer encoding problem
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	44 can now write records in any order
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	45 traceback shown on failure
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	46 better TXT record parsing
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	47 server is now separate from name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	48 can cancel a service browser
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	49
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	50 modified some unit tests to accommodate these changes"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	51
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	52 """0.09 update - remove all records on service unregistration
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	53 fix DOS security problem with readName"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	54
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	55 """0.08 update - changed licensing to LGPL"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	56
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	57 """0.07 update - faster shutdown on engine
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	58 pointer encoding of outgoing names
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	59 ServiceBrowser now works
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	60 new unit tests"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	61
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	62 """0.06 update - small improvements with unit tests
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	63 added defined exception types
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	64 new style objects
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	65 fixed hostname/interface problem
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	66 fixed socket timeout problem
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	67 fixed addServiceListener() typo bug
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	68 using select() for socket reads
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	69 tested on Debian unstable with Python 2.2.2"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	70
17424 e7cfe3587ea4 fix trivial spelling errors Mads Kiilerich <mads@kiilerich.com> parents: 15782 diff changeset	71 """0.05 update - ensure case insensitivity on domain names
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	72 support for unicast DNS queries"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	73
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	74 """0.04 update - added some unit tests
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	75 added __ne__ adjuncts where required
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	76 ensure names end in '.local.'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	77 timeout on receiving socket for clean shutdown"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	78
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	79 __author__ = "Paul Scott-Murphy"
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	80 __email__ = "paul at scott dash murphy dot com"
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	81 __version__ = "0.12"
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	82
28296 a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	83 import select
a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	84 import socket
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	85 import string
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	86 import struct
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	87 import threading
28296 a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	88 import time
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	89 import traceback
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	90
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	91 __all__ = ["Zeroconf", "ServiceInfo", "ServiceBrowser"]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	92
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	93 # hook for threads
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	94
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	95 globals()['_GLOBAL_DONE'] = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	96
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	97 # Some timing constants
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	98
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	99 _UNREGISTER_TIME = 125
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	100 _CHECK_TIME = 175
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	101 _REGISTER_TIME = 225
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	102 _LISTENER_TIME = 200
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	103 _BROWSER_TIME = 500
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	104
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	105 # Some DNS constants
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	106
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	107 _MDNS_ADDR = '224.0.0.251'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	108 _MDNS_PORT = 5353;
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	109 _DNS_PORT = 53;
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	110 _DNS_TTL = 60 * 60; # one hour default TTL
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	111
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	112 _MAX_MSG_TYPICAL = 1460 # unused
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	113 _MAX_MSG_ABSOLUTE = 8972
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	114
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	115 _FLAGS_QR_MASK = 0x8000 # query response mask
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	116 _FLAGS_QR_QUERY = 0x0000 # query
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	117 _FLAGS_QR_RESPONSE = 0x8000 # response
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	118
17476 210acc861c40 spelling: Authoritative timeless@mozdev.org parents: 15782 diff changeset	119 _FLAGS_AA = 0x0400 # Authoritative answer
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	120 _FLAGS_TC = 0x0200 # Truncated
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	121 _FLAGS_RD = 0x0100 # Recursion desired
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	122 _FLAGS_RA = 0x8000 # Recursion available
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	123
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	124 _FLAGS_Z = 0x0040 # Zero
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	125 _FLAGS_AD = 0x0020 # Authentic data
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	126 _FLAGS_CD = 0x0010 # Checking disabled
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	127
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	128 _CLASS_IN = 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	129 _CLASS_CS = 2
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	130 _CLASS_CH = 3
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	131 _CLASS_HS = 4
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	132 _CLASS_NONE = 254
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	133 _CLASS_ANY = 255
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	134 _CLASS_MASK = 0x7FFF
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	135 _CLASS_UNIQUE = 0x8000
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	136
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	137 _TYPE_A = 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	138 _TYPE_NS = 2
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	139 _TYPE_MD = 3
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	140 _TYPE_MF = 4
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	141 _TYPE_CNAME = 5
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	142 _TYPE_SOA = 6
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	143 _TYPE_MB = 7
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	144 _TYPE_MG = 8
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	145 _TYPE_MR = 9
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	146 _TYPE_NULL = 10
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	147 _TYPE_WKS = 11
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	148 _TYPE_PTR = 12
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	149 _TYPE_HINFO = 13
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	150 _TYPE_MINFO = 14
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	151 _TYPE_MX = 15
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	152 _TYPE_TXT = 16
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	153 _TYPE_AAAA = 28
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	154 _TYPE_SRV = 33
27637 b502138f5faa cleanup: remove superfluous space after space after equals (python) timeless <timeless@mozdev.org> parents: 17537 diff changeset	155 _TYPE_ANY = 255
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	156
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	157 # Mapping constants to names
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	158
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	159 _CLASSES = { _CLASS_IN : "in",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	160 _CLASS_CS : "cs",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	161 _CLASS_CH : "ch",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	162 _CLASS_HS : "hs",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	163 _CLASS_NONE : "none",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	164 _CLASS_ANY : "any" }
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	165
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	166 _TYPES = { _TYPE_A : "a",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	167 _TYPE_NS : "ns",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	168 _TYPE_MD : "md",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	169 _TYPE_MF : "mf",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	170 _TYPE_CNAME : "cname",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	171 _TYPE_SOA : "soa",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	172 _TYPE_MB : "mb",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	173 _TYPE_MG : "mg",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	174 _TYPE_MR : "mr",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	175 _TYPE_NULL : "null",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	176 _TYPE_WKS : "wks",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	177 _TYPE_PTR : "ptr",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	178 _TYPE_HINFO : "hinfo",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	179 _TYPE_MINFO : "minfo",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	180 _TYPE_MX : "mx",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	181 _TYPE_TXT : "txt",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	182 _TYPE_AAAA : "quada",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	183 _TYPE_SRV : "srv",
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	184 _TYPE_ANY : "any" }
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	185
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	186 # utility functions
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	187
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	188 def currentTimeMillis():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	189 """Current system time in milliseconds"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	190 return time.time() * 1000
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	191
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	192 # Exceptions
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	193
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	194 class NonLocalNameException(Exception):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	195 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	196
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	197 class NonUniqueNameException(Exception):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	198 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	199
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	200 class NamePartTooLongException(Exception):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	201 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	202
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	203 class AbstractMethodException(Exception):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	204 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	205
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	206 class BadTypeInNameException(Exception):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	207 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	208
11435 7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	209 class BadDomainName(Exception):
7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	210 def __init__(self, pos):
7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	211 Exception.__init__(self, "at position %s" % pos)
7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	212
7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	213 class BadDomainNameCircular(BadDomainName):
7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	214 pass
7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	215
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	216 # implementation classes
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	217
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	218 class DNSEntry(object):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	219 """A DNS entry"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	220
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	221 def __init__(self, name, type, clazz):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	222 self.key = string.lower(name)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	223 self.name = name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	224 self.type = type
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	225 self.clazz = clazz & _CLASS_MASK
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	226 self.unique = (clazz & _CLASS_UNIQUE) != 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	227
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	228 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	229 """Equality test on name, type, and class"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	230 if isinstance(other, DNSEntry):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	231 return self.name == other.name and self.type == other.type and self.clazz == other.clazz
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	232 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	233
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	234 def __ne__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	235 """Non-equality test"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	236 return not self.__eq__(other)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	237
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	238 def getClazz(self, clazz):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	239 """Class accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	240 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	241 return _CLASSES[clazz]
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	242 except KeyError:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	243 return "?(%s)" % (clazz)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	244
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	245 def getType(self, type):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	246 """Type accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	247 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	248 return _TYPES[type]
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	249 except KeyError:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	250 return "?(%s)" % (type)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	251
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	252 def toString(self, hdr, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	253 """String representation with additional information"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	254 result = "%s[%s,%s" % (hdr, self.getType(self.type), self.getClazz(self.clazz))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	255 if self.unique:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	256 result += "-unique,"
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	257 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	258 result += ","
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	259 result += self.name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	260 if other is not None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	261 result += ",%s]" % (other)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	262 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	263 result += "]"
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	264 return result
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	265
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	266 class DNSQuestion(DNSEntry):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	267 """A DNS question entry"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	268
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	269 def __init__(self, name, type, clazz):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	270 if not name.endswith(".local."):
8131 b0d945b95105 zeroconf: don't fail on non-local names Alexander Solovyov <piranha@piranha.org.ua> parents: 7877 diff changeset	271 raise NonLocalNameException(name)
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	272 DNSEntry.__init__(self, name, type, clazz)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	273
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	274 def answeredBy(self, rec):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	275 """Returns true if the question is answered by the record"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	276 return self.clazz == rec.clazz and (self.type == rec.type or self.type == _TYPE_ANY) and self.name == rec.name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	277
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	278 def __repr__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	279 """String representation"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	280 return DNSEntry.toString(self, "question", None)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	281
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	282
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	283 class DNSRecord(DNSEntry):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	284 """A DNS record - like a DNS entry, but has a TTL"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	285
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	286 def __init__(self, name, type, clazz, ttl):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	287 DNSEntry.__init__(self, name, type, clazz)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	288 self.ttl = ttl
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	289 self.created = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	290
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	291 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	292 """Tests equality as per DNSRecord"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	293 if isinstance(other, DNSRecord):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	294 return DNSEntry.__eq__(self, other)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	295 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	296
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	297 def suppressedBy(self, msg):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	298 """Returns true if any answer in a message can suffice for the
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	299 information held in this record."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	300 for record in msg.answers:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	301 if self.suppressedByAnswer(record):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	302 return 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	303 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	304
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	305 def suppressedByAnswer(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	306 """Returns true if another record has same name, type and class,
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	307 and if its TTL is at least half of this record's."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	308 if self == other and other.ttl > (self.ttl / 2):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	309 return 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	310 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	311
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	312 def getExpirationTime(self, percent):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	313 """Returns the time at which this record will have expired
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	314 by a certain percentage."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	315 return self.created + (percent * self.ttl * 10)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	316
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	317 def getRemainingTTL(self, now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	318 """Returns the remaining TTL in seconds."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	319 return max(0, (self.getExpirationTime(100) - now) / 1000)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	320
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	321 def isExpired(self, now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	322 """Returns true if this record has expired."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	323 return self.getExpirationTime(100) <= now
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	324
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	325 def isStale(self, now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	326 """Returns true if this record is at least half way expired."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	327 return self.getExpirationTime(50) <= now
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	328
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	329 def resetTTL(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	330 """Sets this record's TTL and created time to that of
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	331 another record."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	332 self.created = other.created
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	333 self.ttl = other.ttl
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	334
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	335 def write(self, out):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	336 """Abstract method"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	337 raise AbstractMethodException
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	338
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	339 def toString(self, other):
17424 e7cfe3587ea4 fix trivial spelling errors Mads Kiilerich <mads@kiilerich.com> parents: 15782 diff changeset	340 """String representation with additional information"""
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	341 arg = "%s/%s,%s" % (self.ttl, self.getRemainingTTL(currentTimeMillis()), other)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	342 return DNSEntry.toString(self, "record", arg)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	343
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	344 class DNSAddress(DNSRecord):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	345 """A DNS address record"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	346
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	347 def __init__(self, name, type, clazz, ttl, address):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	348 DNSRecord.__init__(self, name, type, clazz, ttl)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	349 self.address = address
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	350
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	351 def write(self, out):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	352 """Used in constructing an outgoing packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	353 out.writeString(self.address, len(self.address))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	354
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	355 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	356 """Tests equality on address"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	357 if isinstance(other, DNSAddress):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	358 return self.address == other.address
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	359 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	360
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	361 def __repr__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	362 """String representation"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	363 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	364 return socket.inet_ntoa(self.address)
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	365 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	366 return self.address
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	367
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	368 class DNSHinfo(DNSRecord):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	369 """A DNS host information record"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	370
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	371 def __init__(self, name, type, clazz, ttl, cpu, os):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	372 DNSRecord.__init__(self, name, type, clazz, ttl)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	373 self.cpu = cpu
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	374 self.os = os
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	375
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	376 def write(self, out):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	377 """Used in constructing an outgoing packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	378 out.writeString(self.cpu, len(self.cpu))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	379 out.writeString(self.os, len(self.os))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	380
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	381 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	382 """Tests equality on cpu and os"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	383 if isinstance(other, DNSHinfo):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	384 return self.cpu == other.cpu and self.os == other.os
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	385 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	386
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	387 def __repr__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	388 """String representation"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	389 return self.cpu + " " + self.os
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	390
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	391 class DNSPointer(DNSRecord):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	392 """A DNS pointer record"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	393
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	394 def __init__(self, name, type, clazz, ttl, alias):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	395 DNSRecord.__init__(self, name, type, clazz, ttl)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	396 self.alias = alias
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	397
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	398 def write(self, out):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	399 """Used in constructing an outgoing packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	400 out.writeName(self.alias)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	401
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	402 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	403 """Tests equality on alias"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	404 if isinstance(other, DNSPointer):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	405 return self.alias == other.alias
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	406 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	407
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	408 def __repr__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	409 """String representation"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	410 return self.toString(self.alias)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	411
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	412 class DNSText(DNSRecord):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	413 """A DNS text record"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	414
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	415 def __init__(self, name, type, clazz, ttl, text):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	416 DNSRecord.__init__(self, name, type, clazz, ttl)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	417 self.text = text
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	418
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	419 def write(self, out):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	420 """Used in constructing an outgoing packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	421 out.writeString(self.text, len(self.text))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	422
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	423 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	424 """Tests equality on text"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	425 if isinstance(other, DNSText):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	426 return self.text == other.text
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	427 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	428
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	429 def __repr__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	430 """String representation"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	431 if len(self.text) > 10:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	432 return self.toString(self.text[:7] + "...")
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	433 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	434 return self.toString(self.text)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	435
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	436 class DNSService(DNSRecord):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	437 """A DNS service record"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	438
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	439 def __init__(self, name, type, clazz, ttl, priority, weight, port, server):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	440 DNSRecord.__init__(self, name, type, clazz, ttl)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	441 self.priority = priority
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	442 self.weight = weight
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	443 self.port = port
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	444 self.server = server
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	445
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	446 def write(self, out):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	447 """Used in constructing an outgoing packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	448 out.writeShort(self.priority)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	449 out.writeShort(self.weight)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	450 out.writeShort(self.port)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	451 out.writeName(self.server)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	452
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	453 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	454 """Tests equality on priority, weight, port and server"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	455 if isinstance(other, DNSService):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	456 return self.priority == other.priority and self.weight == other.weight and self.port == other.port and self.server == other.server
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	457 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	458
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	459 def __repr__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	460 """String representation"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	461 return self.toString("%s:%s" % (self.server, self.port))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	462
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	463 class DNSIncoming(object):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	464 """Object representation of an incoming DNS packet"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	465
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	466 def __init__(self, data):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	467 """Constructor from string holding bytes of packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	468 self.offset = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	469 self.data = data
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	470 self.questions = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	471 self.answers = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	472 self.numQuestions = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	473 self.numAnswers = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	474 self.numAuthorities = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	475 self.numAdditionals = 0
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	476
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	477 self.readHeader()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	478 self.readQuestions()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	479 self.readOthers()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	480
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	481 def readHeader(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	482 """Reads header portion of packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	483 format = '!HHHHHH'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	484 length = struct.calcsize(format)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	485 info = struct.unpack(format, self.data[self.offset:self.offset+length])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	486 self.offset += length
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	487
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	488 self.id = info[0]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	489 self.flags = info[1]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	490 self.numQuestions = info[2]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	491 self.numAnswers = info[3]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	492 self.numAuthorities = info[4]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	493 self.numAdditionals = info[5]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	494
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	495 def readQuestions(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	496 """Reads questions section of packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	497 format = '!HH'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	498 length = struct.calcsize(format)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	499 for i in range(0, self.numQuestions):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	500 name = self.readName()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	501 info = struct.unpack(format, self.data[self.offset:self.offset+length])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	502 self.offset += length
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	503
8131 b0d945b95105 zeroconf: don't fail on non-local names Alexander Solovyov <piranha@piranha.org.ua> parents: 7877 diff changeset	504 try:
b0d945b95105 zeroconf: don't fail on non-local names Alexander Solovyov <piranha@piranha.org.ua> parents: 7877 diff changeset	505 question = DNSQuestion(name, info[0], info[1])
b0d945b95105 zeroconf: don't fail on non-local names Alexander Solovyov <piranha@piranha.org.ua> parents: 7877 diff changeset	506 self.questions.append(question)
b0d945b95105 zeroconf: don't fail on non-local names Alexander Solovyov <piranha@piranha.org.ua> parents: 7877 diff changeset	507 except NonLocalNameException:
b0d945b95105 zeroconf: don't fail on non-local names Alexander Solovyov <piranha@piranha.org.ua> parents: 7877 diff changeset	508 pass
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	509
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	510 def readInt(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	511 """Reads an integer from the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	512 format = '!I'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	513 length = struct.calcsize(format)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	514 info = struct.unpack(format, self.data[self.offset:self.offset+length])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	515 self.offset += length
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	516 return info[0]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	517
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	518 def readCharacterString(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	519 """Reads a character string from the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	520 length = ord(self.data[self.offset])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	521 self.offset += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	522 return self.readString(length)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	523
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	524 def readString(self, len):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	525 """Reads a string of a given length from the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	526 format = '!' + str(len) + 's'
27637 b502138f5faa cleanup: remove superfluous space after space after equals (python) timeless <timeless@mozdev.org> parents: 17537 diff changeset	527 length = struct.calcsize(format)
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	528 info = struct.unpack(format, self.data[self.offset:self.offset+length])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	529 self.offset += length
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	530 return info[0]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	531
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	532 def readUnsignedShort(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	533 """Reads an unsigned short from the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	534 format = '!H'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	535 length = struct.calcsize(format)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	536 info = struct.unpack(format, self.data[self.offset:self.offset+length])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	537 self.offset += length
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	538 return info[0]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	539
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	540 def readOthers(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	541 """Reads the answers, authorities and additionals section of the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	542 format = '!HHiH'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	543 length = struct.calcsize(format)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	544 n = self.numAnswers + self.numAuthorities + self.numAdditionals
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	545 for i in range(0, n):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	546 domain = self.readName()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	547 info = struct.unpack(format, self.data[self.offset:self.offset+length])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	548 self.offset += length
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	549
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	550 rec = None
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	551 if info[0] == _TYPE_A:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	552 rec = DNSAddress(domain, info[0], info[1], info[2], self.readString(4))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	553 elif info[0] == _TYPE_CNAME or info[0] == _TYPE_PTR:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	554 rec = DNSPointer(domain, info[0], info[1], info[2], self.readName())
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	555 elif info[0] == _TYPE_TXT:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	556 rec = DNSText(domain, info[0], info[1], info[2], self.readString(info[3]))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	557 elif info[0] == _TYPE_SRV:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	558 rec = DNSService(domain, info[0], info[1], info[2], self.readUnsignedShort(), self.readUnsignedShort(), self.readUnsignedShort(), self.readName())
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	559 elif info[0] == _TYPE_HINFO:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	560 rec = DNSHinfo(domain, info[0], info[1], info[2], self.readCharacterString(), self.readCharacterString())
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	561 elif info[0] == _TYPE_AAAA:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	562 rec = DNSAddress(domain, info[0], info[1], info[2], self.readString(16))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	563 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	564 # Try to ignore types we don't know about
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	565 # this may mean the rest of the name is
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	566 # unable to be parsed, and may show errors
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	567 # so this is left for debugging. New types
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	568 # encountered need to be parsed properly.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	569 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	570 #print "UNKNOWN TYPE = " + str(info[0])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	571 #raise BadTypeInNameException
10387 9b87c5f4c634 zeroconf: use DNS length field to safely skip unknown record types Brendan Cully <brendan@kublai.com> parents: 10386 diff changeset	572 self.offset += info[3]
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	573
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	574 if rec is not None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	575 self.answers.append(rec)
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	576
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	577 def isQuery(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	578 """Returns true if this is a query"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	579 return (self.flags & _FLAGS_QR_MASK) == _FLAGS_QR_QUERY
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	580
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	581 def isResponse(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	582 """Returns true if this is a response"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	583 return (self.flags & _FLAGS_QR_MASK) == _FLAGS_QR_RESPONSE
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	584
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	585 def readUTF(self, offset, len):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	586 """Reads a UTF-8 string of a given length from the packet"""
10386 1ddb0ae26a55 zeroconf: better fix for readName error Brendan Cully <brendan@kublai.com> parents: 10385 diff changeset	587 return self.data[offset:offset+len].decode('utf-8')
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	588
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	589 def readName(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	590 """Reads a domain name from the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	591 result = ''
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	592 off = self.offset
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	593 next = -1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	594 first = off
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	595
14494 1ffeeb91c55d check-code: flag 0/1 used as constant Boolean expression Martin Geisler <mg@lazybytes.net> parents: 14063 diff changeset	596 while True:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	597 len = ord(self.data[off])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	598 off += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	599 if len == 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	600 break
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	601 t = len & 0xC0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	602 if t == 0x00:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	603 result = ''.join((result, self.readUTF(off, len) + '.'))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	604 off += len
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	605 elif t == 0xC0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	606 if next < 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	607 next = off + 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	608 off = ((len & 0x3F) << 8) \| ord(self.data[off])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	609 if off >= first:
11435 7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	610 raise BadDomainNameCircular(off)
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	611 first = off
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	612 else:
11435 7c58cde598fe zeroconf: Use BadDomainName exception instead of string exceptions Javi Merino <cibervicho@gmail.com> parents: 10514 diff changeset	613 raise BadDomainName(off)
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	614
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	615 if next >= 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	616 self.offset = next
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	617 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	618 self.offset = off
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	619
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	620 return result
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	621
eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	622
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	623 class DNSOutgoing(object):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	624 """Object representation of an outgoing packet"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	625
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	626 def __init__(self, flags, multicast = 1):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	627 self.finished = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	628 self.id = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	629 self.multicast = multicast
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	630 self.flags = flags
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	631 self.names = {}
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	632 self.data = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	633 self.size = 12
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	634
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	635 self.questions = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	636 self.answers = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	637 self.authorities = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	638 self.additionals = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	639
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	640 def addQuestion(self, record):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	641 """Adds a question"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	642 self.questions.append(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	643
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	644 def addAnswer(self, inp, record):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	645 """Adds an answer"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	646 if not record.suppressedBy(inp):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	647 self.addAnswerAtTime(record, 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	648
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	649 def addAnswerAtTime(self, record, now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	650 """Adds an answer if if does not expire by a certain time"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	651 if record is not None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	652 if now == 0 or not record.isExpired(now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	653 self.answers.append((record, now))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	654
17476 210acc861c40 spelling: Authoritative timeless@mozdev.org parents: 15782 diff changeset	655 def addAuthoritativeAnswer(self, record):
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	656 """Adds an authoritative answer"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	657 self.authorities.append(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	658
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	659 def addAdditionalAnswer(self, record):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	660 """Adds an additional answer"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	661 self.additionals.append(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	662
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	663 def writeByte(self, value):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	664 """Writes a single byte to the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	665 format = '!c'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	666 self.data.append(struct.pack(format, chr(value)))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	667 self.size += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	668
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	669 def insertShort(self, index, value):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	670 """Inserts an unsigned short in a certain position in the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	671 format = '!H'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	672 self.data.insert(index, struct.pack(format, value))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	673 self.size += 2
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	674
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	675 def writeShort(self, value):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	676 """Writes an unsigned short to the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	677 format = '!H'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	678 self.data.append(struct.pack(format, value))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	679 self.size += 2
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	680
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	681 def writeInt(self, value):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	682 """Writes an unsigned integer to the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	683 format = '!I'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	684 self.data.append(struct.pack(format, int(value)))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	685 self.size += 4
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	686
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	687 def writeString(self, value, length):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	688 """Writes a string to the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	689 format = '!' + str(length) + 's'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	690 self.data.append(struct.pack(format, value))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	691 self.size += length
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	692
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	693 def writeUTF(self, s):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	694 """Writes a UTF-8 string of a given length to the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	695 utfstr = s.encode('utf-8')
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	696 length = len(utfstr)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	697 if length > 64:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	698 raise NamePartTooLongException
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	699 self.writeByte(length)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	700 self.writeString(utfstr, length)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	701
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	702 def writeName(self, name):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	703 """Writes a domain name to the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	704
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	705 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	706 # Find existing instance of this name in packet
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	707 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	708 index = self.names[name]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	709 except KeyError:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	710 # No record of this name already, so write it
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	711 # out as normal, recording the location of the name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	712 # for future pointers to it.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	713 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	714 self.names[name] = self.size
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	715 parts = name.split('.')
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	716 if parts[-1] == '':
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	717 parts = parts[:-1]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	718 for part in parts:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	719 self.writeUTF(part)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	720 self.writeByte(0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	721 return
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	722
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	723 # An index was found, so write a pointer to it
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	724 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	725 self.writeByte((index >> 8) \| 0xC0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	726 self.writeByte(index)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	727
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	728 def writeQuestion(self, question):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	729 """Writes a question to the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	730 self.writeName(question.name)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	731 self.writeShort(question.type)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	732 self.writeShort(question.clazz)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	733
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	734 def writeRecord(self, record, now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	735 """Writes a record (answer, authoritative answer, additional) to
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	736 the packet"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	737 self.writeName(record.name)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	738 self.writeShort(record.type)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	739 if record.unique and self.multicast:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	740 self.writeShort(record.clazz \| _CLASS_UNIQUE)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	741 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	742 self.writeShort(record.clazz)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	743 if now == 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	744 self.writeInt(record.ttl)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	745 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	746 self.writeInt(record.getRemainingTTL(now))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	747 index = len(self.data)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	748 # Adjust size for the short we will write before this record
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	749 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	750 self.size += 2
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	751 record.write(self)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	752 self.size -= 2
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	753
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	754 length = len(''.join(self.data[index:]))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	755 self.insertShort(index, length) # Here is the short we adjusted for
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	756
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	757 def packet(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	758 """Returns a string containing the packet's bytes
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	759
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	760 No further parts should be added to the packet once this
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	761 is done."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	762 if not self.finished:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	763 self.finished = 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	764 for question in self.questions:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	765 self.writeQuestion(question)
28296 a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	766 for answer, time_ in self.answers:
a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	767 self.writeRecord(answer, time_)
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	768 for authority in self.authorities:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	769 self.writeRecord(authority, 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	770 for additional in self.additionals:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	771 self.writeRecord(additional, 0)
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	772
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	773 self.insertShort(0, len(self.additionals))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	774 self.insertShort(0, len(self.authorities))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	775 self.insertShort(0, len(self.answers))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	776 self.insertShort(0, len(self.questions))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	777 self.insertShort(0, self.flags)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	778 if self.multicast:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	779 self.insertShort(0, 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	780 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	781 self.insertShort(0, self.id)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	782 return ''.join(self.data)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	783
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	784
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	785 class DNSCache(object):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	786 """A cache of DNS entries"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	787
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	788 def __init__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	789 self.cache = {}
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	790
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	791 def add(self, entry):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	792 """Adds an entry"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	793 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	794 list = self.cache[entry.key]
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	795 except KeyError:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	796 list = self.cache[entry.key] = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	797 list.append(entry)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	798
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	799 def remove(self, entry):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	800 """Removes an entry"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	801 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	802 list = self.cache[entry.key]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	803 list.remove(entry)
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	804 except KeyError:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	805 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	806
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	807 def get(self, entry):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	808 """Gets an entry by key. Will return None if there is no
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	809 matching entry."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	810 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	811 list = self.cache[entry.key]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	812 return list[list.index(entry)]
14063 87ebf72878ed Zeroconf: catch both ValueError and KeyError in get() Augie Fackler <durin42@gmail.com> parents: 13930 diff changeset	813 except (KeyError, ValueError):
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	814 return None
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	815
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	816 def getByDetails(self, name, type, clazz):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	817 """Gets an entry by details. Will return None if there is
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	818 no matching entry."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	819 entry = DNSEntry(name, type, clazz)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	820 return self.get(entry)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	821
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	822 def entriesWithName(self, name):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	823 """Returns a list of entries whose key matches the name."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	824 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	825 return self.cache[name]
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	826 except KeyError:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	827 return []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	828
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	829 def entries(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	830 """Returns a list of all entries"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	831 def add(x, y): return x+y
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	832 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	833 return reduce(add, self.cache.values())
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	834 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	835 return []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	836
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	837
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	838 class Engine(threading.Thread):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	839 """An engine wraps read access to sockets, allowing objects that
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	840 need to receive data from sockets to be called back when the
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	841 sockets are ready.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	842
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	843 A reader needs a handle_read() method, which is called when the socket
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	844 it is interested in is ready for reading.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	845
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	846 Writers are not implemented here, because we only send short
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	847 packets.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	848 """
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	849
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	850 def __init__(self, zeroconf):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	851 threading.Thread.__init__(self)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	852 self.zeroconf = zeroconf
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	853 self.readers = {} # maps socket to reader
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	854 self.timeout = 5
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	855 self.condition = threading.Condition()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	856 self.start()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	857
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	858 def run(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	859 while not globals()['_GLOBAL_DONE']:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	860 rs = self.getReaders()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	861 if len(rs) == 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	862 # No sockets to manage, but we wait for the timeout
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	863 # or addition of a socket
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	864 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	865 self.condition.acquire()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	866 self.condition.wait(self.timeout)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	867 self.condition.release()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	868 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	869 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	870 rr, wr, er = select.select(rs, [], [], self.timeout)
28296 a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	871 for sock in rr:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	872 try:
28296 a73394e7b47c zeroconf: use absolute_import timeless <timeless@mozdev.org> parents: 28295 diff changeset	873 self.readers[sock].handle_read()
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	874 except Exception:
10514 1426b4807fc8 zeroconf: suppress traceback during shutdown Brendan Cully <brendan@kublai.com> parents: 10387 diff changeset	875 if not globals()['_GLOBAL_DONE']:
1426b4807fc8 zeroconf: suppress traceback during shutdown Brendan Cully <brendan@kublai.com> parents: 10387 diff changeset	876 traceback.print_exc()
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	877 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	878 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	879
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	880 def getReaders(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	881 self.condition.acquire()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	882 result = self.readers.keys()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	883 self.condition.release()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	884 return result
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	885
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	886 def addReader(self, reader, socket):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	887 self.condition.acquire()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	888 self.readers[socket] = reader
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	889 self.condition.notify()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	890 self.condition.release()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	891
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	892 def delReader(self, socket):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	893 self.condition.acquire()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	894 del(self.readers[socket])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	895 self.condition.notify()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	896 self.condition.release()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	897
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	898 def notify(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	899 self.condition.acquire()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	900 self.condition.notify()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	901 self.condition.release()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	902
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	903 class Listener(object):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	904 """A Listener is used by this module to listen on the multicast
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	905 group to which DNS messages are sent, allowing the implementation
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	906 to cache information as it arrives.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	907
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	908 It requires registration with an Engine object in order to have
17424 e7cfe3587ea4 fix trivial spelling errors Mads Kiilerich <mads@kiilerich.com> parents: 15782 diff changeset	909 the read() method called when a socket is available for reading."""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	910
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	911 def __init__(self, zeroconf):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	912 self.zeroconf = zeroconf
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	913 self.zeroconf.engine.addReader(self, self.zeroconf.socket)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	914
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	915 def handle_read(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	916 data, (addr, port) = self.zeroconf.socket.recvfrom(_MAX_MSG_ABSOLUTE)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	917 self.data = data
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	918 msg = DNSIncoming(data)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	919 if msg.isQuery():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	920 # Always multicast responses
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	921 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	922 if port == _MDNS_PORT:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	923 self.zeroconf.handleQuery(msg, _MDNS_ADDR, _MDNS_PORT)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	924 # If it's not a multicast query, reply via unicast
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	925 # and multicast
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	926 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	927 elif port == _DNS_PORT:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	928 self.zeroconf.handleQuery(msg, addr, port)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	929 self.zeroconf.handleQuery(msg, _MDNS_ADDR, _MDNS_PORT)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	930 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	931 self.zeroconf.handleResponse(msg)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	932
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	933
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	934 class Reaper(threading.Thread):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	935 """A Reaper is used by this module to remove cache entries that
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	936 have expired."""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	937
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	938 def __init__(self, zeroconf):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	939 threading.Thread.__init__(self)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	940 self.zeroconf = zeroconf
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	941 self.start()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	942
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	943 def run(self):
14494 1ffeeb91c55d check-code: flag 0/1 used as constant Boolean expression Martin Geisler <mg@lazybytes.net> parents: 14063 diff changeset	944 while True:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	945 self.zeroconf.wait(10 * 1000)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	946 if globals()['_GLOBAL_DONE']:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	947 return
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	948 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	949 for record in self.zeroconf.cache.entries():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	950 if record.isExpired(now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	951 self.zeroconf.updateRecord(now, record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	952 self.zeroconf.cache.remove(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	953
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	954
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	955 class ServiceBrowser(threading.Thread):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	956 """Used to browse for a service of a specific type.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	957
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	958 The listener object will have its addService() and
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	959 removeService() methods called when this browser
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	960 discovers changes in the services availability."""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	961
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	962 def __init__(self, zeroconf, type, listener):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	963 """Creates a browser for a specific type"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	964 threading.Thread.__init__(self)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	965 self.zeroconf = zeroconf
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	966 self.type = type
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	967 self.listener = listener
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	968 self.services = {}
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	969 self.nextTime = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	970 self.delay = _BROWSER_TIME
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	971 self.list = []
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	972
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	973 self.done = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	974
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	975 self.zeroconf.addListener(self, DNSQuestion(self.type, _TYPE_PTR, _CLASS_IN))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	976 self.start()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	977
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	978 def updateRecord(self, zeroconf, now, record):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	979 """Callback invoked by Zeroconf when new information arrives.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	980
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	981 Updates information required by browser in the Zeroconf cache."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	982 if record.type == _TYPE_PTR and record.name == self.type:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	983 expired = record.isExpired(now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	984 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	985 oldrecord = self.services[record.alias.lower()]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	986 if not expired:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	987 oldrecord.resetTTL(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	988 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	989 del(self.services[record.alias.lower()])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	990 callback = lambda x: self.listener.removeService(x, self.type, record.alias)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	991 self.list.append(callback)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	992 return
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	993 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	994 if not expired:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	995 self.services[record.alias.lower()] = record
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	996 callback = lambda x: self.listener.addService(x, self.type, record.alias)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	997 self.list.append(callback)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	998
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	999 expires = record.getExpirationTime(75)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1000 if expires < self.nextTime:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1001 self.nextTime = expires
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1002
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1003 def cancel(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1004 self.done = 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1005 self.zeroconf.notifyAll()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1006
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1007 def run(self):
14494 1ffeeb91c55d check-code: flag 0/1 used as constant Boolean expression Martin Geisler <mg@lazybytes.net> parents: 14063 diff changeset	1008 while True:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1009 event = None
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1010 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1011 if len(self.list) == 0 and self.nextTime > now:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1012 self.zeroconf.wait(self.nextTime - now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1013 if globals()['_GLOBAL_DONE'] or self.done:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1014 return
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1015 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1016
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1017 if self.nextTime <= now:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1018 out = DNSOutgoing(_FLAGS_QR_QUERY)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1019 out.addQuestion(DNSQuestion(self.type, _TYPE_PTR, _CLASS_IN))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1020 for record in self.services.values():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1021 if not record.isExpired(now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1022 out.addAnswerAtTime(record, now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1023 self.zeroconf.send(out)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1024 self.nextTime = now + self.delay
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1025 self.delay = min(20 * 1000, self.delay * 2)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1026
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1027 if len(self.list) > 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1028 event = self.list.pop(0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1029
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1030 if event is not None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1031 event(self.zeroconf)
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1032
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1033
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1034 class ServiceInfo(object):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1035 """Service information"""
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1036
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1037 def __init__(self, type, name, address=None, port=None, weight=0, priority=0, properties=None, server=None):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1038 """Create a service description.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1039
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1040 type: fully qualified service type name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1041 name: fully qualified service name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1042 address: IP address as unsigned short, network byte order
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1043 port: port that the service runs on
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1044 weight: weight of the service
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1045 priority: priority of the service
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1046 properties: dictionary of properties (or a string holding the bytes for the text field)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1047 server: fully qualified name for service host (defaults to name)"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1048
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1049 if not name.endswith(type):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1050 raise BadTypeInNameException
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1051 self.type = type
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1052 self.name = name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1053 self.address = address
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1054 self.port = port
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1055 self.weight = weight
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1056 self.priority = priority
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1057 if server:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1058 self.server = server
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1059 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1060 self.server = name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1061 self.setProperties(properties)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1062
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1063 def setProperties(self, properties):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1064 """Sets properties and text of this info from a dictionary"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1065 if isinstance(properties, dict):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1066 self.properties = properties
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1067 list = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1068 result = ''
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1069 for key in properties:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1070 value = properties[key]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1071 if value is None:
10343 b8e3aeb7542c zeroconf: do not try to encode encoded strings (issue1942) Henrik Stuart <hg@hstuart.dk> parents: 10288 diff changeset	1072 suffix = ''
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1073 elif isinstance(value, str):
10343 b8e3aeb7542c zeroconf: do not try to encode encoded strings (issue1942) Henrik Stuart <hg@hstuart.dk> parents: 10288 diff changeset	1074 suffix = value
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1075 elif isinstance(value, int):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1076 if value:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1077 suffix = 'true'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1078 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1079 suffix = 'false'
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1080 else:
10343 b8e3aeb7542c zeroconf: do not try to encode encoded strings (issue1942) Henrik Stuart <hg@hstuart.dk> parents: 10288 diff changeset	1081 suffix = ''
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1082 list.append('='.join((key, suffix)))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1083 for item in list:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1084 result = ''.join((result, struct.pack('!c', chr(len(item))), item))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1085 self.text = result
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1086 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1087 self.text = properties
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1088
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1089 def setText(self, text):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1090 """Sets properties and text given a text field"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1091 self.text = text
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1092 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1093 result = {}
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1094 end = len(text)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1095 index = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1096 strs = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1097 while index < end:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1098 length = ord(text[index])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1099 index += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1100 strs.append(text[index:index+length])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1101 index += length
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1102
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1103 for s in strs:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1104 eindex = s.find('=')
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1105 if eindex == -1:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1106 # No equals sign at all
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1107 key = s
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1108 value = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1109 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1110 key = s[:eindex]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1111 value = s[eindex+1:]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1112 if value == 'true':
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1113 value = 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1114 elif value == 'false' or not value:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1115 value = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1116
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1117 # Only update non-existent properties
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1118 if key and result.get(key) == None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1119 result[key] = value
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1120
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1121 self.properties = result
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	1122 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1123 traceback.print_exc()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1124 self.properties = None
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1125
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1126 def getType(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1127 """Type accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1128 return self.type
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1129
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1130 def getName(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1131 """Name accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1132 if self.type is not None and self.name.endswith("." + self.type):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1133 return self.name[:len(self.name) - len(self.type) - 1]
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1134 return self.name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1135
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1136 def getAddress(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1137 """Address accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1138 return self.address
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1139
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1140 def getPort(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1141 """Port accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1142 return self.port
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1143
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1144 def getPriority(self):
17424 e7cfe3587ea4 fix trivial spelling errors Mads Kiilerich <mads@kiilerich.com> parents: 15782 diff changeset	1145 """Priority accessor"""
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1146 return self.priority
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1147
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1148 def getWeight(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1149 """Weight accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1150 return self.weight
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1151
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1152 def getProperties(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1153 """Properties accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1154 return self.properties
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1155
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1156 def getText(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1157 """Text accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1158 return self.text
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1159
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1160 def getServer(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1161 """Server accessor"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1162 return self.server
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1163
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1164 def updateRecord(self, zeroconf, now, record):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1165 """Updates service information from a DNS record"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1166 if record is not None and not record.isExpired(now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1167 if record.type == _TYPE_A:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1168 #if record.name == self.name:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1169 if record.name == self.server:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1170 self.address = record.address
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1171 elif record.type == _TYPE_SRV:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1172 if record.name == self.name:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1173 self.server = record.server
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1174 self.port = record.port
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1175 self.weight = record.weight
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1176 self.priority = record.priority
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1177 #self.address = None
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1178 self.updateRecord(zeroconf, now, zeroconf.cache.getByDetails(self.server, _TYPE_A, _CLASS_IN))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1179 elif record.type == _TYPE_TXT:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1180 if record.name == self.name:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1181 self.setText(record.text)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1182
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1183 def request(self, zeroconf, timeout):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1184 """Returns true if the service could be discovered on the
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1185 network, and updates this object with details discovered.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1186 """
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1187 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1188 delay = _LISTENER_TIME
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1189 next = now + delay
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1190 last = now + timeout
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1191 result = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1192 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1193 zeroconf.addListener(self, DNSQuestion(self.name, _TYPE_ANY, _CLASS_IN))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1194 while self.server is None or self.address is None or self.text is None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1195 if last <= now:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1196 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1197 if next <= now:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1198 out = DNSOutgoing(_FLAGS_QR_QUERY)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1199 out.addQuestion(DNSQuestion(self.name, _TYPE_SRV, _CLASS_IN))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1200 out.addAnswerAtTime(zeroconf.cache.getByDetails(self.name, _TYPE_SRV, _CLASS_IN), now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1201 out.addQuestion(DNSQuestion(self.name, _TYPE_TXT, _CLASS_IN))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1202 out.addAnswerAtTime(zeroconf.cache.getByDetails(self.name, _TYPE_TXT, _CLASS_IN), now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1203 if self.server is not None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1204 out.addQuestion(DNSQuestion(self.server, _TYPE_A, _CLASS_IN))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1205 out.addAnswerAtTime(zeroconf.cache.getByDetails(self.server, _TYPE_A, _CLASS_IN), now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1206 zeroconf.send(out)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1207 next = now + delay
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1208 delay = delay * 2
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1209
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1210 zeroconf.wait(min(next, last) - now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1211 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1212 result = 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1213 finally:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1214 zeroconf.removeListener(self)
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1215
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1216 return result
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1217
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1218 def __eq__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1219 """Tests equality of service name"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1220 if isinstance(other, ServiceInfo):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1221 return other.name == self.name
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1222 return 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1223
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1224 def __ne__(self, other):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1225 """Non-equality test"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1226 return not self.__eq__(other)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1227
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1228 def __repr__(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1229 """String representation"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1230 result = "service[%s,%s:%s," % (self.name, socket.inet_ntoa(self.getAddress()), self.port)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1231 if self.text is None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1232 result += "None"
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1233 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1234 if len(self.text) < 20:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1235 result += self.text
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1236 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1237 result += self.text[:17] + "..."
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1238 result += "]"
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1239 return result
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1240
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1241
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1242 class Zeroconf(object):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1243 """Implementation of Zeroconf Multicast DNS Service Discovery
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1244
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1245 Supports registration, unregistration, queries and browsing.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1246 """
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1247 def __init__(self, bindaddress=None):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1248 """Creates an instance of the Zeroconf class, establishing
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1249 multicast communications, listening and reaping threads."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1250 globals()['_GLOBAL_DONE'] = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1251 if bindaddress is None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1252 self.intf = socket.gethostbyname(socket.gethostname())
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1253 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1254 self.intf = bindaddress
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1255 self.group = ('', _MDNS_PORT)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1256 self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1257 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1258 self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1259 self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	1260 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1261 # SO_REUSEADDR should be equivalent to SO_REUSEPORT for
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1262 # multicast UDP sockets (p 731, "TCP/IP Illustrated,
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1263 # Volume 2"), but some BSD-derived systems require
17424 e7cfe3587ea4 fix trivial spelling errors Mads Kiilerich <mads@kiilerich.com> parents: 15782 diff changeset	1264 # SO_REUSEPORT to be specified explicitly. Also, not all
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1265 # versions of Python have SO_REUSEPORT available. So
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1266 # if you're on a BSD-based system, and haven't upgraded
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1267 # to Python 2.3 yet, you may find this library doesn't
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1268 # work as expected.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1269 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1270 pass
28249 c16949fcb566 zeroconf: fix setsockopt() call on Solaris to send payload of correct length Danek Duvall <danek.duvall@oracle.com> parents: 27637 diff changeset	1271 self.socket.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_TTL, "\xff")
c16949fcb566 zeroconf: fix setsockopt() call on Solaris to send payload of correct length Danek Duvall <danek.duvall@oracle.com> parents: 27637 diff changeset	1272 self.socket.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_LOOP, "\x01")
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1273 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1274 self.socket.bind(self.group)
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	1275 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1276 # Some versions of linux raise an exception even though
17428 72803c8edaa4 avoid using abbreviations that look like spelling errors Mads Kiilerich <mads@kiilerich.com> parents: 17426 diff changeset	1277 # SO_REUSEADDR and SO_REUSEPORT have been set, so ignore it
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1278 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1279 self.socket.setsockopt(socket.SOL_IP, socket.IP_ADD_MEMBERSHIP, socket.inet_aton(_MDNS_ADDR) + socket.inet_aton('0.0.0.0'))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1280
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1281 self.listeners = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1282 self.browsers = []
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1283 self.services = {}
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1284 self.servicetypes = {}
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1285
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1286 self.cache = DNSCache()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1287
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1288 self.condition = threading.Condition()
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1289
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1290 self.engine = Engine(self)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1291 self.listener = Listener(self)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1292 self.reaper = Reaper(self)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1293
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1294 def isLoopback(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1295 return self.intf.startswith("127.0.0.1")
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1296
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1297 def isLinklocal(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1298 return self.intf.startswith("169.254.")
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1299
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1300 def wait(self, timeout):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1301 """Calling thread waits for a given number of milliseconds or
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1302 until notified."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1303 self.condition.acquire()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1304 self.condition.wait(timeout/1000)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1305 self.condition.release()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1306
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1307 def notifyAll(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1308 """Notifies all waiting threads"""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1309 self.condition.acquire()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1310 self.condition.notifyAll()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1311 self.condition.release()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1312
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1313 def getServiceInfo(self, type, name, timeout=3000):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1314 """Returns network's service information for a particular
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1315 name and type, or None if no service matches by the timeout,
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1316 which defaults to 3 seconds."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1317 info = ServiceInfo(type, name)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1318 if info.request(self, timeout):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1319 return info
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1320 return None
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1321
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1322 def addServiceListener(self, type, listener):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1323 """Adds a listener for a particular service type. This object
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1324 will then have its updateRecord method called when information
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1325 arrives for that type."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1326 self.removeServiceListener(listener)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1327 self.browsers.append(ServiceBrowser(self, type, listener))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1328
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1329 def removeServiceListener(self, listener):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1330 """Removes a listener from the set that is currently listening."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1331 for browser in self.browsers:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1332 if browser.listener == listener:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1333 browser.cancel()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1334 del(browser)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1335
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1336 def registerService(self, info, ttl=_DNS_TTL):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1337 """Registers service information to the network with a default TTL
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1338 of 60 seconds. Zeroconf will then respond to requests for
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1339 information for that service. The name of the service may be
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1340 changed if needed to make it unique on the network."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1341 self.checkService(info)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1342 self.services[info.name.lower()] = info
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1343 if self.servicetypes.has_key(info.type):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1344 self.servicetypes[info.type]+=1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1345 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1346 self.servicetypes[info.type]=1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1347 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1348 nextTime = now
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1349 i = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1350 while i < 3:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1351 if now < nextTime:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1352 self.wait(nextTime - now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1353 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1354 continue
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1355 out = DNSOutgoing(_FLAGS_QR_RESPONSE \| _FLAGS_AA)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1356 out.addAnswerAtTime(DNSPointer(info.type, _TYPE_PTR, _CLASS_IN, ttl, info.name), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1357 out.addAnswerAtTime(DNSService(info.name, _TYPE_SRV, _CLASS_IN, ttl, info.priority, info.weight, info.port, info.server), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1358 out.addAnswerAtTime(DNSText(info.name, _TYPE_TXT, _CLASS_IN, ttl, info.text), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1359 if info.address:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1360 out.addAnswerAtTime(DNSAddress(info.server, _TYPE_A, _CLASS_IN, ttl, info.address), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1361 self.send(out)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1362 i += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1363 nextTime += _REGISTER_TIME
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1364
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1365 def unregisterService(self, info):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1366 """Unregister a service."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1367 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1368 del(self.services[info.name.lower()])
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1369 if self.servicetypes[info.type]>1:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1370 self.servicetypes[info.type]-=1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1371 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1372 del self.servicetypes[info.type]
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	1373 except KeyError:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1374 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1375 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1376 nextTime = now
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1377 i = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1378 while i < 3:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1379 if now < nextTime:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1380 self.wait(nextTime - now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1381 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1382 continue
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1383 out = DNSOutgoing(_FLAGS_QR_RESPONSE \| _FLAGS_AA)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1384 out.addAnswerAtTime(DNSPointer(info.type, _TYPE_PTR, _CLASS_IN, 0, info.name), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1385 out.addAnswerAtTime(DNSService(info.name, _TYPE_SRV, _CLASS_IN, 0, info.priority, info.weight, info.port, info.name), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1386 out.addAnswerAtTime(DNSText(info.name, _TYPE_TXT, _CLASS_IN, 0, info.text), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1387 if info.address:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1388 out.addAnswerAtTime(DNSAddress(info.server, _TYPE_A, _CLASS_IN, 0, info.address), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1389 self.send(out)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1390 i += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1391 nextTime += _UNREGISTER_TIME
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1392
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1393 def unregisterAllServices(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1394 """Unregister all registered services."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1395 if len(self.services) > 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1396 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1397 nextTime = now
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1398 i = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1399 while i < 3:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1400 if now < nextTime:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1401 self.wait(nextTime - now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1402 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1403 continue
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1404 out = DNSOutgoing(_FLAGS_QR_RESPONSE \| _FLAGS_AA)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1405 for info in self.services.values():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1406 out.addAnswerAtTime(DNSPointer(info.type, _TYPE_PTR, _CLASS_IN, 0, info.name), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1407 out.addAnswerAtTime(DNSService(info.name, _TYPE_SRV, _CLASS_IN, 0, info.priority, info.weight, info.port, info.server), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1408 out.addAnswerAtTime(DNSText(info.name, _TYPE_TXT, _CLASS_IN, 0, info.text), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1409 if info.address:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1410 out.addAnswerAtTime(DNSAddress(info.server, _TYPE_A, _CLASS_IN, 0, info.address), 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1411 self.send(out)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1412 i += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1413 nextTime += _UNREGISTER_TIME
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1414
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1415 def checkService(self, info):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1416 """Checks the network for a unique service name, modifying the
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1417 ServiceInfo passed in if it is not unique."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1418 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1419 nextTime = now
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1420 i = 0
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1421 while i < 3:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1422 for record in self.cache.entriesWithName(info.type):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1423 if record.type == _TYPE_PTR and not record.isExpired(now) and record.alias == info.name:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1424 if (info.name.find('.') < 0):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1425 info.name = info.name + ".[" + info.address + ":" + info.port + "]." + info.type
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1426 self.checkService(info)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1427 return
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1428 raise NonUniqueNameException
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1429 if now < nextTime:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1430 self.wait(nextTime - now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1431 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1432 continue
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1433 out = DNSOutgoing(_FLAGS_QR_QUERY \| _FLAGS_AA)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1434 self.debug = out
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1435 out.addQuestion(DNSQuestion(info.type, _TYPE_PTR, _CLASS_IN))
17476 210acc861c40 spelling: Authoritative timeless@mozdev.org parents: 15782 diff changeset	1436 out.addAuthoritativeAnswer(DNSPointer(info.type, _TYPE_PTR, _CLASS_IN, _DNS_TTL, info.name))
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1437 self.send(out)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1438 i += 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1439 nextTime += _CHECK_TIME
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1440
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1441 def addListener(self, listener, question):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1442 """Adds a listener for a given question. The listener will have
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1443 its updateRecord method called when information is available to
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1444 answer the question."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1445 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1446 self.listeners.append(listener)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1447 if question is not None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1448 for record in self.cache.entriesWithName(question.name):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1449 if question.answeredBy(record) and not record.isExpired(now):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1450 listener.updateRecord(self, now, record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1451 self.notifyAll()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1452
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1453 def removeListener(self, listener):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1454 """Removes a listener."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1455 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1456 self.listeners.remove(listener)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1457 self.notifyAll()
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	1458 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1459 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1460
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1461 def updateRecord(self, now, rec):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1462 """Used to notify listeners of new information that has updated
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1463 a record."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1464 for listener in self.listeners:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1465 listener.updateRecord(self, now, rec)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1466 self.notifyAll()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1467
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1468 def handleResponse(self, msg):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1469 """Deal with incoming response packets. All answers
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1470 are held in the cache, and listeners are notified."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1471 now = currentTimeMillis()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1472 for record in msg.answers:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1473 expired = record.isExpired(now)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1474 if record in self.cache.entries():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1475 if expired:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1476 self.cache.remove(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1477 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1478 entry = self.cache.get(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1479 if entry is not None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1480 entry.resetTTL(record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1481 record = entry
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1482 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1483 self.cache.add(record)
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1484
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1485 self.updateRecord(now, record)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1486
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1487 def handleQuery(self, msg, addr, port):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1488 """Deal with incoming query packets. Provides a response if
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1489 possible."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1490 out = None
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1491
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1492 # Support unicast client responses
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1493 #
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1494 if port != _MDNS_PORT:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1495 out = DNSOutgoing(_FLAGS_QR_RESPONSE \| _FLAGS_AA, 0)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1496 for question in msg.questions:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1497 out.addQuestion(question)
7874 d812029cda85 cleanup: drop variables for unused return values Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7071 diff changeset	1498
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1499 for question in msg.questions:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1500 if question.type == _TYPE_PTR:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1501 if question.name == "_services._dns-sd._udp.local.":
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1502 for stype in self.servicetypes.keys():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1503 if out is None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1504 out = DNSOutgoing(_FLAGS_QR_RESPONSE \| _FLAGS_AA)
7874 d812029cda85 cleanup: drop variables for unused return values Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7071 diff changeset	1505 out.addAnswer(msg, DNSPointer("_services._dns-sd._udp.local.", _TYPE_PTR, _CLASS_IN, _DNS_TTL, stype))
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1506 for service in self.services.values():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1507 if question.name == service.type:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1508 if out is None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1509 out = DNSOutgoing(_FLAGS_QR_RESPONSE \| _FLAGS_AA)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1510 out.addAnswer(msg, DNSPointer(service.type, _TYPE_PTR, _CLASS_IN, _DNS_TTL, service.name))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1511 else:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1512 try:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1513 if out is None:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1514 out = DNSOutgoing(_FLAGS_QR_RESPONSE \| _FLAGS_AA)
7874 d812029cda85 cleanup: drop variables for unused return values Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7071 diff changeset	1515
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1516 # Answer A record queries for any service addresses we know
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1517 if question.type == _TYPE_A or question.type == _TYPE_ANY:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1518 for service in self.services.values():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1519 if service.server == question.name.lower():
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1520 out.addAnswer(msg, DNSAddress(question.name, _TYPE_A, _CLASS_IN \| _CLASS_UNIQUE, _DNS_TTL, service.address))
7874 d812029cda85 cleanup: drop variables for unused return values Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7071 diff changeset	1521
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1522 service = self.services.get(question.name.lower(), None)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1523 if not service: continue
7874 d812029cda85 cleanup: drop variables for unused return values Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7071 diff changeset	1524
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1525 if question.type == _TYPE_SRV or question.type == _TYPE_ANY:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1526 out.addAnswer(msg, DNSService(question.name, _TYPE_SRV, _CLASS_IN \| _CLASS_UNIQUE, _DNS_TTL, service.priority, service.weight, service.port, service.server))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1527 if question.type == _TYPE_TXT or question.type == _TYPE_ANY:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1528 out.addAnswer(msg, DNSText(question.name, _TYPE_TXT, _CLASS_IN \| _CLASS_UNIQUE, _DNS_TTL, service.text))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1529 if question.type == _TYPE_SRV:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1530 out.addAdditionalAnswer(DNSAddress(service.server, _TYPE_A, _CLASS_IN \| _CLASS_UNIQUE, _DNS_TTL, service.address))
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	1531 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1532 traceback.print_exc()
7874 d812029cda85 cleanup: drop variables for unused return values Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7071 diff changeset	1533
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1534 if out is not None and out.answers:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1535 out.id = msg.id
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1536 self.send(out, addr, port)
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1537
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1538 def send(self, out, addr = _MDNS_ADDR, port = _MDNS_PORT):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1539 """Sends an outgoing packet."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1540 # This is a quick test to see if we can parse the packets we generate
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1541 #temp = DNSIncoming(out.packet())
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1542 try:
7874 d812029cda85 cleanup: drop variables for unused return values Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7071 diff changeset	1543 self.socket.sendto(out.packet(), 0, (addr, port))
13930 0b7853aee5ae zeroconf: clean up naked exceptions Matt Mackall <mpm@selenic.com> parents: 11435 diff changeset	1544 except Exception:
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1545 # Ignore this, it may be a temporary loss of network connection
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1546 pass
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1547
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1548 def close(self):
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1549 """Ends the background threads, and prevent this instance from
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1550 servicing further queries."""
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1551 if globals()['_GLOBAL_DONE'] == 0:
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1552 globals()['_GLOBAL_DONE'] = 1
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1553 self.notifyAll()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1554 self.engine.notify()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1555 self.unregisterAllServices()
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1556 self.socket.setsockopt(socket.SOL_IP, socket.IP_DROP_MEMBERSHIP, socket.inet_aton(_MDNS_ADDR) + socket.inet_aton('0.0.0.0'))
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1557 self.socket.close()
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1558
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1559 # Test a few module features, including service registration, service
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1560 # query (for Zoe), and service unregistration.
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1561
7877 eba7f12b0c51 cleanup: whitespace cleanup Peter Arrenbrecht <peter.arrenbrecht@gmail.com> parents: 7874 diff changeset	1562 if __name__ == '__main__':
28295 01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1563 print("Multicast DNS Service Discovery for Python, version", __version__)
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1564 r = Zeroconf()
28295 01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1565 print("1. Testing registration of a service...")
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1566 desc = {'version':'0.10','a':'test value', 'b':'another value'}
643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1567 info = ServiceInfo("_http._tcp.local.", "My Service Name._http._tcp.local.", socket.inet_aton("127.0.0.1"), 1234, 0, 0, desc)
28295 01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1568 print(" Registering service...")
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1569 r.registerService(info)
28295 01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1570 print(" Registration done.")
01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1571 print("2. Testing query of service information...")
01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1572 print(" Getting ZOE service:", str(r.getServiceInfo("_http._tcp.local.", "ZOE._http._tcp.local.")))
01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1573 print(" Query done.")
01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1574 print("3. Testing query of own service...")
01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1575 print(" Getting self:", str(r.getServiceInfo("_http._tcp.local.", "My Service Name._http._tcp.local.")))
01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1576 print(" Query done.")
01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1577 print("4. Testing unregister of service information...")
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1578 r.unregisterService(info)
28295 01815c159856 zeroconf: use print function timeless <timeless@mozdev.org> parents: 28249 diff changeset	1579 print(" Unregister done.")
7071 643c751e60b2 zeroconf: initial implementation Matt Mackall <mpm@selenic.com> parents: diff changeset	1580 r.close()
10288 e0121ad90b57 zeroconf: flag third-party Zeroconf.py for ignore in check-code Matt Mackall <mpm@selenic.com> parents: 8131 diff changeset	1581
e0121ad90b57 zeroconf: flag third-party Zeroconf.py for ignore in check-code Matt Mackall <mpm@selenic.com> parents: 8131 diff changeset	1582 # no-check-code

Mercurial > public > mercurial-scm > hg-stable

annotate hgext/zeroconf/Zeroconf.py @ 28296:a73394e7b47c