/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/win32utils.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2005-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
21
21
 
22
22
import glob
23
23
import os
24
 
import re
25
 
import shlex
26
24
import struct
27
 
import StringIO
28
25
import sys
29
26
 
 
27
from bzrlib import cmdline
30
28
 
31
29
# Windows version
32
30
if sys.platform == 'win32':
137
135
                'WorkingSetSize': mem_struct.WorkingSetSize,
138
136
                'QuotaPeakPagedPoolUsage': mem_struct.QuotaPeakPagedPoolUsage,
139
137
                'QuotaPagedPoolUsage': mem_struct.QuotaPagedPoolUsage,
140
 
                'QuotaPeakNonPagedPoolUsage': mem_struct.QuotaPeakNonPagedPoolUsage,
 
138
                'QuotaPeakNonPagedPoolUsage':
 
139
                    mem_struct.QuotaPeakNonPagedPoolUsage,
141
140
                'QuotaNonPagedPoolUsage': mem_struct.QuotaNonPagedPoolUsage,
142
141
                'PagefileUsage': mem_struct.PagefileUsage,
143
142
                'PeakPagefileUsage': mem_struct.PeakPagefileUsage,
154
153
                   ' or win32process')
155
154
        return
156
155
    if short:
157
 
        trace.note('WorkingSize %7dKB'
158
 
                   '\tPeakWorking %7dKB\t%s',
 
156
        # using base-2 units (see HACKING.txt).
 
157
        trace.note('WorkingSize %7dKiB'
 
158
                   '\tPeakWorking %7dKiB\t%s',
159
159
                   info['WorkingSetSize'] / 1024,
160
160
                   info['PeakWorkingSetSize'] / 1024,
161
161
                   message)
162
162
        return
163
163
    if message:
164
164
        trace.note('%s', message)
165
 
    trace.note('WorkingSize       %8d KB', info['WorkingSetSize'] / 1024)
166
 
    trace.note('PeakWorking       %8d KB', info['PeakWorkingSetSize'] / 1024)
167
 
    trace.note('PagefileUsage     %8d KB', info.get('PagefileUsage', 0) / 1024)
168
 
    trace.note('PeakPagefileUsage %8d KB', info.get('PeakPagefileUsage', 0) / 1024)
169
 
    trace.note('PrivateUsage      %8d KB', info.get('PrivateUsage', 0) / 1024)
 
165
    trace.note('WorkingSize       %8d KiB', info['WorkingSetSize'] / 1024)
 
166
    trace.note('PeakWorking       %8d KiB', info['PeakWorkingSetSize'] / 1024)
 
167
    trace.note('PagefileUsage     %8d KiB', info.get('PagefileUsage', 0) / 1024)
 
168
    trace.note('PeakPagefileUsage %8d KiB',
 
169
               info.get('PeakPagefileUsage', 0) / 1024)
 
170
    trace.note('PrivateUsage      %8d KiB', info.get('PrivateUsage', 0) / 1024)
170
171
    trace.note('PageFaultCount    %8d', info.get('PageFaultCount', 0))
171
172
 
172
173
 
182
183
        return (defaultx, defaulty)
183
184
 
184
185
    # To avoid problem with redirecting output via pipe
185
 
    # need to use stderr instead of stdout
 
186
    # we need to use stderr instead of stdout
186
187
    h = ctypes.windll.kernel32.GetStdHandle(WIN32_STDERR_HANDLE)
187
188
    csbi = ctypes.create_string_buffer(22)
188
189
    res = ctypes.windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
189
190
 
190
191
    if res:
191
192
        (bufx, bufy, curx, cury, wattr,
192
 
        left, top, right, bottom, maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)
 
193
        left, top, right, bottom, maxx, maxy) = struct.unpack(
 
194
            "hhhhHhhhhhh", csbi.raw)
193
195
        sizex = right - left + 1
194
196
        sizey = bottom - top + 1
