コード例 #1
0
ファイル: process.py プロジェクト: txrproject/stem
    def test_version_argument(tor_cmd):
        """
    Check that 'tor --version' matches 'GETINFO version'.
    """

        assert_equal('Tor version %s.\n' % test.tor_version(),
                     run_tor(tor_cmd, '--version'))
コード例 #2
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_unanonymous_hidden_service_config_must_match(tor_cmd):
    """
    Checking that startup fails if HiddenServiceNonAnonymousMode and
    HiddenServiceSingleHopMode don't match.
    """

    try:
      stem.process.launch_tor_with_config(
        tor_cmd = tor_cmd,
        config = {'HiddenServiceNonAnonymousMode': '1'},
      )

      raise AssertionError("Tor shouldn't start with 'HiddenServiceNonAnonymousMode' set but not 'HiddenServiceSingleHopMode'")
    except OSError as exc:
      assert_equal('Process terminated: HiddenServiceNonAnonymousMode does not provide any server anonymity. It must be used with HiddenServiceSingleHopMode set to 1.', str(exc))

    try:
      stem.process.launch_tor_with_config(
        tor_cmd = tor_cmd,
        config = {'HiddenServiceSingleHopMode': '1'},
      )

      raise AssertionError("Tor shouldn't start with 'HiddenServiceSingleHopMode' set but not 'HiddenServiceNonAnonymousMode'")
    except OSError as exc:
      assert_equal('Process terminated: HiddenServiceSingleHopMode does not provide any server anonymity. It must be used with HiddenServiceNonAnonymousMode set to 1.', str(exc))
コード例 #3
0
ファイル: process.py プロジェクト: txrproject/stem
    def test_hash_password(tor_cmd):
        """
    Hash a controller password. It's salted so can't assert that we get a
    particular value. Also, tor's output is unnecessarily verbose so including
    hush to cut it down.
    """

        output = run_tor(tor_cmd, '--hush', '--hash-password',
                         'my_password').splitlines()[-1]

        if not re.match('^16:[0-9A-F]{58}$', output):
            raise AssertionError(
                "Unexpected response from 'tor --hash-password my_password': %s"
                % output)

        # I'm not gonna even pretend to understand the following. Ported directly
        # from tor's test_cmdline_args.py.

        output_hex = binascii.a2b_hex(
            stem.util.str_tools._to_bytes(output).strip()[3:])
        salt, how, hashed = output_hex[:8], output_hex[8], output_hex[9:]

        count = (16 + (how & 15)) << ((how >> 4) + 6)
        stuff = salt + b'my_password'
        repetitions = count // len(stuff) + 1
        inp = (stuff * repetitions)[:count]
        assert_equal(hashlib.sha1(inp).digest(), hashed)
コード例 #4
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_torrc_arguments(tor_cmd):
    """
    Pass configuration options on the commandline.
    """

    with torrc() as torrc_path:
      config_args = [
        '+SocksPort', '9090',  # append an extra SocksPort
        '/ExtORPort',  # drops our ExtORPort
        '/TransPort',  # drops a port we didn't originally have
        '+ControlPort', '9005',  # appends a ControlPort where we didn't have any before
      ]

      output = run_tor(tor_cmd, '-f', torrc_path, '--dump-config', 'short', *config_args)
      result = [line for line in output.splitlines() if not line.startswith('DataDirectory')]

      expected = [
        'ControlPort 9005',
        'ExitPolicy reject *:*',
        'Nickname stemIntegTest',
        'PublishServerDescriptor 0',
        'SocksPort 9089',
        'SocksPort 9090',
      ]

      assert_equal(expected, result)
コード例 #5
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_hash_password(tor_cmd):
    """
    Hash a controller password. It's salted so can't assert that we get a
    particular value. Also, tor's output is unnecessarily verbose so including
    hush to cut it down.
    """

    output = run_tor(tor_cmd, '--hush', '--hash-password', 'my_password').splitlines()[-1]

    if not re.match('^16:[0-9A-F]{58}$', output):
      raise AssertionError("Unexpected response from 'tor --hash-password my_password': %s" % output)

    # I'm not gonna even pretend to understand the following. Ported directly
    # from tor's test_cmdline_args.py.

    if stem.prereq.is_python_3():
      output_hex = binascii.a2b_hex(stem.util.str_tools._to_bytes(output).strip()[3:])
      salt, how, hashed = output_hex[:8], output_hex[8], output_hex[9:]
    else:
      output_hex = binascii.a2b_hex(output.strip()[3:])
      salt, how, hashed = output_hex[:8], ord(output_hex[8]), output_hex[9:]

    count = (16 + (how & 15)) << ((how >> 4) + 6)
    stuff = salt + b'my_password'
    repetitions = count // len(stuff) + 1
    inp = (stuff * repetitions)[:count]
    assert_equal(hashlib.sha1(inp).digest(), hashed)
