19
19
Only one dependency: ctypes should be installed.
22
from __future__ import absolute_import
27
from bzrlib import cmdline
30
if sys.platform == 'win32':
31
_major,_minor,_build,_platform,_text = sys.getwindowsversion()
34
# The operating system platform.
35
# This member can be one of the following values.
36
# ========================== ======================================
38
# -------------------------- --------------------------------------
39
# VER_PLATFORM_WIN32_NT The operating system is Windows Vista,
40
# 2 Windows Server "Longhorn",
41
# Windows Server 2003, Windows XP,
42
# Windows 2000, or Windows NT.
44
# VER_PLATFORM_WIN32_WINDOWS The operating system is Windows Me,
45
# 1 Windows 98, or Windows 95.
46
# ========================== ======================================
50
# don't care about real Windows name, just to force safe operations
33
from breezy.i18n import gettext
34
from breezy.sixish import text_type
56
36
# We can cope without it; use a separate variable to help pyflakes
149
131
proc = win32process.GetCurrentProcess()
150
132
info = win32process.GetProcessMemoryInfo(proc)
152
trace.note('Cannot debug memory on win32 without ctypes'
134
trace.note(gettext('Cannot debug memory on win32 without ctypes'
156
138
# using base-2 units (see HACKING.txt).
157
trace.note('WorkingSize %7dKiB'
158
'\tPeakWorking %7dKiB\t%s',
139
trace.note(gettext('WorkingSize {0:>7}KiB'
140
'\tPeakWorking {1:>7}KiB\t{2}').format(
159
141
info['WorkingSetSize'] / 1024,
160
142
info['PeakWorkingSetSize'] / 1024,
164
146
trace.note('%s', message)
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',
147
trace.note(gettext('WorkingSize %8d KiB'), info['WorkingSetSize'] / 1024)
148
trace.note(gettext('PeakWorking %8d KiB'), info['PeakWorkingSetSize'] / 1024)
149
trace.note(gettext('PagefileUsage %8d KiB'), info.get('PagefileUsage', 0) / 1024)
150
trace.note(gettext('PeakPagefileUsage %8d KiB'),
169
151
info.get('PeakPagefileUsage', 0) / 1024)
170
trace.note('PrivateUsage %8d KiB', info.get('PrivateUsage', 0) / 1024)
171
trace.note('PageFaultCount %8d', info.get('PageFaultCount', 0))
152
trace.note(gettext('PrivateUsage %8d KiB'), info.get('PrivateUsage', 0) / 1024)
153
trace.note(gettext('PageFaultCount %8d'), info.get('PageFaultCount', 0))
174
156
def get_console_size(defaultx=80, defaulty=25):
242
224
one that moves with the user as they logon to different machines, and
243
225
a 'local' one that stays local to the machine. This returns the 'roaming'
244
226
directory, and thus is suitable for storing user-preferences, etc.
246
Returned value can be unicode or plain string.
247
To convert plain string to unicode use
248
s.decode(osutils.get_user_encoding())
249
(XXX - but see bug 262874, which asserts the correct encoding is 'mbcs')
251
228
appdata = _get_sh_special_folder_path(CSIDL_APPDATA)
255
appdata = os.environ.get('APPDATA')
258
# if we fall to this point we on win98
259
# at least try C:/WINDOWS/Application Data
260
windir = os.environ.get('windir')
262
appdata = os.path.join(windir, 'Application Data')
263
if os.path.isdir(appdata):
265
# did not find anything
231
# Use APPDATA if defined, will return None if not
232
return get_environ_unicode('APPDATA')
269
235
def get_local_appdata_location():
275
241
a 'local' one that stays local to the machine. This returns the 'local'
276
242
directory, and thus is suitable for caches, temp files and other things
277
243
which don't need to move with the user.
279
Returned value can be unicode or plain string.
280
To convert plain string to unicode use
281
s.decode(osutils.get_user_encoding())
282
(XXX - but see bug 262874, which asserts the correct encoding is 'mbcs')
284
245
local = _get_sh_special_folder_path(CSIDL_LOCAL_APPDATA)
287
248
# Vista supplies LOCALAPPDATA, but XP and earlier do not.
288
local = os.environ.get('LOCALAPPDATA')
249
local = get_environ_unicode('LOCALAPPDATA')
291
252
return get_appdata_location()
296
257
Assume on win32 it's the <My Documents> folder.
297
258
If location cannot be obtained return system drive root,
300
Returned value can be unicode or plain string.
301
To convert plain string to unicode use
302
s.decode(osutils.get_user_encoding())
304
261
home = _get_sh_special_folder_path(CSIDL_PERSONAL)
307
# try for HOME env variable
308
home = os.path.expanduser('~')
264
home = get_environ_unicode('HOME')
267
homepath = get_environ_unicode('HOMEPATH')
268
if homepath is not None:
269
return os.path.join(get_environ_unicode('HOMEDIR', ''), home)
311
270
# at least return windows root directory
312
windir = os.environ.get('windir')
271
windir = get_environ_unicode('WINDIR')
314
273
return os.path.splitdrive(windir)[0] + '/'
315
274
# otherwise C:\ is good enough for 98% users
275
return text_type('C:/')
319
278
def get_user_name():
320
279
"""Return user name as login name.
321
280
If name cannot be obtained return None.
323
Returned value can be unicode or plain string.
324
To convert plain string to unicode use
325
s.decode(osutils.get_user_encoding())
382
336
if (GetComputerName is not None
383
337
and GetComputerName(buf, ctypes.byref(n))):
385
# otherwise try env variables, which will be 'mbcs' encoded
386
# on Windows (Python doesn't expose the native win32 unicode environment)
388
# http://msdn.microsoft.com/en-us/library/aa246807.aspx
389
# environment variables should always be encoded in 'mbcs'.
391
return os.environ['COMPUTERNAME'].decode("mbcs")
396
def _ensure_unicode(s):
397
if s and type(s) != unicode:
398
from bzrlib import osutils
399
s = s.decode(osutils.get_user_encoding())
403
def get_appdata_location_unicode():
404
return _ensure_unicode(get_appdata_location())
406
def get_home_location_unicode():
407
return _ensure_unicode(get_home_location())
409
def get_user_name_unicode():
410
return _ensure_unicode(get_user_name())
412
def get_host_name_unicode():
413
return _ensure_unicode(get_host_name())
338
return extract_buffer(buf)
339
return get_environ_unicode('COMPUTERNAME')
416
342
def _ensure_with_dir(path):
511
436
def set_file_attr_hidden(path):
512
437
"""Set file attributes to hidden if possible"""
513
438
if has_win32file:
514
if winver != 'Windows 98':
515
SetFileAttributes = win32file.SetFileAttributesW
517
SetFileAttributes = win32file.SetFileAttributes
439
SetFileAttributes = win32file.SetFileAttributesW
519
441
SetFileAttributes(path, win32file.FILE_ATTRIBUTE_HIDDEN)
520
except pywintypes.error, e:
521
from bzrlib import trace
442
except pywintypes.error as e:
522
444
trace.mutter('Unable to set hidden attribute on %r: %s', path, e)
525
def _command_line_to_argv(command_line, single_quotes_allowed=False):
447
def _command_line_to_argv(command_line, argv, single_quotes_allowed=False):
526
448
"""Convert a Unicode command line into a list of argv arguments.
528
450
It performs wildcard expansion to make wildcards act closer to how they
536
458
:return: A list of unicode strings.
460
# First, split the command line
538
461
s = cmdline.Splitter(command_line, single_quotes_allowed=single_quotes_allowed)
539
# Now that we've split the content, expand globs if necessary
463
# Bug #587868 Now make sure that the length of s agrees with sys.argv
464
# we do this by simply counting the number of arguments in each. The counts should
465
# agree no matter what encoding sys.argv is in (AFAIK)
466
# len(arguments) < len(sys.argv) should be an impossibility since python gets
467
# args from the very same PEB as does GetCommandLineW
470
# Now shorten the command line we get from GetCommandLineW to match sys.argv
471
if len(arguments) < len(argv):
472
raise AssertionError("Split command line can't be shorter than argv")
473
arguments = arguments[len(arguments) - len(argv):]
475
# Carry on to process globs (metachars) in the command line
476
# expand globs if necessary
540
477
# TODO: Use 'globbing' instead of 'glob.glob', this gives us stuff like
541
478
# '**/' style globs
543
for is_quoted, arg in s:
480
for is_quoted, arg in arguments:
544
481
if is_quoted or not glob.has_magic(arg):
557
494
if command_line is None:
558
495
raise ctypes.WinError()
559
496
# Skip the first argument, since we only care about parameters
560
argv = _command_line_to_argv(command_line)[1:]
561
if getattr(sys, 'frozen', None) is None:
562
# Invoked via 'python.exe' which takes the form:
563
# python.exe [PYTHON_OPTIONS] C:\Path\bzr [BZR_OPTIONS]
564
# we need to get only BZR_OPTIONS part,
565
# We already removed 'python.exe' so we remove everything up to and
566
# including the first non-option ('-') argument.
567
for idx in xrange(len(argv)):
568
if argv[idx][:1] != '-':
497
argv = _command_line_to_argv(command_line, sys.argv)[1:]
573
get_unicode_argv = None
501
def get_environ_unicode(key, default=None):
502
"""Get `key` from environment as unicode or `default` if unset
504
The environment is natively unicode on modern windows versions but
505
Python 2 only accesses it through the legacy bytestring api.
507
Environmental variable names are case insenstive on Windows.
509
A large enough buffer will be allocated to retrieve the value, though
510
it may take two calls to the underlying library function.
512
This needs ctypes because pywin32 does not expose the wide version.
514
cfunc = getattr(get_environ_unicode, "_c_function", None)
516
from ctypes.wintypes import DWORD, LPCWSTR, LPWSTR
517
cfunc = ctypes.WINFUNCTYPE(DWORD, LPCWSTR, LPWSTR, DWORD)(
518
("GetEnvironmentVariableW", ctypes.windll.kernel32))
519
get_environ_unicode._c_function = cfunc
520
buffer_size = 256 # heuristic, 256 characters often enough
522
buffer = ctypes.create_unicode_buffer(buffer_size)
523
length = cfunc(key, buffer, buffer_size)
525
code = ctypes.GetLastError()
526
if code == 203: # ERROR_ENVVAR_NOT_FOUND
528
raise ctypes.WinError(code)
529
if buffer_size > length:
530
return buffer[:length]
535
def _pywin32_is_local_pid_dead(pid):
536
"""True if pid doesn't correspond to live process on this machine"""
538
handle = win32api.OpenProcess(1, False, pid) # PROCESS_TERMINATE
539
except pywintypes.error as e:
540
if e[0] == 5: # ERROR_ACCESS_DENIED
541
# Probably something alive we're not allowed to kill
543
elif e[0] == 87: # ERROR_INVALID_PARAMETER
548
is_local_pid_dead = _pywin32_is_local_pid_dead
549
elif has_ctypes and sys.platform == 'win32':
550
from ctypes.wintypes import BOOL, DWORD, HANDLE
551
_kernel32 = ctypes.windll.kernel32
552
_CloseHandle = ctypes.WINFUNCTYPE(BOOL, HANDLE)(
553
("CloseHandle", _kernel32))
554
_OpenProcess = ctypes.WINFUNCTYPE(HANDLE, DWORD, BOOL, DWORD)(
555
("OpenProcess", _kernel32))
556
def _ctypes_is_local_pid_dead(pid):
557
"""True if pid doesn't correspond to live process on this machine"""
558
handle = _OpenProcess(1, False, pid) # PROCESS_TERMINATE
560
errorcode = ctypes.GetLastError()
561
if errorcode == 5: # ERROR_ACCESS_DENIED
562
# Probably something alive we're not allowed to kill
564
elif errorcode == 87: # ERROR_INVALID_PARAMETER
566
raise ctypes.WinError(errorcode)
569
is_local_pid_dead = _ctypes_is_local_pid_dead
572
def _is_pywintypes_error(evalue):
573
"""True if exception instance is an error from pywin32"""
574
if has_pywintypes and isinstance(evalue, pywintypes.error):