195
197
        return (sizex, sizey)
392
394
 
393
395
 
394
396
def _ensure_unicode(s):
395
 
    from bzrlib import osutils
396
397
    if s and type(s) != unicode:
397
398
        from bzrlib import osutils
398
399
        s = s.decode(osutils.get_user_encoding())
413
414
 
414
415
 
415
416
def _ensure_with_dir(path):
416
 
    if not os.path.split(path)[0] or path.startswith(u'*') or path.startswith(u'?'):
 
417
    if (not os.path.split(path)[0] or path.startswith(u'*')
 
418
        or path.startswith(u'?')):
417
419
        return u'./' + path, True
418
420
    else:
419
421
        return path, False
520
522
            trace.mutter('Unable to set hidden attribute on %r: %s', path, e)
521
523
 
522
524
 
523
 
 
524
 
class UnicodeShlex(object):
525
 
    """This is a very simplified version of shlex.shlex.
526
 
 
527
 
    The main change is that it supports non-ascii input streams. The internal
528
 
    structure is quite simplified relative to shlex.shlex, since we aren't
529
 
    trying to handle multiple input streams, etc. In fact, we don't use a
530
 
    file-like api either.
531
 
    """
532
 
 
533
 
    def __init__(self, uni_string):
534
 
        self._input = uni_string
535
 
        self._input_iter = iter(self._input)
536
 
        self._whitespace_match = re.compile(u'\s').match
537
 
        self._word_match = re.compile(u'\S').match
538
 
        self._quote_chars = u'"'
539
 
        # self._quote_match = re.compile(u'[\'"]').match
540
 
        self._escape_match = lambda x: None # Never matches
541
 
        self._escape = '\\'
542
 
        # State can be
543
 
        #   ' ' - after whitespace, starting a new token
544
 
        #   'a' - after text, currently working on a token
545
 
        #   '"' - after ", currently in a "-delimited quoted section
546
 
        #   "\" - after '\', checking the next char
547
 
        self._state = ' '
548
 
        self._token = [] # Current token being parsed
549
 
 
550
 
    def _get_token(self):
551
 
        # Were there quote chars as part of this token?
552
 
        quoted = False
553
 
        quoted_state = None
554
 
        for nextchar in self._input_iter:
555
 
            if self._state == ' ':
556
 
                if self._whitespace_match(nextchar):
557
 
                    # if self._token: return token
558
 
                    continue
559
 
                elif nextchar in self._quote_chars:
560
 
                    self._state = nextchar # quoted state
561
 
                elif self._word_match(nextchar):
562
 
                    self._token.append(nextchar)
563
 
                    self._state = 'a'
564
 
                else:
565
 
                    raise AssertionError('wtttf?')
566
 
            elif self._state in self._quote_chars:
567
 
                quoted = True
568
 
                if nextchar == self._state: # End of quote
569
 
                    self._state = 'a' # posix allows 'foo'bar to translate to
570
 
                                      # foobar
571
 
                elif self._state == '"' and nextchar == self._escape:
572
 
                    quoted_state = self._state
573
 
                    self._state = nextchar
574
 
                else:
575
 
                    self._token.append(nextchar)
576
 
            elif self._state == self._escape:
577
 
                if nextchar == '\\':
578
 
                    self._token.append('\\')
579
 
                elif nextchar == '"':
580
 
                    self._token.append(nextchar)
581
 
                else:
582
 
                    self._token.append('\\' + nextchar)
583
 
                self._state = quoted_state
584
 
            elif self._state == 'a':
585
 
                if self._whitespace_match(nextchar):
586
 
                    if self._token:
587
 
                        break # emit this token
588
 
                    else:
589
 
                        continue # no token to emit
590
 
                elif nextchar in self._quote_chars:
591
 
                    # Start a new quoted section
592
 
                    self._state = nextchar
593
 
                # escape?
594
 
                elif (self._word_match(nextchar)
595
 
                      or nextchar in self._quote_chars
596
 
                      # or whitespace_split?
597
 
                      ):