コード例 #6
0
ファイル: process.py プロジェクト: txrproject/stem
    def test_no_orphaned_process(tor_cmd):
        """
    Check that when an exception arises in the middle of spawning tor that we
    don't leave a lingering process.
    """

        if not stem.util.system.is_available('sleep'):
            raise unittest.case.SkipTest('(sleep unavailable)')

        with patch('re.compile', Mock(side_effect=KeyboardInterrupt('nope'))):
            # We don't need to actually run tor for this test. Rather, any process will
            # do the trick. Picking sleep so this'll clean itself up if our test fails.

            mock_tor_process = subprocess.Popen(['sleep', '60'])

            with patch('subprocess.Popen',
                       Mock(return_value=mock_tor_process)):
                try:
                    stem.process.launch_tor(tor_cmd)
                    raise AssertionError("tor shoudn't have started")
                except KeyboardInterrupt as exc:
                    if os.path.exists('/proc/%s' % mock_tor_process.pid):
                        raise AssertionError(
                            'launch_tor() left a lingering tor process')

                    assert_equal('nope', str(exc))
コード例 #7
0
ファイル: process.py プロジェクト: txrproject/stem
    def test_torrc_arguments(tor_cmd):
        """
    Pass configuration options on the commandline.
    """

        with torrc() as torrc_path:
            config_args = [
                '+SocksPort',
                '9090',  # append an extra SocksPort
                '/ExtORPort',  # drops our ExtORPort
                '/TransPort',  # drops a port we didn't originally have
                '+ControlPort',
                '9005',  # appends a ControlPort where we didn't have any before
            ]

            output = run_tor(tor_cmd, '-f', torrc_path, '--dump-config',
                             'short', *config_args)
            result = [
                line for line in output.splitlines()
                if not line.startswith('DataDirectory')
            ]

            expected = [
                'ControlPort 9005',
                'ExitPolicy reject *:*',
                'Nickname stemIntegTest',
                'PublishServerDescriptor 0',
                'SocksPort 9089',
                'SocksPort 9090',
            ]

            assert_equal(expected, result)
コード例 #8
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_quiet_argument(tor_cmd):
    """
    Check that we don't provide anything on stdout when running 'tor --quiet'.
    """

    quiet_output = run_tor(tor_cmd, '--quiet', '--invalid_argument', 'true', expect_failure = True)
    assert_equal('', quiet_output, 'No output should be provided with the --quiet argument')
コード例 #9
0
ファイル: process.py プロジェクト: zmchu/stem
  async def test_launch_tor_with_config_via_stdin(tor_cmd):
    """
    Exercises launch_tor_with_config when we provide our torrc via stdin.
    """

    with tempfile.TemporaryDirectory() as data_directory:
      control_port = random_port()
      control_socket, tor_process = None, None

      try:
        tor_process = stem.process.launch_tor_with_config(
          tor_cmd = tor_cmd,
          config = {
            'SocksPort': random_port(),
            'ControlPort': control_port,
            'DataDirectory': data_directory,
          },
          completion_percent = 0
        )

        control_socket = stem.socket.ControlPort(port = int(control_port))
        await stem.connection.authenticate(control_socket)

        # exercises the socket
        await control_socket.send('GETCONF ControlPort')
        getconf_response = await control_socket.recv()

        assert_equal('ControlPort=%s' % control_port, str(getconf_response))
      finally:
        if control_socket:
          await control_socket.close()

        if tor_process:
          tor_process.kill()
          tor_process.wait()
