Source Code for Module deefuzzer.deefuzzer.tools.relay

 1  #!/usr/bin/python 
 2  # -*- coding: utf-8 -*- 
 3  # 
 4  # Copyright (C) 2006-2011 Guillaume Pellerin 
 5   
 6  # <yomguy@parisson.com> 
 7   
 8  # This software is a computer program whose purpose is to stream audio 
 9  # and video data through icecast2 servers. 
10   
11  # This software is governed by the CeCILL license under French law and 
12  # abiding by the rules of distribution of free software. You can use, 
13  # modify and/ or redistribute the software under the terms of the CeCILL 
14  # license as circulated by CEA, CNRS and INRIA at the following URL 
15  # "http://www.cecill.info". 
16   
17  # As a counterpart to the access to the source code and  rights to copy, 
18  # modify and redistribute granted by the license, users are provided only 
19  # with a limited warranty and the software's author, the holder of the 
20  # economic rights, and the successive licensors have only limited 
21  # liability. 
22   
23  # In this respect, the user's attention is drawn to the risks associated 
24  # with loading, using,  modifying and/or developing or reproducing the 
25  # software by the user in light of its specific status of free software, 
26  # that may mean that it is complicated to manipulate, and that also 
27  # therefore means that it is reserved for developers and  experienced 
28  # professionals having in-depth computer knowledge. Users are therefore 
29  # encouraged to load and test the software's suitability as regards their 
30  # requirements in conditions enabling the security of their systems and/or 
31  # data to be ensured and, more generally, to use and operate it in the 
32  # same conditions as regards security. 
33   
34  # The fact that you are presently reading this means that you have had 
35  # knowledge of the CeCILL license and that you accept its terms. 
36   
37  # Author: Guillaume Pellerin <yomguy@parisson.com> 
38   
39  from threading import Thread 
40  import Queue 
41  import urllib 
42   
43 -class Relay(Thread): 
44   
45 -    def __init__(self, sub_buffer_size, queue_size): 
46          Thread.__init__(self) 
47          self.sub_buffer_size = sub_buffer_size 
48          self.queue_size = queue_size 
49          self.queue = Queue.Queue(self.queue_size) 
50   
51 -    def set_url(self, url): 
52          self.url = url 
53   
54 -    def open(self): 
55          try: 
56              self.stream = urllib.urlopen(self.url) 
57              self.isopen = True 
58          except: 
59              self.isopen = False 
60   
61 -    def close(self): 
62          if self.stream: 
63              self.isopen = False 
64   
65 -    def run(self): 
66          while True: 
67              if self.isopen: 
68                  self.chunk = self.stream.read(self.sub_buffer_size) 
69                  self.queue.put_nowait(self.chunk) 
70  #                print self.queue.qsize() 
71              else: 
72                  self.stream.close() 
73                  break 
74