git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/python/gnuradio/gruimpl/socket_stuff.py

   1 #
   2 # Copyright 2005 Free Software Foundation, Inc.
   3 #
   4 # This file is part of GNU Radio
   5 #
   6 # GNU Radio is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2, or (at your option)
   9 # any later version.
  10 #
  11 # GNU Radio is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with GNU Radio; see the file COPYING.  If not, write to
  18 # the Free Software Foundation, Inc., 51 Franklin Street,
  19 # Boston, MA 02110-1301, USA.
  20 #
  21
  22 # random socket related stuff
  23
  24 import socket
  25 import os
  26 import sys
  27
  28 def tcp_connect_or_die(sock_addr):
  29     """
  30     @param sock_addr: (host, port) to connect to
  31     @type sock_addr: tuple
  32     @returns: socket or exits
  33     """
  34     s = socket.socket (socket.AF_INET, socket.SOCK_STREAM)
  35     try:
  36         s.connect(sock_addr)
  37     except socket.error, err:
  38         sys.stderr.write('Failed to connect to %s: %s\n' %
  39                          (sock_addr, os.strerror (err.args[0]),))
  40         sys.exit(1)
  41     return s
  42
  43 def udp_connect_or_die(sock_addr):
  44     """
  45     @param sock_addr: (host, port) to connect to
  46     @type sock_addr: tuple
  47     @returns: socket or exits
  48     """
  49     s = socket.socket (socket.AF_INET, socket.SOCK_DGRAM)
  50     try:
  51         s.connect(sock_addr)
  52     except socket.error, err:
  53         sys.stderr.write('Failed to connect to %s: %s\n' %
  54                          (sock_addr, os.strerror (err.args[0]),))
  55         sys.exit(1)
  56     return s