コード例 #10
0
ファイル: process.py プロジェクト: txrproject/stem
    def test_unanonymous_hidden_service_config_must_match(tor_cmd):
        """
    Checking that startup fails if HiddenServiceNonAnonymousMode and
    HiddenServiceSingleHopMode don't match.
    """

        try:
            stem.process.launch_tor_with_config(
                tor_cmd=tor_cmd,
                config={'HiddenServiceNonAnonymousMode': '1'},
            )

            raise AssertionError(
                "Tor shouldn't start with 'HiddenServiceNonAnonymousMode' set but not 'HiddenServiceSingleHopMode'"
            )
        except OSError as exc:
            assert_equal(
                'Process terminated: HiddenServiceNonAnonymousMode does not provide any server anonymity. It must be used with HiddenServiceSingleHopMode set to 1.',
                str(exc))

        try:
            stem.process.launch_tor_with_config(
                tor_cmd=tor_cmd,
                config={'HiddenServiceSingleHopMode': '1'},
            )

            raise AssertionError(
                "Tor shouldn't start with 'HiddenServiceSingleHopMode' set but not 'HiddenServiceNonAnonymousMode'"
            )
        except OSError as exc:
            assert_equal(
                'Process terminated: HiddenServiceSingleHopMode does not provide any server anonymity. It must be used with HiddenServiceNonAnonymousMode set to 1.',
                str(exc))
コード例 #11
0
ファイル: process.py プロジェクト: zmchu/stem
  def test_quiet_argument(tor_cmd):
    """
    Check that we don't provide anything on stdout when running 'tor --quiet'.
    """

    quiet_output = run_tor(tor_cmd, '--quiet', '--invalid_argument', 'true', expect_failure = True)
    assert_equal('', quiet_output, 'No output should be provided with the --quiet argument')
コード例 #12
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_with_invalid_config(tor_cmd):
    """
    Spawn a tor process with a configuration that should make it dead on arrival.
    """

    # Set the same SocksPort and ControlPort, this should fail with...
    #
    #   [warn] Failed to parse/validate config: Failed to bind one of the listener ports.
    #   [err] Reading config failed--see warnings above.

    with tmp_directory() as data_directory:
      both_ports = random_port()

      try:
        stem.process.launch_tor_with_config(
          tor_cmd = tor_cmd,
          config = {
            'SocksPort': both_ports,
            'ControlPort': both_ports,
            'DataDirectory': data_directory,
          },
        )

        raise AssertionError('Tor should fail to launch')
      except OSError as exc:
        assert_equal('Process terminated: Failed to bind one of the listener ports.', str(exc))
コード例 #13
0
ファイル: process.py プロジェクト: txrproject/stem
    def test_with_invalid_config(tor_cmd):
        """
    Spawn a tor process with a configuration that should make it dead on arrival.
    """

        # Set the same SocksPort and ControlPort, this should fail with...
        #
        #   [warn] Failed to parse/validate config: Failed to bind one of the listener ports.
        #   [err] Reading config failed--see warnings above.

        with tempfile.TemporaryDirectory() as data_directory:
            both_ports = random_port()

            try:
                stem.process.launch_tor_with_config(
                    tor_cmd=tor_cmd,
                    config={
                        'SocksPort': both_ports,
                        'ControlPort': both_ports,
                        'DataDirectory': data_directory,
                    },
                )

                raise AssertionError('Tor should fail to launch')
            except OSError as exc:
                assert_equal(
                    'Process terminated: Failed to bind one of the listener ports.',
                    str(exc))
