13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
"""Win32-specific helper functions
19
19
Only one dependency: ctypes should be installed.
28
from bzrlib import cmdline
28
31
if sys.platform == 'win32':
97
101
MAX_COMPUTERNAME_LENGTH = 31
103
# Registry data type ids
108
def debug_memory_win32api(message='', short=True):
109
"""Use trace.note() to dump the running memory info."""
110
from bzrlib import trace
112
class PROCESS_MEMORY_COUNTERS_EX(ctypes.Structure):
113
"""Used by GetProcessMemoryInfo"""
114
_fields_ = [('cb', ctypes.c_ulong),
115
('PageFaultCount', ctypes.c_ulong),
116
('PeakWorkingSetSize', ctypes.c_size_t),
117
('WorkingSetSize', ctypes.c_size_t),
118
('QuotaPeakPagedPoolUsage', ctypes.c_size_t),
119
('QuotaPagedPoolUsage', ctypes.c_size_t),
120
('QuotaPeakNonPagedPoolUsage', ctypes.c_size_t),
121
('QuotaNonPagedPoolUsage', ctypes.c_size_t),
122
('PagefileUsage', ctypes.c_size_t),
123
('PeakPagefileUsage', ctypes.c_size_t),
124
('PrivateUsage', ctypes.c_size_t),
126
cur_process = ctypes.windll.kernel32.GetCurrentProcess()
127
mem_struct = PROCESS_MEMORY_COUNTERS_EX()
128
ret = ctypes.windll.psapi.GetProcessMemoryInfo(cur_process,
129
ctypes.byref(mem_struct),
130
ctypes.sizeof(mem_struct))
132
trace.note('Failed to GetProcessMemoryInfo()')
134
info = {'PageFaultCount': mem_struct.PageFaultCount,
135
'PeakWorkingSetSize': mem_struct.PeakWorkingSetSize,
136
'WorkingSetSize': mem_struct.WorkingSetSize,
137
'QuotaPeakPagedPoolUsage': mem_struct.QuotaPeakPagedPoolUsage,
138
'QuotaPagedPoolUsage': mem_struct.QuotaPagedPoolUsage,
139
'QuotaPeakNonPagedPoolUsage':
140
mem_struct.QuotaPeakNonPagedPoolUsage,
141
'QuotaNonPagedPoolUsage': mem_struct.QuotaNonPagedPoolUsage,
142
'PagefileUsage': mem_struct.PagefileUsage,
143
'PeakPagefileUsage': mem_struct.PeakPagefileUsage,
144
'PrivateUsage': mem_struct.PrivateUsage,
148
# win32process does not return PrivateUsage, because it doesn't use
149
# PROCESS_MEMORY_COUNTERS_EX (it uses the one without _EX).
150
proc = win32process.GetCurrentProcess()
151
info = win32process.GetProcessMemoryInfo(proc)
153
trace.note('Cannot debug memory on win32 without ctypes'
157
# using base-2 units (see HACKING.txt).
158
trace.note('WorkingSize %7dKiB'
159
'\tPeakWorking %7dKiB\t%s',
160
info['WorkingSetSize'] / 1024,
161
info['PeakWorkingSetSize'] / 1024,
165
trace.note('%s', message)
166
trace.note('WorkingSize %8d KiB', info['WorkingSetSize'] / 1024)
167
trace.note('PeakWorking %8d KiB', info['PeakWorkingSetSize'] / 1024)
168
trace.note('PagefileUsage %8d KiB', info.get('PagefileUsage', 0) / 1024)
169
trace.note('PeakPagefileUsage %8d KiB',
170
info.get('PeakPagefileUsage', 0) / 1024)
171
trace.note('PrivateUsage %8d KiB', info.get('PrivateUsage', 0) / 1024)
172
trace.note('PageFaultCount %8d', info.get('PageFaultCount', 0))
100
175
def get_console_size(defaultx=80, defaulty=25):
101
176
"""Return size of current console.
109
184
return (defaultx, defaulty)
111
186
# To avoid problem with redirecting output via pipe
112
# need to use stderr instead of stdout
187
# we need to use stderr instead of stdout
113
188
h = ctypes.windll.kernel32.GetStdHandle(WIN32_STDERR_HANDLE)
114
189
csbi = ctypes.create_string_buffer(22)
115
190
res = ctypes.windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
118
193
(bufx, bufy, curx, cury, wattr,
119
left, top, right, bottom, maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)
194
left, top, right, bottom, maxx, maxy) = struct.unpack(
195
"hhhhHhhhhhh", csbi.raw)
120
196
sizex = right - left + 1
121
197
sizey = bottom - top + 1
122
198
return (sizex, sizey)
432
def glob_one(possible_glob):
433
"""Same as glob.glob().
435
work around bugs in glob.glob()
436
- Python bug #1001604 ("glob doesn't return unicode with ...")
437
- failing expansion for */* with non-iso-8859-* chars
439
corrected_glob, corrected = _ensure_with_dir(possible_glob)
440
glob_files = glob.glob(corrected_glob)
443
# special case to let the normal code path handle
444
# files that do not exist, etc.
445
glob_files = [possible_glob]
447
glob_files = [_undo_ensure_with_dir(elem, corrected)
448
for elem in glob_files]
449
return [elem.replace(u'\\', u'/') for elem in glob_files]
355
452
def glob_expand(file_list):
356
453
"""Replacement for glob expansion by the shell.
366
463
if not file_list:
369
465
expanded_file_list = []
370
466
for possible_glob in file_list:
372
# work around bugs in glob.glob()
373
# - Python bug #1001604 ("glob doesn't return unicode with ...")
374
# - failing expansion for */* with non-iso-8859-* chars
375
possible_glob, corrected = _ensure_with_dir(possible_glob)
376
glob_files = glob.glob(possible_glob)
379
# special case to let the normal code path handle
380
# files that do not exists
381
expanded_file_list.append(
382
_undo_ensure_with_dir(possible_glob, corrected))
384
glob_files = [_undo_ensure_with_dir(elem, corrected) for elem in glob_files]
385
expanded_file_list += glob_files
387
return [elem.replace(u'\\', u'/') for elem in expanded_file_list]
467
expanded_file_list.extend(glob_one(possible_glob))
468
return expanded_file_list
390
471
def get_app_path(appname):
391
472
"""Look up in Windows registry for full path to application executable.
392
Typicaly, applications create subkey with their basename
473
Typically, applications create subkey with their basename
393
474
in HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\App Paths\
395
476
:param appname: name of application (if no filename extension
398
479
or appname itself if nothing found.
484
if not os.path.splitext(basename)[1]:
485
basename = appname + '.exe'
402
488
hkey = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE,
403
r'SOFTWARE\Microsoft\Windows'
404
r'\CurrentVersion\App Paths')
489
'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths\\' +
405
491
except EnvironmentError:
409
if not os.path.splitext(basename)[1]:
410
basename = appname + '.exe'
413
fullpath = _winreg.QueryValue(hkey, basename)
496
path, type_id = _winreg.QueryValueEx(hkey, '')
414
497
except WindowsError:
417
500
_winreg.CloseKey(hkey)
502
if type_id == REG_SZ:
504
if type_id == REG_EXPAND_SZ and has_win32api:
505
fullpath = win32api.ExpandEnvironmentStrings(path)
506
if len(fullpath) > 1 and fullpath[0] == '"' and fullpath[-1] == '"':
507
fullpath = fullpath[1:-1] # remove quotes around value
422
512
def set_file_attr_hidden(path):
423
513
"""Set file attributes to hidden if possible"""
424
514
if has_win32file:
425
win32file.SetFileAttributes(path, win32file.FILE_ATTRIBUTE_HIDDEN)
515
if winver != 'Windows 98':
516
SetFileAttributes = win32file.SetFileAttributesW
518
SetFileAttributes = win32file.SetFileAttributes
520
SetFileAttributes(path, win32file.FILE_ATTRIBUTE_HIDDEN)
521
except pywintypes.error, e:
522
from bzrlib import trace
523
trace.mutter('Unable to set hidden attribute on %r: %s', path, e)
526
def _command_line_to_argv(command_line, single_quotes_allowed=False):
527
"""Convert a Unicode command line into a list of argv arguments.
529
It performs wildcard expansion to make wildcards act closer to how they
530
work in posix shells, versus how they work by default on Windows. Quoted
531
arguments are left untouched.
533
:param command_line: The unicode string to split into an arg list.
534
:param single_quotes_allowed: Whether single quotes are accepted as quoting
535
characters like double quotes. False by
537
:return: A list of unicode strings.
539
s = cmdline.Splitter(command_line, single_quotes_allowed=single_quotes_allowed)
540
# Now that we've split the content, expand globs if necessary
541
# TODO: Use 'globbing' instead of 'glob.glob', this gives us stuff like
544
for is_quoted, arg in s:
545
if is_quoted or not glob.has_magic(arg):
548
args.extend(glob_one(arg))
552
if has_ctypes and winver != 'Windows 98':
553
def get_unicode_argv():
554
prototype = ctypes.WINFUNCTYPE(ctypes.c_wchar_p)
555
GetCommandLineW = prototype(("GetCommandLineW",
556
ctypes.windll.kernel32))
557
command_line = GetCommandLineW()
558
if command_line is None:
559
raise ctypes.WinError()
560
# Skip the first argument, since we only care about parameters
561
argv = _command_line_to_argv(command_line)[1:]
562
if getattr(sys, 'frozen', None) is None:
563
# Invoked via 'python.exe' which takes the form:
564
# python.exe [PYTHON_OPTIONS] C:\Path\bzr [BZR_OPTIONS]
565
# we need to get only BZR_OPTIONS part,
566
# We already removed 'python.exe' so we remove everything up to and
567
# including the first non-option ('-') argument.
568
for idx in xrange(len(argv)):
569
if argv[idx][:1] != '-':
574
get_unicode_argv = None
added
removed
bzrlib/win32utils.py