598
 
                    self._token.append(nextchar)
599
 
                else:
600
 
                    raise AssertionError('state == "a", char: %r'
601
 
                                         % (nextchar,))
602
 
            else:
603
 
                raise AssertionError('unknown state: %r' % (self._state,))
604
 
        result = ''.join(self._token)
605
 
        self._token = []
606
 
        if not quoted and result == '':
607
 
            result = None
608
 
        return quoted, result
609
 
 
610
 
    def __iter__(self):
611
 
        return self
612
 
 
613
 
    def next(self):
614
 
        quoted, token = self._get_token()
615
 
        if token is None:
616
 
            raise StopIteration
617
 
        return quoted, token
618
 
 
619
 
 
620
 
def _command_line_to_argv(command_line):
621
 
    """Convert a Unicode command line into a set of argv arguments.
622
 
 
623
 
    This does wildcard expansion, etc. It is intended to make wildcards act
624
 
    closer to how they work in posix shells, versus how they work by default on
625
 
    Windows.
626
 
    """
627
 
    s = UnicodeShlex(command_line)
628
 
    # Now that we've split the content, expand globs
 
525
def _command_line_to_argv(command_line, single_quotes_allowed=False):
 
526
    """Convert a Unicode command line into a list of argv arguments.
 
527
 
 
528
    It performs wildcard expansion to make wildcards act closer to how they
 
529
    work in posix shells, versus how they work by default on Windows. Quoted
 
530
    arguments are left untouched.
 
531
 
 
532
    :param command_line: The unicode string to split into an arg list.
 
533
    :param single_quotes_allowed: Whether single quotes are accepted as quoting
 
534
                                  characters like double quotes. False by
 
535
                                  default.
 
536
    :return: A list of unicode strings.
 
537
    """
 
538
    s = cmdline.Splitter(command_line, single_quotes_allowed=single_quotes_allowed)
 
539
    # Now that we've split the content, expand globs if necessary
629
540
    # TODO: Use 'globbing' instead of 'glob.glob', this gives us stuff like
630
541
    #       '**/' style globs
631
542
    args = []
632
543
    for is_quoted, arg in s:
633
544
        if is_quoted or not glob.has_magic(arg):
634
 
            args.append(arg.replace(u'\\', u'/'))
 
545
            args.append(arg)
635
546
        else:
636
547
            args.extend(glob_one(arg))
637
548
    return args
639
550
 
640
551
if has_ctypes and winver != 'Windows 98':
641
552
    def get_unicode_argv():
642
 
        LPCWSTR = ctypes.c_wchar_p
643
 
        INT = ctypes.c_int
644
 
        POINTER = ctypes.POINTER
645
 
        prototype = ctypes.WINFUNCTYPE(LPCWSTR)
646
 
        GetCommandLine = prototype(("GetCommandLineW",
647
 
                                    ctypes.windll.kernel32))
648
 
        prototype = ctypes.WINFUNCTYPE(POINTER(LPCWSTR), LPCWSTR, POINTER(INT))
649
 
        command_line = GetCommandLine()
 
553
        prototype = ctypes.WINFUNCTYPE(ctypes.c_wchar_p)
 
554
        GetCommandLineW = prototype(("GetCommandLineW",
 
555
                                     ctypes.windll.kernel32))
 
556
        command_line = GetCommandLineW()
 
557
        if command_line is None:
 
558
            raise ctypes.WinError()
650
559
        # Skip the first argument, since we only care about parameters
651
 
        argv = _command_line_to_argv(GetCommandLine())[1:]
 
560
        argv = _command_line_to_argv(command_line)[1:]
652
561
        if getattr(sys, 'frozen', None) is None:
653
562
            # Invoked via 'python.exe' which takes the form:
654
563
            #   python.exe [PYTHON_OPTIONS] C:\Path\bzr [BZR_OPTIONS]