コード例 #14
0
ファイル: process.py プロジェクト: txrproject/stem
    def test_can_run_multithreaded(tor_cmd):
        """
    Our launch_tor() function uses signal to support its timeout argument.
    This only works in the main thread so ensure we give a useful message when
    it isn't.
    """

        with tempfile.TemporaryDirectory() as data_directory:
            # Tries running tor in another thread with the given timeout argument. This
            # issues an invalid torrc so we terminate right away if we get to the point
            # of actually invoking tor.
            #
            # Returns None if launching tor is successful, and otherwise returns the
            # exception we raised.

            def launch_async_with_timeout(timeout_arg):
                raised_exc = [None]

                def short_launch():
                    try:
                        stem.process.launch_tor_with_config(
                            tor_cmd=tor_cmd,
                            config={
                                'SocksPort': 'invalid',
                                'DataDirectory': data_directory,
                            },
                            completion_percent=100,
                            timeout=timeout_arg,
                        )
                    except Exception as exc:
                        raised_exc[0] = exc

                t = threading.Thread(target=short_launch)
                t.start()
                t.join()

                if 'Invalid SocksPort' in str(raised_exc[0]):
                    return None  # got to the point of invoking tor
                else:
                    return raised_exc[0]

            exc = launch_async_with_timeout(0.5)
            assert_equal(OSError, type(exc))
            assert_equal(
                'Launching tor with a timeout can only be done in the main thread',
                str(exc))

            # We should launch successfully if no timeout is specified or we specify it
            # to be 'None'.

            if launch_async_with_timeout(None) is not None:
                raise AssertionError(
                    'Launching tor without a timeout should be successful')

            if launch_async_with_timeout(
                    stem.process.DEFAULT_INIT_TIMEOUT) is not None:
                raise AssertionError(
                    'Launching tor with the default timeout should be successful'
                )
コード例 #15
0
ファイル: process.py プロジェクト: zmchu/stem
  def test_torrc_arguments_via_stdin(tor_cmd):
    """
    Pass configuration options via stdin.
    """

    with tempfile.TemporaryDirectory() as data_directory:
      torrc = BASIC_RELAY_TORRC % data_directory
      output = run_tor(tor_cmd, '-f', '-', '--dump-config', 'short', stdin = torrc)
      assert_equal(sorted(torrc.splitlines()), sorted(output.splitlines()))
コード例 #16
0
ファイル: process.py プロジェクト: zmchu/stem
  def test_version_argument(tor_cmd):
    """
    Check that 'tor --version' matches 'GETINFO version'.
    """

    # We're only interested in the first line of output:
    # Other lines are about libraries and compilers.
    tor_version = run_tor(tor_cmd,  '--version').split("\n")[0]

    assert_equal('Tor version %s.' % test.tor_version(), tor_version)
コード例 #17
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_help_argument(tor_cmd):
    """
    Check that 'tor --help' provides the expected output.
    """

    help_output = run_tor(tor_cmd, '--help')

    if not help_output.startswith('Copyright (c) 2001') or not help_output.endswith('tor -f <torrc> [args]\nSee man page for options, or https://www.torproject.org/ for documentation.\n'):
      raise AssertionError("Help output didn't have the expected strings: %s" % help_output)

    assert_equal(help_output, run_tor(tor_cmd, '-h'), "'tor -h' should simply be an alias for 'tor --help'")
コード例 #18
0
ファイル: process.py プロジェクト: zmchu/stem
  def test_help_argument(tor_cmd):
    """
    Check that 'tor --help' provides the expected output.
    """

    help_output = run_tor(tor_cmd, '--help')

    if not help_output.startswith('Copyright (c) 2001') or not help_output.endswith('tor -f <torrc> [args]\nSee man page for options, or https://www.torproject.org/ for documentation.\n'):
      raise AssertionError("Help output didn't have the expected strings: %s" % help_output)

    assert_equal(help_output, run_tor(tor_cmd, '-h'), "'tor -h' should simply be an alias for 'tor --help'")
コード例 #19
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_can_run_multithreaded(tor_cmd):
    """
    Our launch_tor() function uses signal to support its timeout argument.
    This only works in the main thread so ensure we give a useful message when
    it isn't.
    """

    with tmp_directory() as data_directory:
      # Tries running tor in another thread with the given timeout argument. This
      # issues an invalid torrc so we terminate right away if we get to the point
      # of actually invoking tor.
      #
      # Returns None if launching tor is successful, and otherwise returns the
      # exception we raised.

      def launch_async_with_timeout(timeout_arg):
        raised_exc = [None]

        def short_launch():
          try:
            stem.process.launch_tor_with_config(
              tor_cmd = tor_cmd,
              config = {
                'SocksPort': 'invalid',
                'DataDirectory': data_directory,
              },
              completion_percent = 100,
              timeout = timeout_arg,
            )
          except Exception as exc:
            raised_exc[0] = exc

        t = threading.Thread(target = short_launch)
        t.start()
        t.join()

        if 'Invalid SocksPort' in str(raised_exc[0]):
          return None  # got to the point of invoking tor
        else:
          return raised_exc[0]

      exc = launch_async_with_timeout(0.5)
      assert_equal(OSError, type(exc))
      assert_equal('Launching tor with a timeout can only be done in the main thread', str(exc))

      # We should launch successfully if no timeout is specified or we specify it
      # to be 'None'.

      if launch_async_with_timeout(None) is not None:
        raise AssertionError('Launching tor without a timeout should be successful')

      if launch_async_with_timeout(stem.process.DEFAULT_INIT_TIMEOUT) is not None:
        raise AssertionError('Launching tor with the default timeout should be successful')
コード例 #20
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_torrc_arguments_via_stdin(tor_cmd):
    """
    Pass configuration options via stdin.
    """

    if test.tor_version() < stem.version.Requirement.TORRC_VIA_STDIN:
      skip('(requires )' % stem.version.Requirement.TORRC_VIA_STDIN)

    with tmp_directory() as data_directory:
      torrc = BASIC_RELAY_TORRC % data_directory
      output = run_tor(tor_cmd, '-f', '-', '--dump-config', 'short', stdin = torrc)
      assert_equal(sorted(torrc.splitlines()), sorted(output.splitlines()))
コード例 #21
0
ファイル: process.py プロジェクト: teor2345/stem
    def test_torrc_arguments_via_stdin(tor_cmd):
        """
    Pass configuration options via stdin.
    """

        if test.tor_version() < stem.version.Requirement.TORRC_VIA_STDIN:
            skip('(requires %s)' % stem.version.Requirement.TORRC_VIA_STDIN)

        with tmp_directory() as data_directory:
            torrc = BASIC_RELAY_TORRC % data_directory
            output = run_tor(tor_cmd,
                             '-f',
                             '-',
                             '--dump-config',
                             'short',
                             stdin=torrc)
            assert_equal(sorted(torrc.splitlines()),
                         sorted(output.splitlines()))
コード例 #22
0
ファイル: process.py プロジェクト: txrproject/stem
    async def test_launch_tor_with_config_via_file(tor_cmd):
        """
    Exercises launch_tor_with_config when we write a torrc to disk.
    """

        with tempfile.TemporaryDirectory() as data_directory:
            control_port = random_port()
            control_socket, tor_process = None, None

            try:
                # Launch tor without a torrc, but with a control port. Confirms that this
                # works by checking that we're still able to access the new instance.

                with patch('stem.version.get_system_tor_version',
                           Mock(return_value=stem.version.Version('0.0.0.1'))):
                    tor_process = stem.process.launch_tor_with_config(
                        tor_cmd=tor_cmd,
                        config={
                            'SocksPort': random_port(),
                            'ControlPort': control_port,
                            'DataDirectory': data_directory,
                        },
                        completion_percent=0)

                control_socket = stem.socket.ControlPort(
                    port=int(control_port))
                await stem.connection.authenticate(control_socket)

                # exercises the socket
                await control_socket.send('GETCONF ControlPort')
                getconf_response = await control_socket.recv()

                assert_equal('ControlPort=%s' % control_port,
                             str(getconf_response))
            finally:
                if control_socket:
                    await control_socket.close()

                if tor_process:
                    tor_process.kill()
                    tor_process.wait()
コード例 #23
0
ファイル: process.py プロジェクト: zmchu/stem
  def test_list_fingerprint_argument(tor_cmd):
    """
    Exercise our 'tor --list-fingerprint' argument.
    """

    # This command should only work with a relay (which our test instance isn't).

    output = run_tor(tor_cmd, '--list-fingerprint', with_torrc = True, expect_failure = True)
    assert_in("Clients don't have long-term identity keys. Exiting.", output, 'Should fail to start due to lacking an ORPort')

    with tempfile.TemporaryDirectory() as data_directory:
      torrc_path = os.path.join(data_directory, 'torrc')

      with open(torrc_path, 'w') as torrc_file:
        torrc_file.write(BASIC_RELAY_TORRC % data_directory + '\nORPort 6954')

      output = run_tor(tor_cmd, '--list-fingerprint', '-f', torrc_path)
      nickname, fingerprint_with_spaces = output.splitlines()[-1].split(' ', 1)
      fingerprint = fingerprint_with_spaces.replace(' ', '')

      assert_equal('stemIntegTest', nickname)
      assert_equal(49, len(fingerprint_with_spaces))

      if not stem.util.tor_tools.is_valid_fingerprint(fingerprint):
        raise AssertionError('We should have a valid fingerprint: %s' % fingerprint)

      with open(os.path.join(data_directory, 'fingerprint')) as fingerprint_file:
        expected = 'stemIntegTest %s\n' % fingerprint
        assert_equal(expected, fingerprint_file.read())
コード例 #24
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_list_fingerprint_argument(tor_cmd):
    """
    Exercise our 'tor --list-fingerprint' argument.
    """

    # This command should only work with a relay (which our test instance isn't).

    output = run_tor(tor_cmd, '--list-fingerprint', with_torrc = True, expect_failure = True)
    assert_in("Clients don't have long-term identity keys. Exiting.", output, 'Should fail to start due to lacking an ORPort')

    with tmp_directory() as data_directory:
      torrc_path = os.path.join(data_directory, 'torrc')

      with open(torrc_path, 'w') as torrc_file:
        torrc_file.write(BASIC_RELAY_TORRC % data_directory + '\nORPort 6954')

      output = run_tor(tor_cmd, '--list-fingerprint', '-f', torrc_path)
      nickname, fingerprint_with_spaces = output.splitlines()[-1].split(' ', 1)
      fingerprint = fingerprint_with_spaces.replace(' ', '')

      assert_equal('stemIntegTest', nickname)
      assert_equal(49, len(fingerprint_with_spaces))

      if not stem.util.tor_tools.is_valid_fingerprint(fingerprint):
        raise AssertionError('We should have a valid fingerprint: %s' % fingerprint)

      with open(os.path.join(data_directory, 'fingerprint')) as fingerprint_file:
        expected = 'stemIntegTest %s\n' % fingerprint
        assert_equal(expected, fingerprint_file.read())
コード例 #25
0
ファイル: process.py プロジェクト: teor2345/stem
    def test_launch_tor_with_config_via_stdin(tor_cmd):
        """
    Exercises launch_tor_with_config when we provide our torrc via stdin.
    """

        if test.tor_version() < stem.version.Requirement.TORRC_VIA_STDIN:
            skip('(requires %s)' % stem.version.Requirement.TORRC_VIA_STDIN)

        with tmp_directory() as data_directory:
            control_port = random_port()
            control_socket, tor_process = None, None

            try:
                tor_process = stem.process.launch_tor_with_config(
                    tor_cmd=tor_cmd,
                    config={
                        'SocksPort': random_port(),
                        'ControlPort': control_port,
                        'DataDirectory': data_directory,
                    },
                    completion_percent=0)

                control_socket = stem.socket.ControlPort(
                    port=int(control_port))
                stem.connection.authenticate(control_socket)

                # exercises the socket
                control_socket.send('GETCONF ControlPort')
                getconf_response = control_socket.recv()

                assert_equal('ControlPort=%s' % control_port,
                             str(getconf_response))
            finally:
                if control_socket:
                    control_socket.close()

                if tor_process:
                    tor_process.kill()
                    tor_process.wait()
コード例 #26
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_launch_tor_with_config_via_file(tor_cmd):
    """
    Exercises launch_tor_with_config when we write a torrc to disk.
    """

    with tmp_directory() as data_directory:
      control_port = random_port()
      control_socket, tor_process = None, None

      try:
        # Launch tor without a torrc, but with a control port. Confirms that this
        # works by checking that we're still able to access the new instance.

        with patch('stem.version.get_system_tor_version', Mock(return_value = stem.version.Version('0.0.0.1'))):
          tor_process = stem.process.launch_tor_with_config(
            tor_cmd = tor_cmd,
            config = {
              'SocksPort': random_port(),
              'ControlPort': control_port,
              'DataDirectory': data_directory,
            },
            completion_percent = 5
          )

        control_socket = stem.socket.ControlPort(port = int(control_port))
        stem.connection.authenticate(control_socket)

        # exercises the socket
        control_socket.send('GETCONF ControlPort')
        getconf_response = control_socket.recv()

        assert_equal('ControlPort=%s' % control_port, str(getconf_response))
      finally:
        if control_socket:
          control_socket.close()

        if tor_process:
          tor_process.kill()
          tor_process.wait()
コード例 #27
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_launch_tor_with_config_via_stdin(tor_cmd):
    """
    Exercises launch_tor_with_config when we provide our torrc via stdin.
    """

    if test.tor_version() < stem.version.Requirement.TORRC_VIA_STDIN:
      skip('(requires )' % stem.version.Requirement.TORRC_VIA_STDIN)

    with tmp_directory() as data_directory:
      control_port = random_port()
      control_socket, tor_process = None, None

      try:
        tor_process = stem.process.launch_tor_with_config(
          tor_cmd = tor_cmd,
          config = {
            'SocksPort': random_port(),
            'ControlPort': control_port,
            'DataDirectory': data_directory,
          },
          completion_percent = 5
        )

        control_socket = stem.socket.ControlPort(port = int(control_port))
        stem.connection.authenticate(control_socket)

        # exercises the socket
        control_socket.send('GETCONF ControlPort')
        getconf_response = control_socket.recv()

        assert_equal('ControlPort=%s' % control_port, str(getconf_response))
      finally:
        if control_socket:
          control_socket.close()

        if tor_process:
          tor_process.kill()
          tor_process.wait()
コード例 #28
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_no_orphaned_process(tor_cmd):
    """
    Check that when an exception arises in the middle of spawning tor that we
    don't leave a lingering process.
    """

    if not stem.util.system.is_available('sleep'):
      skip('(sleep unavailable)')

    with patch('re.compile', Mock(side_effect = KeyboardInterrupt('nope'))):
      # We don't need to actually run tor for this test. Rather, any process will
      # do the trick. Picking sleep so this'll clean itself up if our test fails.

      mock_tor_process = subprocess.Popen(['sleep', '60'])

      with patch('subprocess.Popen', Mock(return_value = mock_tor_process)):
        try:
          stem.process.launch_tor(tor_cmd)
          raise AssertionError("tor shoudn't have started")
        except KeyboardInterrupt as exc:
          if os.path.exists('/proc/%s' % mock_tor_process.pid):
            raise AssertionError('launch_tor() left a lingering tor process')

          assert_equal('nope', str(exc))
コード例 #29
0
ファイル: process.py プロジェクト: teor2345/stem
    def test_unanonymous_hidden_service_config_must_match(tor_cmd):
        """
    Checking that startup fails if HiddenServiceNonAnonymousMode and
    HiddenServiceSingleHopMode don't match.
    """

        try:
            stem.process.launch_tor_with_config(
                tor_cmd=tor_cmd,
                config={'HiddenServiceNonAnonymousMode': '1'},
            )

            raise AssertionError(
                "Tor shouldn't start with 'HiddenServiceNonAnonymousMode' set but not 'HiddenServiceSingleHopMode'"
            )
        except OSError as exc:
            if test.tor_version(
            ) >= stem.version.Requirement.ADD_ONION_NON_ANONYMOUS:
                assert_equal(
                    'Process terminated: HiddenServiceNonAnonymousMode does not provide any server anonymity. It must be used with HiddenServiceSingleHopMode set to 1.',
                    str(exc))
            else:
                assert_equal(
                    "Process terminated: Unknown option 'HiddenServiceNonAnonymousMode'.  Failing.",
                    str(exc))

        try:
            stem.process.launch_tor_with_config(
                tor_cmd=tor_cmd,
                config={'HiddenServiceSingleHopMode': '1'},
            )

            raise AssertionError(
                "Tor shouldn't start with 'HiddenServiceSingleHopMode' set but not 'HiddenServiceNonAnonymousMode'"
            )
        except OSError as exc:
            if test.tor_version(
            ) >= stem.version.Requirement.ADD_ONION_NON_ANONYMOUS:
                assert_equal(
                    'Process terminated: HiddenServiceSingleHopMode does not provide any server anonymity. It must be used with HiddenServiceNonAnonymousMode set to 1.',
                    str(exc))
            else:
                assert_equal(
                    "Process terminated: Unknown option 'HiddenServiceSingleHopMode'.  Failing.",
                    str(exc))
コード例 #30
0
ファイル: process.py プロジェクト: patrickod/stem
  def test_version_argument(tor_cmd):
    """
    Check that 'tor --version' matches 'GETINFO version'.
    """

    assert_equal('Tor version %s.\n' % test.tor_version(), run_tor(tor_cmd, '--version'))