##// END OF EJS Templates
sha1dc: initial implementation of Python extension...
Augie Fackler -
r44510:bde1cd4c default
parent child Browse files
Show More
@@ -0,0 +1,197 b''
1 #define PY_SSIZE_T_CLEAN
2 #include <Python.h>
3
4 #include "lib/sha1.h"
5
6 #if PY_MAJOR_VERSION >= 3
7 #define IS_PY3K
8 #endif
9
10 /* helper to switch things like string literal depending on Python version */
11 #ifdef IS_PY3K
12 #define PY23(py2, py3) py3
13 #else
14 #define PY23(py2, py3) py2
15 #endif
16
17 static char sha1dc_doc[] = "Efficient detection of SHA1 collision constructs.";
18
19 /* clang-format off */
20 typedef struct {
21 PyObject_HEAD
22 SHA1_CTX ctx;
23 } pysha1ctx;
24 /* clang-format on */
25
26 static int pysha1ctx_init(pysha1ctx *self, PyObject *args)
27 {
28 const char *data = NULL;
29 Py_ssize_t len;
30
31 SHA1DCInit(&(self->ctx));
32 /* We don't want "safe" sha1s, wherein sha1dc can give you a
33 different hash for something that's trying to give you a
34 collision. We just want to detect collisions.
35 */
36 SHA1DCSetSafeHash(&(self->ctx), 0);
37 if (!PyArg_ParseTuple(args, PY23("|s#", "|y#"), &data, &len)) {
38 return -1;
39 }
40 if (data) {
41 SHA1DCUpdate(&(self->ctx), data, len);
42 }
43 return 0;
44 }
45
46 static void pysha1ctx_dealloc(pysha1ctx *self)
47 {
48 PyObject_Del(self);
49 }
50
51 static PyObject *pysha1ctx_update(pysha1ctx *self, PyObject *args)
52 {
53 const char *data;
54 Py_ssize_t len;
55 if (!PyArg_ParseTuple(args, PY23("s#", "y#"), &data, &len)) {
56 return NULL;
57 }
58 SHA1DCUpdate(&(self->ctx), data, len);
59 Py_RETURN_NONE;
60 }
61
62 /* it is intentional that this take a ctx by value, as that clones the
63 context so we can keep using .update() without poisoning the state
64 with padding.
65 */
66 static int finalize(SHA1_CTX ctx, unsigned char *hash_out)
67 {
68 if (SHA1DCFinal(hash_out, &ctx)) {
69 PyErr_SetString(PyExc_OverflowError,
70 "sha1 collision attack detected");
71 return 0;
72 }
73 return 1;
74 }
75
76 static PyObject *pysha1ctx_digest(pysha1ctx *self)
77 {
78 unsigned char hash[20];
79 if (!finalize(self->ctx, hash)) {
80 return NULL;
81 }
82 return PyBytes_FromStringAndSize((char *)hash, 20);
83 }
84
85 static PyObject *pysha1ctx_hexdigest(pysha1ctx *self)
86 {
87 unsigned char hash[20];
88 if (!finalize(self->ctx, hash)) {
89 return NULL;
90 }
91 char hexhash[40];
92 static const char hexdigit[] = "0123456789abcdef";
93 for (int i = 0; i < 20; ++i) {
94 hexhash[i * 2] = hexdigit[hash[i] >> 4];
95 hexhash[i * 2 + 1] = hexdigit[hash[i] & 15];
96 }
97 return PyString_FromStringAndSize(hexhash, 40);
98 }
99
100 static PyTypeObject sha1ctxType;
101
102 static PyObject *pysha1ctx_copy(pysha1ctx *self)
103 {
104 pysha1ctx *clone = (pysha1ctx *)PyObject_New(pysha1ctx, &sha1ctxType);
105 if (!clone) {
106 return NULL;
107 }
108 clone->ctx = self->ctx;
109 return (PyObject *)clone;
110 }
111
112 static PyMethodDef pysha1ctx_methods[] = {
113 {"update", (PyCFunction)pysha1ctx_update, METH_VARARGS,
114 "Update this hash object's state with the provided bytes."},
115 {"digest", (PyCFunction)pysha1ctx_digest, METH_NOARGS,
116 "Return the digest value as a string of binary data."},
117 {"hexdigest", (PyCFunction)pysha1ctx_hexdigest, METH_NOARGS,
118 "Return the digest value as a string of hexadecimal digits."},
119 {"copy", (PyCFunction)pysha1ctx_copy, METH_NOARGS,
120 "Return a copy of the hash object."},
121 {NULL},
122 };
123
124 /* clang-format off */
125 static PyTypeObject sha1ctxType = {
126 PyVarObject_HEAD_INIT(NULL, 0) /* header */
127 "sha1dc.sha1", /* tp_name */
128 sizeof(pysha1ctx), /* tp_basicsize */
129 0, /* tp_itemsize */
130 (destructor)pysha1ctx_dealloc, /* tp_dealloc */
131 0, /* tp_print */
132 0, /* tp_getattr */
133 0, /* tp_setattr */
134 0, /* tp_compare */
135 0, /* tp_repr */
136 0, /* tp_as_number */
137 0, /* tp_as_sequence */
138 0, /* tp_as_mapping */
139 0, /* tp_hash */
140 0, /* tp_call */
141 0, /* tp_str */
142 0, /* tp_getattro */
143 0, /* tp_setattro */
144 0, /* tp_as_buffer */
145 Py_TPFLAGS_DEFAULT, /* tp_flags */
146 "sha1 implementation that looks for collisions", /* tp_doc */
147 0, /* tp_traverse */
148 0, /* tp_clear */
149 0, /* tp_richcompare */
150 0, /* tp_weaklistoffset */
151 0, /* tp_iter */
152 0, /* tp_iternext */
153 pysha1ctx_methods, /* tp_methods */
154 0, /* tp_members */
155 0, /* tp_getset */
156 0, /* tp_base */
157 0, /* tp_dict */
158 0, /* tp_descr_get */
159 0, /* tp_descr_set */
160 0, /* tp_dictoffset */
161 (initproc)pysha1ctx_init, /* tp_init */
162 0, /* tp_alloc */
163 };
164 /* clang-format on */
165
166 static PyMethodDef methods[] = {
167 {NULL, NULL},
168 };
169
170 static void module_init(PyObject *mod)
171 {
172 sha1ctxType.tp_new = PyType_GenericNew;
173 if (PyType_Ready(&sha1ctxType) < 0) {
174 return;
175 }
176 Py_INCREF(&sha1ctxType);
177
178 PyModule_AddObject(mod, "sha1", (PyObject *)&sha1ctxType);
179 }
180
181 #ifdef IS_PY3K
182 static struct PyModuleDef sha1dc_module = {PyModuleDef_HEAD_INIT, "sha1dc",
183 sha1dc_doc, -1, methods};
184
185 PyMODINIT_FUNC PyInit_sha1dc(void)
186 {
187 PyObject *mod = PyModule_Create(&sha1dc_module);
188 module_init(mod);
189 return mod;
190 }
191 #else
192 PyMODINIT_FUNC initsha1dc(void)
193 {
194 PyObject *mod = Py_InitModule3("sha1dc", methods, sha1dc_doc);
195 module_init(mod);
196 }
197 #endif
@@ -0,0 +1,60 b''
1 # Tests to ensure that sha1dc.sha1 is exactly a drop-in for
2 # hashlib.sha1 for our needs.
3 from __future__ import absolute_import
4
5 import hashlib
6 import unittest
7
8 import silenttestrunner
9
10 try:
11 from mercurial.thirdparty import sha1dc
12 except ImportError:
13 sha1dc = None
14
15
16 class hashertestsbase(object):
17 def test_basic_hash(self):
18 h = self.hasher()
19 h.update(b'foo')
20 self.assertEqual(
21 '0beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a33', h.hexdigest()
22 )
23 h.update(b'bar')
24 self.assertEqual(
25 '8843d7f92416211de9ebb963ff4ce28125932878', h.hexdigest()
26 )
27
28 def test_copy_hasher(self):
29 h = self.hasher()
30 h.update(b'foo')
31 h2 = h.copy()
32 h.update(b'baz')
33 h2.update(b'bar')
34 self.assertEqual(
35 '21eb6533733a5e4763acacd1d45a60c2e0e404e1', h.hexdigest()
36 )
37 self.assertEqual(
38 '8843d7f92416211de9ebb963ff4ce28125932878', h2.hexdigest()
39 )
40
41 def test_init_hasher(self):
42 h = self.hasher(b'initial string')
43 self.assertEqual(
44 b'\xc9y|n\x1f3S\xa4:\xbaJ\xca,\xc1\x1a\x9e\xb8\xd8\xdd\x86',
45 h.digest(),
46 )
47
48
49 class hashlibtests(unittest.TestCase, hashertestsbase):
50 hasher = hashlib.sha1
51
52
53 if sha1dc:
54
55 class sha1dctests(unittest.TestCase, hashertestsbase):
56 hasher = sha1dc.sha1
57
58
59 if __name__ == '__main__':
60 silenttestrunner.main(__name__)
@@ -1,1723 +1,1731 b''
1 #
1 #
2 # This is the mercurial setup script.
2 # This is the mercurial setup script.
3 #
3 #
4 # 'python setup.py install', or
4 # 'python setup.py install', or
5 # 'python setup.py --help' for more options
5 # 'python setup.py --help' for more options
6
6
7 import os
7 import os
8
8
9 # Mercurial will never work on Python 3 before 3.5 due to a lack
9 # Mercurial will never work on Python 3 before 3.5 due to a lack
10 # of % formatting on bytestrings, and can't work on 3.6.0 or 3.6.1
10 # of % formatting on bytestrings, and can't work on 3.6.0 or 3.6.1
11 # due to a bug in % formatting in bytestrings.
11 # due to a bug in % formatting in bytestrings.
12 # We cannot support Python 3.5.0, 3.5.1, 3.5.2 because of bug in
12 # We cannot support Python 3.5.0, 3.5.1, 3.5.2 because of bug in
13 # codecs.escape_encode() where it raises SystemError on empty bytestring
13 # codecs.escape_encode() where it raises SystemError on empty bytestring
14 # bug link: https://bugs.python.org/issue25270
14 # bug link: https://bugs.python.org/issue25270
15 supportedpy = ','.join(
15 supportedpy = ','.join(
16 [
16 [
17 '>=2.7',
17 '>=2.7',
18 '!=3.0.*',
18 '!=3.0.*',
19 '!=3.1.*',
19 '!=3.1.*',
20 '!=3.2.*',
20 '!=3.2.*',
21 '!=3.3.*',
21 '!=3.3.*',
22 '!=3.4.*',
22 '!=3.4.*',
23 '!=3.5.0',
23 '!=3.5.0',
24 '!=3.5.1',
24 '!=3.5.1',
25 '!=3.5.2',
25 '!=3.5.2',
26 '!=3.6.0',
26 '!=3.6.0',
27 '!=3.6.1',
27 '!=3.6.1',
28 ]
28 ]
29 )
29 )
30
30
31 import sys, platform
31 import sys, platform
32 import sysconfig
32 import sysconfig
33
33
34 if sys.version_info[0] >= 3:
34 if sys.version_info[0] >= 3:
35 printf = eval('print')
35 printf = eval('print')
36 libdir_escape = 'unicode_escape'
36 libdir_escape = 'unicode_escape'
37
37
38 def sysstr(s):
38 def sysstr(s):
39 return s.decode('latin-1')
39 return s.decode('latin-1')
40
40
41
41
42 else:
42 else:
43 libdir_escape = 'string_escape'
43 libdir_escape = 'string_escape'
44
44
45 def printf(*args, **kwargs):
45 def printf(*args, **kwargs):
46 f = kwargs.get('file', sys.stdout)
46 f = kwargs.get('file', sys.stdout)
47 end = kwargs.get('end', '\n')
47 end = kwargs.get('end', '\n')
48 f.write(b' '.join(args) + end)
48 f.write(b' '.join(args) + end)
49
49
50 def sysstr(s):
50 def sysstr(s):
51 return s
51 return s
52
52
53
53
54 # Attempt to guide users to a modern pip - this means that 2.6 users
54 # Attempt to guide users to a modern pip - this means that 2.6 users
55 # should have a chance of getting a 4.2 release, and when we ratchet
55 # should have a chance of getting a 4.2 release, and when we ratchet
56 # the version requirement forward again hopefully everyone will get
56 # the version requirement forward again hopefully everyone will get
57 # something that works for them.
57 # something that works for them.
58 if sys.version_info < (2, 7, 0, 'final'):
58 if sys.version_info < (2, 7, 0, 'final'):
59 pip_message = (
59 pip_message = (
60 'This may be due to an out of date pip. '
60 'This may be due to an out of date pip. '
61 'Make sure you have pip >= 9.0.1.'
61 'Make sure you have pip >= 9.0.1.'
62 )
62 )
63 try:
63 try:
64 import pip
64 import pip
65
65
66 pip_version = tuple([int(x) for x in pip.__version__.split('.')[:3]])
66 pip_version = tuple([int(x) for x in pip.__version__.split('.')[:3]])
67 if pip_version < (9, 0, 1):
67 if pip_version < (9, 0, 1):
68 pip_message = (
68 pip_message = (
69 'Your pip version is out of date, please install '
69 'Your pip version is out of date, please install '
70 'pip >= 9.0.1. pip {} detected.'.format(pip.__version__)
70 'pip >= 9.0.1. pip {} detected.'.format(pip.__version__)
71 )
71 )
72 else:
72 else:
73 # pip is new enough - it must be something else
73 # pip is new enough - it must be something else
74 pip_message = ''
74 pip_message = ''
75 except Exception:
75 except Exception:
76 pass
76 pass
77 error = """
77 error = """
78 Mercurial does not support Python older than 2.7.
78 Mercurial does not support Python older than 2.7.
79 Python {py} detected.
79 Python {py} detected.
80 {pip}
80 {pip}
81 """.format(
81 """.format(
82 py=sys.version_info, pip=pip_message
82 py=sys.version_info, pip=pip_message
83 )
83 )
84 printf(error, file=sys.stderr)
84 printf(error, file=sys.stderr)
85 sys.exit(1)
85 sys.exit(1)
86
86
87 if sys.version_info[0] >= 3:
87 if sys.version_info[0] >= 3:
88 DYLIB_SUFFIX = sysconfig.get_config_vars()['EXT_SUFFIX']
88 DYLIB_SUFFIX = sysconfig.get_config_vars()['EXT_SUFFIX']
89 else:
89 else:
90 # deprecated in Python 3
90 # deprecated in Python 3
91 DYLIB_SUFFIX = sysconfig.get_config_vars()['SO']
91 DYLIB_SUFFIX = sysconfig.get_config_vars()['SO']
92
92
93 # Solaris Python packaging brain damage
93 # Solaris Python packaging brain damage
94 try:
94 try:
95 import hashlib
95 import hashlib
96
96
97 sha = hashlib.sha1()
97 sha = hashlib.sha1()
98 except ImportError:
98 except ImportError:
99 try:
99 try:
100 import sha
100 import sha
101
101
102 sha.sha # silence unused import warning
102 sha.sha # silence unused import warning
103 except ImportError:
103 except ImportError:
104 raise SystemExit(
104 raise SystemExit(
105 "Couldn't import standard hashlib (incomplete Python install)."
105 "Couldn't import standard hashlib (incomplete Python install)."
106 )
106 )
107
107
108 try:
108 try:
109 import zlib
109 import zlib
110
110
111 zlib.compressobj # silence unused import warning
111 zlib.compressobj # silence unused import warning
112 except ImportError:
112 except ImportError:
113 raise SystemExit(
113 raise SystemExit(
114 "Couldn't import standard zlib (incomplete Python install)."
114 "Couldn't import standard zlib (incomplete Python install)."
115 )
115 )
116
116
117 # The base IronPython distribution (as of 2.7.1) doesn't support bz2
117 # The base IronPython distribution (as of 2.7.1) doesn't support bz2
118 isironpython = False
118 isironpython = False
119 try:
119 try:
120 isironpython = (
120 isironpython = (
121 platform.python_implementation().lower().find("ironpython") != -1
121 platform.python_implementation().lower().find("ironpython") != -1
122 )
122 )
123 except AttributeError:
123 except AttributeError:
124 pass
124 pass
125
125
126 if isironpython:
126 if isironpython:
127 sys.stderr.write("warning: IronPython detected (no bz2 support)\n")
127 sys.stderr.write("warning: IronPython detected (no bz2 support)\n")
128 else:
128 else:
129 try:
129 try:
130 import bz2
130 import bz2
131
131
132 bz2.BZ2Compressor # silence unused import warning
132 bz2.BZ2Compressor # silence unused import warning
133 except ImportError:
133 except ImportError:
134 raise SystemExit(
134 raise SystemExit(
135 "Couldn't import standard bz2 (incomplete Python install)."
135 "Couldn't import standard bz2 (incomplete Python install)."
136 )
136 )
137
137
138 ispypy = "PyPy" in sys.version
138 ispypy = "PyPy" in sys.version
139
139
140 hgrustext = os.environ.get('HGWITHRUSTEXT')
140 hgrustext = os.environ.get('HGWITHRUSTEXT')
141 # TODO record it for proper rebuild upon changes
141 # TODO record it for proper rebuild upon changes
142 # (see mercurial/__modulepolicy__.py)
142 # (see mercurial/__modulepolicy__.py)
143 if hgrustext != 'cpython' and hgrustext is not None:
143 if hgrustext != 'cpython' and hgrustext is not None:
144 hgrustext = 'direct-ffi'
144 hgrustext = 'direct-ffi'
145
145
146 import ctypes
146 import ctypes
147 import errno
147 import errno
148 import stat, subprocess, time
148 import stat, subprocess, time
149 import re
149 import re
150 import shutil
150 import shutil
151 import tempfile
151 import tempfile
152 from distutils import log
152 from distutils import log
153
153
154 # We have issues with setuptools on some platforms and builders. Until
154 # We have issues with setuptools on some platforms and builders. Until
155 # those are resolved, setuptools is opt-in except for platforms where
155 # those are resolved, setuptools is opt-in except for platforms where
156 # we don't have issues.
156 # we don't have issues.
157 issetuptools = os.name == 'nt' or 'FORCE_SETUPTOOLS' in os.environ
157 issetuptools = os.name == 'nt' or 'FORCE_SETUPTOOLS' in os.environ
158 if issetuptools:
158 if issetuptools:
159 from setuptools import setup
159 from setuptools import setup
160 else:
160 else:
161 from distutils.core import setup
161 from distutils.core import setup
162 from distutils.ccompiler import new_compiler
162 from distutils.ccompiler import new_compiler
163 from distutils.core import Command, Extension
163 from distutils.core import Command, Extension
164 from distutils.dist import Distribution
164 from distutils.dist import Distribution
165 from distutils.command.build import build
165 from distutils.command.build import build
166 from distutils.command.build_ext import build_ext
166 from distutils.command.build_ext import build_ext
167 from distutils.command.build_py import build_py
167 from distutils.command.build_py import build_py
168 from distutils.command.build_scripts import build_scripts
168 from distutils.command.build_scripts import build_scripts
169 from distutils.command.install import install
169 from distutils.command.install import install
170 from distutils.command.install_lib import install_lib
170 from distutils.command.install_lib import install_lib
171 from distutils.command.install_scripts import install_scripts
171 from distutils.command.install_scripts import install_scripts
172 from distutils.spawn import spawn, find_executable
172 from distutils.spawn import spawn, find_executable
173 from distutils import file_util
173 from distutils import file_util
174 from distutils.errors import (
174 from distutils.errors import (
175 CCompilerError,
175 CCompilerError,
176 DistutilsError,
176 DistutilsError,
177 DistutilsExecError,
177 DistutilsExecError,
178 )
178 )
179 from distutils.sysconfig import get_python_inc, get_config_var
179 from distutils.sysconfig import get_python_inc, get_config_var
180 from distutils.version import StrictVersion
180 from distutils.version import StrictVersion
181
181
182 # Explain to distutils.StrictVersion how our release candidates are versionned
182 # Explain to distutils.StrictVersion how our release candidates are versionned
183 StrictVersion.version_re = re.compile(r'^(\d+)\.(\d+)(\.(\d+))?-?(rc(\d+))?$')
183 StrictVersion.version_re = re.compile(r'^(\d+)\.(\d+)(\.(\d+))?-?(rc(\d+))?$')
184
184
185
185
186 def write_if_changed(path, content):
186 def write_if_changed(path, content):
187 """Write content to a file iff the content hasn't changed."""
187 """Write content to a file iff the content hasn't changed."""
188 if os.path.exists(path):
188 if os.path.exists(path):
189 with open(path, 'rb') as fh:
189 with open(path, 'rb') as fh:
190 current = fh.read()
190 current = fh.read()
191 else:
191 else:
192 current = b''
192 current = b''
193
193
194 if current != content:
194 if current != content:
195 with open(path, 'wb') as fh:
195 with open(path, 'wb') as fh:
196 fh.write(content)
196 fh.write(content)
197
197
198
198
199 scripts = ['hg']
199 scripts = ['hg']
200 if os.name == 'nt':
200 if os.name == 'nt':
201 # We remove hg.bat if we are able to build hg.exe.
201 # We remove hg.bat if we are able to build hg.exe.
202 scripts.append('contrib/win32/hg.bat')
202 scripts.append('contrib/win32/hg.bat')
203
203
204
204
205 def cancompile(cc, code):
205 def cancompile(cc, code):
206 tmpdir = tempfile.mkdtemp(prefix='hg-install-')
206 tmpdir = tempfile.mkdtemp(prefix='hg-install-')
207 devnull = oldstderr = None
207 devnull = oldstderr = None
208 try:
208 try:
209 fname = os.path.join(tmpdir, 'testcomp.c')
209 fname = os.path.join(tmpdir, 'testcomp.c')
210 f = open(fname, 'w')
210 f = open(fname, 'w')
211 f.write(code)
211 f.write(code)
212 f.close()
212 f.close()
213 # Redirect stderr to /dev/null to hide any error messages
213 # Redirect stderr to /dev/null to hide any error messages
214 # from the compiler.
214 # from the compiler.
215 # This will have to be changed if we ever have to check
215 # This will have to be changed if we ever have to check
216 # for a function on Windows.
216 # for a function on Windows.
217 devnull = open('/dev/null', 'w')
217 devnull = open('/dev/null', 'w')
218 oldstderr = os.dup(sys.stderr.fileno())
218 oldstderr = os.dup(sys.stderr.fileno())
219 os.dup2(devnull.fileno(), sys.stderr.fileno())
219 os.dup2(devnull.fileno(), sys.stderr.fileno())
220 objects = cc.compile([fname], output_dir=tmpdir)
220 objects = cc.compile([fname], output_dir=tmpdir)
221 cc.link_executable(objects, os.path.join(tmpdir, "a.out"))
221 cc.link_executable(objects, os.path.join(tmpdir, "a.out"))
222 return True
222 return True
223 except Exception:
223 except Exception:
224 return False
224 return False
225 finally:
225 finally:
226 if oldstderr is not None:
226 if oldstderr is not None:
227 os.dup2(oldstderr, sys.stderr.fileno())
227 os.dup2(oldstderr, sys.stderr.fileno())
228 if devnull is not None:
228 if devnull is not None:
229 devnull.close()
229 devnull.close()
230 shutil.rmtree(tmpdir)
230 shutil.rmtree(tmpdir)
231
231
232
232
233 # simplified version of distutils.ccompiler.CCompiler.has_function
233 # simplified version of distutils.ccompiler.CCompiler.has_function
234 # that actually removes its temporary files.
234 # that actually removes its temporary files.
235 def hasfunction(cc, funcname):
235 def hasfunction(cc, funcname):
236 code = 'int main(void) { %s(); }\n' % funcname
236 code = 'int main(void) { %s(); }\n' % funcname
237 return cancompile(cc, code)
237 return cancompile(cc, code)
238
238
239
239
240 def hasheader(cc, headername):
240 def hasheader(cc, headername):
241 code = '#include <%s>\nint main(void) { return 0; }\n' % headername
241 code = '#include <%s>\nint main(void) { return 0; }\n' % headername
242 return cancompile(cc, code)
242 return cancompile(cc, code)
243
243
244
244
245 # py2exe needs to be installed to work
245 # py2exe needs to be installed to work
246 try:
246 try:
247 import py2exe
247 import py2exe
248
248
249 py2exe.Distribution # silence unused import warning
249 py2exe.Distribution # silence unused import warning
250 py2exeloaded = True
250 py2exeloaded = True
251 # import py2exe's patched Distribution class
251 # import py2exe's patched Distribution class
252 from distutils.core import Distribution
252 from distutils.core import Distribution
253 except ImportError:
253 except ImportError:
254 py2exeloaded = False
254 py2exeloaded = False
255
255
256
256
257 def runcmd(cmd, env, cwd=None):
257 def runcmd(cmd, env, cwd=None):
258 p = subprocess.Popen(
258 p = subprocess.Popen(
259 cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env, cwd=cwd
259 cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env, cwd=cwd
260 )
260 )
261 out, err = p.communicate()
261 out, err = p.communicate()
262 return p.returncode, out, err
262 return p.returncode, out, err
263
263
264
264
265 class hgcommand(object):
265 class hgcommand(object):
266 def __init__(self, cmd, env):
266 def __init__(self, cmd, env):
267 self.cmd = cmd
267 self.cmd = cmd
268 self.env = env
268 self.env = env
269
269
270 def run(self, args):
270 def run(self, args):
271 cmd = self.cmd + args
271 cmd = self.cmd + args
272 returncode, out, err = runcmd(cmd, self.env)
272 returncode, out, err = runcmd(cmd, self.env)
273 err = filterhgerr(err)
273 err = filterhgerr(err)
274 if err or returncode != 0:
274 if err or returncode != 0:
275 printf("stderr from '%s':" % (' '.join(cmd)), file=sys.stderr)
275 printf("stderr from '%s':" % (' '.join(cmd)), file=sys.stderr)
276 printf(err, file=sys.stderr)
276 printf(err, file=sys.stderr)
277 return ''
277 return ''
278 return out
278 return out
279
279
280
280
281 def filterhgerr(err):
281 def filterhgerr(err):
282 # If root is executing setup.py, but the repository is owned by
282 # If root is executing setup.py, but the repository is owned by
283 # another user (as in "sudo python setup.py install") we will get
283 # another user (as in "sudo python setup.py install") we will get
284 # trust warnings since the .hg/hgrc file is untrusted. That is
284 # trust warnings since the .hg/hgrc file is untrusted. That is
285 # fine, we don't want to load it anyway. Python may warn about
285 # fine, we don't want to load it anyway. Python may warn about
286 # a missing __init__.py in mercurial/locale, we also ignore that.
286 # a missing __init__.py in mercurial/locale, we also ignore that.
287 err = [
287 err = [
288 e
288 e
289 for e in err.splitlines()
289 for e in err.splitlines()
290 if (
290 if (
291 not e.startswith(b'not trusting file')
291 not e.startswith(b'not trusting file')
292 and not e.startswith(b'warning: Not importing')
292 and not e.startswith(b'warning: Not importing')
293 and not e.startswith(b'obsolete feature not enabled')
293 and not e.startswith(b'obsolete feature not enabled')
294 and not e.startswith(b'*** failed to import extension')
294 and not e.startswith(b'*** failed to import extension')
295 and not e.startswith(b'devel-warn:')
295 and not e.startswith(b'devel-warn:')
296 and not (
296 and not (
297 e.startswith(b'(third party extension')
297 e.startswith(b'(third party extension')
298 and e.endswith(b'or newer of Mercurial; disabling)')
298 and e.endswith(b'or newer of Mercurial; disabling)')
299 )
299 )
300 )
300 )
301 ]
301 ]
302 return b'\n'.join(b' ' + e for e in err)
302 return b'\n'.join(b' ' + e for e in err)
303
303
304
304
305 def findhg():
305 def findhg():
306 """Try to figure out how we should invoke hg for examining the local
306 """Try to figure out how we should invoke hg for examining the local
307 repository contents.
307 repository contents.
308
308
309 Returns an hgcommand object."""
309 Returns an hgcommand object."""
310 # By default, prefer the "hg" command in the user's path. This was
310 # By default, prefer the "hg" command in the user's path. This was
311 # presumably the hg command that the user used to create this repository.
311 # presumably the hg command that the user used to create this repository.
312 #
312 #
313 # This repository may require extensions or other settings that would not
313 # This repository may require extensions or other settings that would not
314 # be enabled by running the hg script directly from this local repository.
314 # be enabled by running the hg script directly from this local repository.
315 hgenv = os.environ.copy()
315 hgenv = os.environ.copy()
316 # Use HGPLAIN to disable hgrc settings that would change output formatting,
316 # Use HGPLAIN to disable hgrc settings that would change output formatting,
317 # and disable localization for the same reasons.
317 # and disable localization for the same reasons.
318 hgenv['HGPLAIN'] = '1'
318 hgenv['HGPLAIN'] = '1'
319 hgenv['LANGUAGE'] = 'C'
319 hgenv['LANGUAGE'] = 'C'
320 hgcmd = ['hg']
320 hgcmd = ['hg']
321 # Run a simple "hg log" command just to see if using hg from the user's
321 # Run a simple "hg log" command just to see if using hg from the user's
322 # path works and can successfully interact with this repository. Windows
322 # path works and can successfully interact with this repository. Windows
323 # gives precedence to hg.exe in the current directory, so fall back to the
323 # gives precedence to hg.exe in the current directory, so fall back to the
324 # python invocation of local hg, where pythonXY.dll can always be found.
324 # python invocation of local hg, where pythonXY.dll can always be found.
325 check_cmd = ['log', '-r.', '-Ttest']
325 check_cmd = ['log', '-r.', '-Ttest']
326 if os.name != 'nt':
326 if os.name != 'nt':
327 try:
327 try:
328 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
328 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
329 except EnvironmentError:
329 except EnvironmentError:
330 retcode = -1
330 retcode = -1
331 if retcode == 0 and not filterhgerr(err):
331 if retcode == 0 and not filterhgerr(err):
332 return hgcommand(hgcmd, hgenv)
332 return hgcommand(hgcmd, hgenv)
333
333
334 # Fall back to trying the local hg installation.
334 # Fall back to trying the local hg installation.
335 hgenv = localhgenv()
335 hgenv = localhgenv()
336 hgcmd = [sys.executable, 'hg']
336 hgcmd = [sys.executable, 'hg']
337 try:
337 try:
338 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
338 retcode, out, err = runcmd(hgcmd + check_cmd, hgenv)
339 except EnvironmentError:
339 except EnvironmentError:
340 retcode = -1
340 retcode = -1
341 if retcode == 0 and not filterhgerr(err):
341 if retcode == 0 and not filterhgerr(err):
342 return hgcommand(hgcmd, hgenv)
342 return hgcommand(hgcmd, hgenv)
343
343
344 raise SystemExit(
344 raise SystemExit(
345 'Unable to find a working hg binary to extract the '
345 'Unable to find a working hg binary to extract the '
346 'version from the repository tags'
346 'version from the repository tags'
347 )
347 )
348
348
349
349
350 def localhgenv():
350 def localhgenv():
351 """Get an environment dictionary to use for invoking or importing
351 """Get an environment dictionary to use for invoking or importing
352 mercurial from the local repository."""
352 mercurial from the local repository."""
353 # Execute hg out of this directory with a custom environment which takes
353 # Execute hg out of this directory with a custom environment which takes
354 # care to not use any hgrc files and do no localization.
354 # care to not use any hgrc files and do no localization.
355 env = {
355 env = {
356 'HGMODULEPOLICY': 'py',
356 'HGMODULEPOLICY': 'py',
357 'HGRCPATH': '',
357 'HGRCPATH': '',
358 'LANGUAGE': 'C',
358 'LANGUAGE': 'C',
359 'PATH': '',
359 'PATH': '',
360 } # make pypi modules that use os.environ['PATH'] happy
360 } # make pypi modules that use os.environ['PATH'] happy
361 if 'LD_LIBRARY_PATH' in os.environ:
361 if 'LD_LIBRARY_PATH' in os.environ:
362 env['LD_LIBRARY_PATH'] = os.environ['LD_LIBRARY_PATH']
362 env['LD_LIBRARY_PATH'] = os.environ['LD_LIBRARY_PATH']
363 if 'SystemRoot' in os.environ:
363 if 'SystemRoot' in os.environ:
364 # SystemRoot is required by Windows to load various DLLs. See:
364 # SystemRoot is required by Windows to load various DLLs. See:
365 # https://bugs.python.org/issue13524#msg148850
365 # https://bugs.python.org/issue13524#msg148850
366 env['SystemRoot'] = os.environ['SystemRoot']
366 env['SystemRoot'] = os.environ['SystemRoot']
367 return env
367 return env
368
368
369
369
370 version = ''
370 version = ''
371
371
372 if os.path.isdir('.hg'):
372 if os.path.isdir('.hg'):
373 hg = findhg()
373 hg = findhg()
374 cmd = ['log', '-r', '.', '--template', '{tags}\n']
374 cmd = ['log', '-r', '.', '--template', '{tags}\n']
375 numerictags = [t for t in sysstr(hg.run(cmd)).split() if t[0:1].isdigit()]
375 numerictags = [t for t in sysstr(hg.run(cmd)).split() if t[0:1].isdigit()]
376 hgid = sysstr(hg.run(['id', '-i'])).strip()
376 hgid = sysstr(hg.run(['id', '-i'])).strip()
377 if not hgid:
377 if not hgid:
378 # Bail out if hg is having problems interacting with this repository,
378 # Bail out if hg is having problems interacting with this repository,
379 # rather than falling through and producing a bogus version number.
379 # rather than falling through and producing a bogus version number.
380 # Continuing with an invalid version number will break extensions
380 # Continuing with an invalid version number will break extensions
381 # that define minimumhgversion.
381 # that define minimumhgversion.
382 raise SystemExit('Unable to determine hg version from local repository')
382 raise SystemExit('Unable to determine hg version from local repository')
383 if numerictags: # tag(s) found
383 if numerictags: # tag(s) found
384 version = numerictags[-1]
384 version = numerictags[-1]
385 if hgid.endswith('+'): # propagate the dirty status to the tag
385 if hgid.endswith('+'): # propagate the dirty status to the tag
386 version += '+'
386 version += '+'
387 else: # no tag found
387 else: # no tag found
388 ltagcmd = ['parents', '--template', '{latesttag}']
388 ltagcmd = ['parents', '--template', '{latesttag}']
389 ltag = sysstr(hg.run(ltagcmd))
389 ltag = sysstr(hg.run(ltagcmd))
390 changessincecmd = ['log', '-T', 'x\n', '-r', "only(.,'%s')" % ltag]
390 changessincecmd = ['log', '-T', 'x\n', '-r', "only(.,'%s')" % ltag]
391 changessince = len(hg.run(changessincecmd).splitlines())
391 changessince = len(hg.run(changessincecmd).splitlines())
392 version = '%s+%s-%s' % (ltag, changessince, hgid)
392 version = '%s+%s-%s' % (ltag, changessince, hgid)
393 if version.endswith('+'):
393 if version.endswith('+'):
394 version += time.strftime('%Y%m%d')
394 version += time.strftime('%Y%m%d')
395 elif os.path.exists('.hg_archival.txt'):
395 elif os.path.exists('.hg_archival.txt'):
396 kw = dict(
396 kw = dict(
397 [[t.strip() for t in l.split(':', 1)] for l in open('.hg_archival.txt')]
397 [[t.strip() for t in l.split(':', 1)] for l in open('.hg_archival.txt')]
398 )
398 )
399 if 'tag' in kw:
399 if 'tag' in kw:
400 version = kw['tag']
400 version = kw['tag']
401 elif 'latesttag' in kw:
401 elif 'latesttag' in kw:
402 if 'changessincelatesttag' in kw:
402 if 'changessincelatesttag' in kw:
403 version = '%(latesttag)s+%(changessincelatesttag)s-%(node).12s' % kw
403 version = '%(latesttag)s+%(changessincelatesttag)s-%(node).12s' % kw
404 else:
404 else:
405 version = '%(latesttag)s+%(latesttagdistance)s-%(node).12s' % kw
405 version = '%(latesttag)s+%(latesttagdistance)s-%(node).12s' % kw
406 else:
406 else:
407 version = kw.get('node', '')[:12]
407 version = kw.get('node', '')[:12]
408
408
409 if version:
409 if version:
410 versionb = version
410 versionb = version
411 if not isinstance(versionb, bytes):
411 if not isinstance(versionb, bytes):
412 versionb = versionb.encode('ascii')
412 versionb = versionb.encode('ascii')
413
413
414 write_if_changed(
414 write_if_changed(
415 'mercurial/__version__.py',
415 'mercurial/__version__.py',
416 b''.join(
416 b''.join(
417 [
417 [
418 b'# this file is autogenerated by setup.py\n'
418 b'# this file is autogenerated by setup.py\n'
419 b'version = b"%s"\n' % versionb,
419 b'version = b"%s"\n' % versionb,
420 ]
420 ]
421 ),
421 ),
422 )
422 )
423
423
424 try:
424 try:
425 oldpolicy = os.environ.get('HGMODULEPOLICY', None)
425 oldpolicy = os.environ.get('HGMODULEPOLICY', None)
426 os.environ['HGMODULEPOLICY'] = 'py'
426 os.environ['HGMODULEPOLICY'] = 'py'
427 from mercurial import __version__
427 from mercurial import __version__
428
428
429 version = __version__.version
429 version = __version__.version
430 except ImportError:
430 except ImportError:
431 version = b'unknown'
431 version = b'unknown'
432 finally:
432 finally:
433 if oldpolicy is None:
433 if oldpolicy is None:
434 del os.environ['HGMODULEPOLICY']
434 del os.environ['HGMODULEPOLICY']
435 else:
435 else:
436 os.environ['HGMODULEPOLICY'] = oldpolicy
436 os.environ['HGMODULEPOLICY'] = oldpolicy
437
437
438
438
439 class hgbuild(build):
439 class hgbuild(build):
440 # Insert hgbuildmo first so that files in mercurial/locale/ are found
440 # Insert hgbuildmo first so that files in mercurial/locale/ are found
441 # when build_py is run next.
441 # when build_py is run next.
442 sub_commands = [('build_mo', None)] + build.sub_commands
442 sub_commands = [('build_mo', None)] + build.sub_commands
443
443
444
444
445 class hgbuildmo(build):
445 class hgbuildmo(build):
446
446
447 description = "build translations (.mo files)"
447 description = "build translations (.mo files)"
448
448
449 def run(self):
449 def run(self):
450 if not find_executable('msgfmt'):
450 if not find_executable('msgfmt'):
451 self.warn(
451 self.warn(
452 "could not find msgfmt executable, no translations "
452 "could not find msgfmt executable, no translations "
453 "will be built"
453 "will be built"
454 )
454 )
455 return
455 return
456
456
457 podir = 'i18n'
457 podir = 'i18n'
458 if not os.path.isdir(podir):
458 if not os.path.isdir(podir):
459 self.warn("could not find %s/ directory" % podir)
459 self.warn("could not find %s/ directory" % podir)
460 return
460 return
461
461
462 join = os.path.join
462 join = os.path.join
463 for po in os.listdir(podir):
463 for po in os.listdir(podir):
464 if not po.endswith('.po'):
464 if not po.endswith('.po'):
465 continue
465 continue
466 pofile = join(podir, po)
466 pofile = join(podir, po)
467 modir = join('locale', po[:-3], 'LC_MESSAGES')
467 modir = join('locale', po[:-3], 'LC_MESSAGES')
468 mofile = join(modir, 'hg.mo')
468 mofile = join(modir, 'hg.mo')
469 mobuildfile = join('mercurial', mofile)
469 mobuildfile = join('mercurial', mofile)
470 cmd = ['msgfmt', '-v', '-o', mobuildfile, pofile]
470 cmd = ['msgfmt', '-v', '-o', mobuildfile, pofile]
471 if sys.platform != 'sunos5':
471 if sys.platform != 'sunos5':
472 # msgfmt on Solaris does not know about -c
472 # msgfmt on Solaris does not know about -c
473 cmd.append('-c')
473 cmd.append('-c')
474 self.mkpath(join('mercurial', modir))
474 self.mkpath(join('mercurial', modir))
475 self.make_file([pofile], mobuildfile, spawn, (cmd,))
475 self.make_file([pofile], mobuildfile, spawn, (cmd,))
476
476
477
477
478 class hgdist(Distribution):
478 class hgdist(Distribution):
479 pure = False
479 pure = False
480 rust = hgrustext is not None
480 rust = hgrustext is not None
481 cffi = ispypy
481 cffi = ispypy
482
482
483 global_options = Distribution.global_options + [
483 global_options = Distribution.global_options + [
484 ('pure', None, "use pure (slow) Python code instead of C extensions"),
484 ('pure', None, "use pure (slow) Python code instead of C extensions"),
485 ('rust', None, "use Rust extensions additionally to C extensions"),
485 ('rust', None, "use Rust extensions additionally to C extensions"),
486 ]
486 ]
487
487
488 def has_ext_modules(self):
488 def has_ext_modules(self):
489 # self.ext_modules is emptied in hgbuildpy.finalize_options which is
489 # self.ext_modules is emptied in hgbuildpy.finalize_options which is
490 # too late for some cases
490 # too late for some cases
491 return not self.pure and Distribution.has_ext_modules(self)
491 return not self.pure and Distribution.has_ext_modules(self)
492
492
493
493
494 # This is ugly as a one-liner. So use a variable.
494 # This is ugly as a one-liner. So use a variable.
495 buildextnegops = dict(getattr(build_ext, 'negative_options', {}))
495 buildextnegops = dict(getattr(build_ext, 'negative_options', {}))
496 buildextnegops['no-zstd'] = 'zstd'
496 buildextnegops['no-zstd'] = 'zstd'
497 buildextnegops['no-rust'] = 'rust'
497 buildextnegops['no-rust'] = 'rust'
498
498
499
499
500 class hgbuildext(build_ext):
500 class hgbuildext(build_ext):
501 user_options = build_ext.user_options + [
501 user_options = build_ext.user_options + [
502 ('zstd', None, 'compile zstd bindings [default]'),
502 ('zstd', None, 'compile zstd bindings [default]'),
503 ('no-zstd', None, 'do not compile zstd bindings'),
503 ('no-zstd', None, 'do not compile zstd bindings'),
504 (
504 (
505 'rust',
505 'rust',
506 None,
506 None,
507 'compile Rust extensions if they are in use '
507 'compile Rust extensions if they are in use '
508 '(requires Cargo) [default]',
508 '(requires Cargo) [default]',
509 ),
509 ),
510 ('no-rust', None, 'do not compile Rust extensions'),
510 ('no-rust', None, 'do not compile Rust extensions'),
511 ]
511 ]
512
512
513 boolean_options = build_ext.boolean_options + ['zstd', 'rust']
513 boolean_options = build_ext.boolean_options + ['zstd', 'rust']
514 negative_opt = buildextnegops
514 negative_opt = buildextnegops
515
515
516 def initialize_options(self):
516 def initialize_options(self):
517 self.zstd = True
517 self.zstd = True
518 self.rust = True
518 self.rust = True
519
519
520 return build_ext.initialize_options(self)
520 return build_ext.initialize_options(self)
521
521
522 def finalize_options(self):
522 def finalize_options(self):
523 # Unless overridden by the end user, build extensions in parallel.
523 # Unless overridden by the end user, build extensions in parallel.
524 # Only influences behavior on Python 3.5+.
524 # Only influences behavior on Python 3.5+.
525 if getattr(self, 'parallel', None) is None:
525 if getattr(self, 'parallel', None) is None:
526 self.parallel = True
526 self.parallel = True
527
527
528 return build_ext.finalize_options(self)
528 return build_ext.finalize_options(self)
529
529
530 def build_extensions(self):
530 def build_extensions(self):
531 ruststandalones = [
531 ruststandalones = [
532 e for e in self.extensions if isinstance(e, RustStandaloneExtension)
532 e for e in self.extensions if isinstance(e, RustStandaloneExtension)
533 ]
533 ]
534 self.extensions = [
534 self.extensions = [
535 e for e in self.extensions if e not in ruststandalones
535 e for e in self.extensions if e not in ruststandalones
536 ]
536 ]
537 # Filter out zstd if disabled via argument.
537 # Filter out zstd if disabled via argument.
538 if not self.zstd:
538 if not self.zstd:
539 self.extensions = [
539 self.extensions = [
540 e for e in self.extensions if e.name != 'mercurial.zstd'
540 e for e in self.extensions if e.name != 'mercurial.zstd'
541 ]
541 ]
542
542
543 # Build Rust standalon extensions if it'll be used
543 # Build Rust standalon extensions if it'll be used
544 # and its build is not explictely disabled (for external build
544 # and its build is not explictely disabled (for external build
545 # as Linux distributions would do)
545 # as Linux distributions would do)
546 if self.distribution.rust and self.rust and hgrustext != 'direct-ffi':
546 if self.distribution.rust and self.rust and hgrustext != 'direct-ffi':
547 for rustext in ruststandalones:
547 for rustext in ruststandalones:
548 rustext.build('' if self.inplace else self.build_lib)
548 rustext.build('' if self.inplace else self.build_lib)
549
549
550 return build_ext.build_extensions(self)
550 return build_ext.build_extensions(self)
551
551
552 def build_extension(self, ext):
552 def build_extension(self, ext):
553 if (
553 if (
554 self.distribution.rust
554 self.distribution.rust
555 and self.rust
555 and self.rust
556 and isinstance(ext, RustExtension)
556 and isinstance(ext, RustExtension)
557 ):
557 ):
558 ext.rustbuild()
558 ext.rustbuild()
559 try:
559 try:
560 build_ext.build_extension(self, ext)
560 build_ext.build_extension(self, ext)
561 except CCompilerError:
561 except CCompilerError:
562 if not getattr(ext, 'optional', False):
562 if not getattr(ext, 'optional', False):
563 raise
563 raise
564 log.warn(
564 log.warn(
565 "Failed to build optional extension '%s' (skipping)", ext.name
565 "Failed to build optional extension '%s' (skipping)", ext.name
566 )
566 )
567
567
568
568
569 class hgbuildscripts(build_scripts):
569 class hgbuildscripts(build_scripts):
570 def run(self):
570 def run(self):
571 if os.name != 'nt' or self.distribution.pure:
571 if os.name != 'nt' or self.distribution.pure:
572 return build_scripts.run(self)
572 return build_scripts.run(self)
573
573
574 exebuilt = False
574 exebuilt = False
575 try:
575 try:
576 self.run_command('build_hgexe')
576 self.run_command('build_hgexe')
577 exebuilt = True
577 exebuilt = True
578 except (DistutilsError, CCompilerError):
578 except (DistutilsError, CCompilerError):
579 log.warn('failed to build optional hg.exe')
579 log.warn('failed to build optional hg.exe')
580
580
581 if exebuilt:
581 if exebuilt:
582 # Copying hg.exe to the scripts build directory ensures it is
582 # Copying hg.exe to the scripts build directory ensures it is
583 # installed by the install_scripts command.
583 # installed by the install_scripts command.
584 hgexecommand = self.get_finalized_command('build_hgexe')
584 hgexecommand = self.get_finalized_command('build_hgexe')
585 dest = os.path.join(self.build_dir, 'hg.exe')
585 dest = os.path.join(self.build_dir, 'hg.exe')
586 self.mkpath(self.build_dir)
586 self.mkpath(self.build_dir)
587 self.copy_file(hgexecommand.hgexepath, dest)
587 self.copy_file(hgexecommand.hgexepath, dest)
588
588
589 # Remove hg.bat because it is redundant with hg.exe.
589 # Remove hg.bat because it is redundant with hg.exe.
590 self.scripts.remove('contrib/win32/hg.bat')
590 self.scripts.remove('contrib/win32/hg.bat')
591
591
592 return build_scripts.run(self)
592 return build_scripts.run(self)
593
593
594
594
595 class hgbuildpy(build_py):
595 class hgbuildpy(build_py):
596 def finalize_options(self):
596 def finalize_options(self):
597 build_py.finalize_options(self)
597 build_py.finalize_options(self)
598
598
599 if self.distribution.pure:
599 if self.distribution.pure:
600 self.distribution.ext_modules = []
600 self.distribution.ext_modules = []
601 elif self.distribution.cffi:
601 elif self.distribution.cffi:
602 from mercurial.cffi import (
602 from mercurial.cffi import (
603 bdiffbuild,
603 bdiffbuild,
604 mpatchbuild,
604 mpatchbuild,
605 )
605 )
606
606
607 exts = [
607 exts = [
608 mpatchbuild.ffi.distutils_extension(),
608 mpatchbuild.ffi.distutils_extension(),
609 bdiffbuild.ffi.distutils_extension(),
609 bdiffbuild.ffi.distutils_extension(),
610 ]
610 ]
611 # cffi modules go here
611 # cffi modules go here
612 if sys.platform == 'darwin':
612 if sys.platform == 'darwin':
613 from mercurial.cffi import osutilbuild
613 from mercurial.cffi import osutilbuild
614
614
615 exts.append(osutilbuild.ffi.distutils_extension())
615 exts.append(osutilbuild.ffi.distutils_extension())
616 self.distribution.ext_modules = exts
616 self.distribution.ext_modules = exts
617 else:
617 else:
618 h = os.path.join(get_python_inc(), 'Python.h')
618 h = os.path.join(get_python_inc(), 'Python.h')
619 if not os.path.exists(h):
619 if not os.path.exists(h):
620 raise SystemExit(
620 raise SystemExit(
621 'Python headers are required to build '
621 'Python headers are required to build '
622 'Mercurial but weren\'t found in %s' % h
622 'Mercurial but weren\'t found in %s' % h
623 )
623 )
624
624
625 def run(self):
625 def run(self):
626 basepath = os.path.join(self.build_lib, 'mercurial')
626 basepath = os.path.join(self.build_lib, 'mercurial')
627 self.mkpath(basepath)
627 self.mkpath(basepath)
628
628
629 rust = self.distribution.rust
629 rust = self.distribution.rust
630 if self.distribution.pure:
630 if self.distribution.pure:
631 modulepolicy = 'py'
631 modulepolicy = 'py'
632 elif self.build_lib == '.':
632 elif self.build_lib == '.':
633 # in-place build should run without rebuilding and Rust extensions
633 # in-place build should run without rebuilding and Rust extensions
634 modulepolicy = 'rust+c-allow' if rust else 'allow'
634 modulepolicy = 'rust+c-allow' if rust else 'allow'
635 else:
635 else:
636 modulepolicy = 'rust+c' if rust else 'c'
636 modulepolicy = 'rust+c' if rust else 'c'
637
637
638 content = b''.join(
638 content = b''.join(
639 [
639 [
640 b'# this file is autogenerated by setup.py\n',
640 b'# this file is autogenerated by setup.py\n',
641 b'modulepolicy = b"%s"\n' % modulepolicy.encode('ascii'),
641 b'modulepolicy = b"%s"\n' % modulepolicy.encode('ascii'),
642 ]
642 ]
643 )
643 )
644 write_if_changed(os.path.join(basepath, '__modulepolicy__.py'), content)
644 write_if_changed(os.path.join(basepath, '__modulepolicy__.py'), content)
645
645
646 build_py.run(self)
646 build_py.run(self)
647
647
648
648
649 class buildhgextindex(Command):
649 class buildhgextindex(Command):
650 description = 'generate prebuilt index of hgext (for frozen package)'
650 description = 'generate prebuilt index of hgext (for frozen package)'
651 user_options = []
651 user_options = []
652 _indexfilename = 'hgext/__index__.py'
652 _indexfilename = 'hgext/__index__.py'
653
653
654 def initialize_options(self):
654 def initialize_options(self):
655 pass
655 pass
656
656
657 def finalize_options(self):
657 def finalize_options(self):
658 pass
658 pass
659
659
660 def run(self):
660 def run(self):
661 if os.path.exists(self._indexfilename):
661 if os.path.exists(self._indexfilename):
662 with open(self._indexfilename, 'w') as f:
662 with open(self._indexfilename, 'w') as f:
663 f.write('# empty\n')
663 f.write('# empty\n')
664
664
665 # here no extension enabled, disabled() lists up everything
665 # here no extension enabled, disabled() lists up everything
666 code = (
666 code = (
667 'import pprint; from mercurial import extensions; '
667 'import pprint; from mercurial import extensions; '
668 'pprint.pprint(extensions.disabled())'
668 'pprint.pprint(extensions.disabled())'
669 )
669 )
670 returncode, out, err = runcmd(
670 returncode, out, err = runcmd(
671 [sys.executable, '-c', code], localhgenv()
671 [sys.executable, '-c', code], localhgenv()
672 )
672 )
673 if err or returncode != 0:
673 if err or returncode != 0:
674 raise DistutilsExecError(err)
674 raise DistutilsExecError(err)
675
675
676 with open(self._indexfilename, 'wb') as f:
676 with open(self._indexfilename, 'wb') as f:
677 f.write(b'# this file is autogenerated by setup.py\n')
677 f.write(b'# this file is autogenerated by setup.py\n')
678 f.write(b'docs = ')
678 f.write(b'docs = ')
679 f.write(out)
679 f.write(out)
680
680
681
681
682 class buildhgexe(build_ext):
682 class buildhgexe(build_ext):
683 description = 'compile hg.exe from mercurial/exewrapper.c'
683 description = 'compile hg.exe from mercurial/exewrapper.c'
684 user_options = build_ext.user_options + [
684 user_options = build_ext.user_options + [
685 (
685 (
686 'long-paths-support',
686 'long-paths-support',
687 None,
687 None,
688 'enable support for long paths on '
688 'enable support for long paths on '
689 'Windows (off by default and '
689 'Windows (off by default and '
690 'experimental)',
690 'experimental)',
691 ),
691 ),
692 ]
692 ]
693
693
694 LONG_PATHS_MANIFEST = """
694 LONG_PATHS_MANIFEST = """
695 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
695 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
696 <assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">
696 <assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">
697 <application>
697 <application>
698 <windowsSettings
698 <windowsSettings
699 xmlns:ws2="http://schemas.microsoft.com/SMI/2016/WindowsSettings">
699 xmlns:ws2="http://schemas.microsoft.com/SMI/2016/WindowsSettings">
700 <ws2:longPathAware>true</ws2:longPathAware>
700 <ws2:longPathAware>true</ws2:longPathAware>
701 </windowsSettings>
701 </windowsSettings>
702 </application>
702 </application>
703 </assembly>"""
703 </assembly>"""
704
704
705 def initialize_options(self):
705 def initialize_options(self):
706 build_ext.initialize_options(self)
706 build_ext.initialize_options(self)
707 self.long_paths_support = False
707 self.long_paths_support = False
708
708
709 def build_extensions(self):
709 def build_extensions(self):
710 if os.name != 'nt':
710 if os.name != 'nt':
711 return
711 return
712 if isinstance(self.compiler, HackedMingw32CCompiler):
712 if isinstance(self.compiler, HackedMingw32CCompiler):
713 self.compiler.compiler_so = self.compiler.compiler # no -mdll
713 self.compiler.compiler_so = self.compiler.compiler # no -mdll
714 self.compiler.dll_libraries = [] # no -lmsrvc90
714 self.compiler.dll_libraries = [] # no -lmsrvc90
715
715
716 pythonlib = None
716 pythonlib = None
717
717
718 if getattr(sys, 'dllhandle', None):
718 if getattr(sys, 'dllhandle', None):
719 # Different Python installs can have different Python library
719 # Different Python installs can have different Python library
720 # names. e.g. the official CPython distribution uses pythonXY.dll
720 # names. e.g. the official CPython distribution uses pythonXY.dll
721 # and MinGW uses libpythonX.Y.dll.
721 # and MinGW uses libpythonX.Y.dll.
722 _kernel32 = ctypes.windll.kernel32
722 _kernel32 = ctypes.windll.kernel32
723 _kernel32.GetModuleFileNameA.argtypes = [
723 _kernel32.GetModuleFileNameA.argtypes = [
724 ctypes.c_void_p,
724 ctypes.c_void_p,
725 ctypes.c_void_p,
725 ctypes.c_void_p,
726 ctypes.c_ulong,
726 ctypes.c_ulong,
727 ]
727 ]
728 _kernel32.GetModuleFileNameA.restype = ctypes.c_ulong
728 _kernel32.GetModuleFileNameA.restype = ctypes.c_ulong
729 size = 1000
729 size = 1000
730 buf = ctypes.create_string_buffer(size + 1)
730 buf = ctypes.create_string_buffer(size + 1)
731 filelen = _kernel32.GetModuleFileNameA(
731 filelen = _kernel32.GetModuleFileNameA(
732 sys.dllhandle, ctypes.byref(buf), size
732 sys.dllhandle, ctypes.byref(buf), size
733 )
733 )
734
734
735 if filelen > 0 and filelen != size:
735 if filelen > 0 and filelen != size:
736 dllbasename = os.path.basename(buf.value)
736 dllbasename = os.path.basename(buf.value)
737 if not dllbasename.lower().endswith(b'.dll'):
737 if not dllbasename.lower().endswith(b'.dll'):
738 raise SystemExit(
738 raise SystemExit(
739 'Python DLL does not end with .dll: %s' % dllbasename
739 'Python DLL does not end with .dll: %s' % dllbasename
740 )
740 )
741 pythonlib = dllbasename[:-4]
741 pythonlib = dllbasename[:-4]
742
742
743 if not pythonlib:
743 if not pythonlib:
744 log.warn(
744 log.warn(
745 'could not determine Python DLL filename; assuming pythonXY'
745 'could not determine Python DLL filename; assuming pythonXY'
746 )
746 )
747
747
748 hv = sys.hexversion
748 hv = sys.hexversion
749 pythonlib = b'python%d%d' % (hv >> 24, (hv >> 16) & 0xFF)
749 pythonlib = b'python%d%d' % (hv >> 24, (hv >> 16) & 0xFF)
750
750
751 log.info('using %s as Python library name' % pythonlib)
751 log.info('using %s as Python library name' % pythonlib)
752 with open('mercurial/hgpythonlib.h', 'wb') as f:
752 with open('mercurial/hgpythonlib.h', 'wb') as f:
753 f.write(b'/* this file is autogenerated by setup.py */\n')
753 f.write(b'/* this file is autogenerated by setup.py */\n')
754 f.write(b'#define HGPYTHONLIB "%s"\n' % pythonlib)
754 f.write(b'#define HGPYTHONLIB "%s"\n' % pythonlib)
755
755
756 macros = None
756 macros = None
757 if sys.version_info[0] >= 3:
757 if sys.version_info[0] >= 3:
758 macros = [('_UNICODE', None), ('UNICODE', None)]
758 macros = [('_UNICODE', None), ('UNICODE', None)]
759
759
760 objects = self.compiler.compile(
760 objects = self.compiler.compile(
761 ['mercurial/exewrapper.c'],
761 ['mercurial/exewrapper.c'],
762 output_dir=self.build_temp,
762 output_dir=self.build_temp,
763 macros=macros,
763 macros=macros,
764 )
764 )
765 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
765 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
766 self.hgtarget = os.path.join(dir, 'hg')
766 self.hgtarget = os.path.join(dir, 'hg')
767 self.compiler.link_executable(
767 self.compiler.link_executable(
768 objects, self.hgtarget, libraries=[], output_dir=self.build_temp
768 objects, self.hgtarget, libraries=[], output_dir=self.build_temp
769 )
769 )
770 if self.long_paths_support:
770 if self.long_paths_support:
771 self.addlongpathsmanifest()
771 self.addlongpathsmanifest()
772
772
773 def addlongpathsmanifest(self):
773 def addlongpathsmanifest(self):
774 r"""Add manifest pieces so that hg.exe understands long paths
774 r"""Add manifest pieces so that hg.exe understands long paths
775
775
776 This is an EXPERIMENTAL feature, use with care.
776 This is an EXPERIMENTAL feature, use with care.
777 To enable long paths support, one needs to do two things:
777 To enable long paths support, one needs to do two things:
778 - build Mercurial with --long-paths-support option
778 - build Mercurial with --long-paths-support option
779 - change HKLM\SYSTEM\CurrentControlSet\Control\FileSystem\
779 - change HKLM\SYSTEM\CurrentControlSet\Control\FileSystem\
780 LongPathsEnabled to have value 1.
780 LongPathsEnabled to have value 1.
781
781
782 Please ignore 'warning 81010002: Unrecognized Element "longPathAware"';
782 Please ignore 'warning 81010002: Unrecognized Element "longPathAware"';
783 it happens because Mercurial uses mt.exe circa 2008, which is not
783 it happens because Mercurial uses mt.exe circa 2008, which is not
784 yet aware of long paths support in the manifest (I think so at least).
784 yet aware of long paths support in the manifest (I think so at least).
785 This does not stop mt.exe from embedding/merging the XML properly.
785 This does not stop mt.exe from embedding/merging the XML properly.
786
786
787 Why resource #1 should be used for .exe manifests? I don't know and
787 Why resource #1 should be used for .exe manifests? I don't know and
788 wasn't able to find an explanation for mortals. But it seems to work.
788 wasn't able to find an explanation for mortals. But it seems to work.
789 """
789 """
790 exefname = self.compiler.executable_filename(self.hgtarget)
790 exefname = self.compiler.executable_filename(self.hgtarget)
791 fdauto, manfname = tempfile.mkstemp(suffix='.hg.exe.manifest')
791 fdauto, manfname = tempfile.mkstemp(suffix='.hg.exe.manifest')
792 os.close(fdauto)
792 os.close(fdauto)
793 with open(manfname, 'w') as f:
793 with open(manfname, 'w') as f:
794 f.write(self.LONG_PATHS_MANIFEST)
794 f.write(self.LONG_PATHS_MANIFEST)
795 log.info("long paths manifest is written to '%s'" % manfname)
795 log.info("long paths manifest is written to '%s'" % manfname)
796 inputresource = '-inputresource:%s;#1' % exefname
796 inputresource = '-inputresource:%s;#1' % exefname
797 outputresource = '-outputresource:%s;#1' % exefname
797 outputresource = '-outputresource:%s;#1' % exefname
798 log.info("running mt.exe to update hg.exe's manifest in-place")
798 log.info("running mt.exe to update hg.exe's manifest in-place")
799 # supplying both -manifest and -inputresource to mt.exe makes
799 # supplying both -manifest and -inputresource to mt.exe makes
800 # it merge the embedded and supplied manifests in the -outputresource
800 # it merge the embedded and supplied manifests in the -outputresource
801 self.spawn(
801 self.spawn(
802 [
802 [
803 'mt.exe',
803 'mt.exe',
804 '-nologo',
804 '-nologo',
805 '-manifest',
805 '-manifest',
806 manfname,
806 manfname,
807 inputresource,
807 inputresource,
808 outputresource,
808 outputresource,
809 ]
809 ]
810 )
810 )
811 log.info("done updating hg.exe's manifest")
811 log.info("done updating hg.exe's manifest")
812 os.remove(manfname)
812 os.remove(manfname)
813
813
814 @property
814 @property
815 def hgexepath(self):
815 def hgexepath(self):
816 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
816 dir = os.path.dirname(self.get_ext_fullpath('dummy'))
817 return os.path.join(self.build_temp, dir, 'hg.exe')
817 return os.path.join(self.build_temp, dir, 'hg.exe')
818
818
819
819
820 class hgbuilddoc(Command):
820 class hgbuilddoc(Command):
821 description = 'build documentation'
821 description = 'build documentation'
822 user_options = [
822 user_options = [
823 ('man', None, 'generate man pages'),
823 ('man', None, 'generate man pages'),
824 ('html', None, 'generate html pages'),
824 ('html', None, 'generate html pages'),
825 ]
825 ]
826
826
827 def initialize_options(self):
827 def initialize_options(self):
828 self.man = None
828 self.man = None
829 self.html = None
829 self.html = None
830
830
831 def finalize_options(self):
831 def finalize_options(self):
832 # If --man or --html are set, only generate what we're told to.
832 # If --man or --html are set, only generate what we're told to.
833 # Otherwise generate everything.
833 # Otherwise generate everything.
834 have_subset = self.man is not None or self.html is not None
834 have_subset = self.man is not None or self.html is not None
835
835
836 if have_subset:
836 if have_subset:
837 self.man = True if self.man else False
837 self.man = True if self.man else False
838 self.html = True if self.html else False
838 self.html = True if self.html else False
839 else:
839 else:
840 self.man = True
840 self.man = True
841 self.html = True
841 self.html = True
842
842
843 def run(self):
843 def run(self):
844 def normalizecrlf(p):
844 def normalizecrlf(p):
845 with open(p, 'rb') as fh:
845 with open(p, 'rb') as fh:
846 orig = fh.read()
846 orig = fh.read()
847
847
848 if b'\r\n' not in orig:
848 if b'\r\n' not in orig:
849 return
849 return
850
850
851 log.info('normalizing %s to LF line endings' % p)
851 log.info('normalizing %s to LF line endings' % p)
852 with open(p, 'wb') as fh:
852 with open(p, 'wb') as fh:
853 fh.write(orig.replace(b'\r\n', b'\n'))
853 fh.write(orig.replace(b'\r\n', b'\n'))
854
854
855 def gentxt(root):
855 def gentxt(root):
856 txt = 'doc/%s.txt' % root
856 txt = 'doc/%s.txt' % root
857 log.info('generating %s' % txt)
857 log.info('generating %s' % txt)
858 res, out, err = runcmd(
858 res, out, err = runcmd(
859 [sys.executable, 'gendoc.py', root], os.environ, cwd='doc'
859 [sys.executable, 'gendoc.py', root], os.environ, cwd='doc'
860 )
860 )
861 if res:
861 if res:
862 raise SystemExit(
862 raise SystemExit(
863 'error running gendoc.py: %s' % '\n'.join([out, err])
863 'error running gendoc.py: %s' % '\n'.join([out, err])
864 )
864 )
865
865
866 with open(txt, 'wb') as fh:
866 with open(txt, 'wb') as fh:
867 fh.write(out)
867 fh.write(out)
868
868
869 def gengendoc(root):
869 def gengendoc(root):
870 gendoc = 'doc/%s.gendoc.txt' % root
870 gendoc = 'doc/%s.gendoc.txt' % root
871
871
872 log.info('generating %s' % gendoc)
872 log.info('generating %s' % gendoc)
873 res, out, err = runcmd(
873 res, out, err = runcmd(
874 [sys.executable, 'gendoc.py', '%s.gendoc' % root],
874 [sys.executable, 'gendoc.py', '%s.gendoc' % root],
875 os.environ,
875 os.environ,
876 cwd='doc',
876 cwd='doc',
877 )
877 )
878 if res:
878 if res:
879 raise SystemExit(
879 raise SystemExit(
880 'error running gendoc: %s' % '\n'.join([out, err])
880 'error running gendoc: %s' % '\n'.join([out, err])
881 )
881 )
882
882
883 with open(gendoc, 'wb') as fh:
883 with open(gendoc, 'wb') as fh:
884 fh.write(out)
884 fh.write(out)
885
885
886 def genman(root):
886 def genman(root):
887 log.info('generating doc/%s' % root)
887 log.info('generating doc/%s' % root)
888 res, out, err = runcmd(
888 res, out, err = runcmd(
889 [
889 [
890 sys.executable,
890 sys.executable,
891 'runrst',
891 'runrst',
892 'hgmanpage',
892 'hgmanpage',
893 '--halt',
893 '--halt',
894 'warning',
894 'warning',
895 '--strip-elements-with-class',
895 '--strip-elements-with-class',
896 'htmlonly',
896 'htmlonly',
897 '%s.txt' % root,
897 '%s.txt' % root,
898 root,
898 root,
899 ],
899 ],
900 os.environ,
900 os.environ,
901 cwd='doc',
901 cwd='doc',
902 )
902 )
903 if res:
903 if res:
904 raise SystemExit(
904 raise SystemExit(
905 'error running runrst: %s' % '\n'.join([out, err])
905 'error running runrst: %s' % '\n'.join([out, err])
906 )
906 )
907
907
908 normalizecrlf('doc/%s' % root)
908 normalizecrlf('doc/%s' % root)
909
909
910 def genhtml(root):
910 def genhtml(root):
911 log.info('generating doc/%s.html' % root)
911 log.info('generating doc/%s.html' % root)
912 res, out, err = runcmd(
912 res, out, err = runcmd(
913 [
913 [
914 sys.executable,
914 sys.executable,
915 'runrst',
915 'runrst',
916 'html',
916 'html',
917 '--halt',
917 '--halt',
918 'warning',
918 'warning',
919 '--link-stylesheet',
919 '--link-stylesheet',
920 '--stylesheet-path',
920 '--stylesheet-path',
921 'style.css',
921 'style.css',
922 '%s.txt' % root,
922 '%s.txt' % root,
923 '%s.html' % root,
923 '%s.html' % root,
924 ],
924 ],
925 os.environ,
925 os.environ,
926 cwd='doc',
926 cwd='doc',
927 )
927 )
928 if res:
928 if res:
929 raise SystemExit(
929 raise SystemExit(
930 'error running runrst: %s' % '\n'.join([out, err])
930 'error running runrst: %s' % '\n'.join([out, err])
931 )
931 )
932
932
933 normalizecrlf('doc/%s.html' % root)
933 normalizecrlf('doc/%s.html' % root)
934
934
935 # This logic is duplicated in doc/Makefile.
935 # This logic is duplicated in doc/Makefile.
936 sources = set(
936 sources = set(
937 f
937 f
938 for f in os.listdir('mercurial/helptext')
938 for f in os.listdir('mercurial/helptext')
939 if re.search(r'[0-9]\.txt$', f)
939 if re.search(r'[0-9]\.txt$', f)
940 )
940 )
941
941
942 # common.txt is a one-off.
942 # common.txt is a one-off.
943 gentxt('common')
943 gentxt('common')
944
944
945 for source in sorted(sources):
945 for source in sorted(sources):
946 assert source[-4:] == '.txt'
946 assert source[-4:] == '.txt'
947 root = source[:-4]
947 root = source[:-4]
948
948
949 gentxt(root)
949 gentxt(root)
950 gengendoc(root)
950 gengendoc(root)
951
951
952 if self.man:
952 if self.man:
953 genman(root)
953 genman(root)
954 if self.html:
954 if self.html:
955 genhtml(root)
955 genhtml(root)
956
956
957
957
958 class hginstall(install):
958 class hginstall(install):
959
959
960 user_options = install.user_options + [
960 user_options = install.user_options + [
961 (
961 (
962 'old-and-unmanageable',
962 'old-and-unmanageable',
963 None,
963 None,
964 'noop, present for eggless setuptools compat',
964 'noop, present for eggless setuptools compat',
965 ),
965 ),
966 (
966 (
967 'single-version-externally-managed',
967 'single-version-externally-managed',
968 None,
968 None,
969 'noop, present for eggless setuptools compat',
969 'noop, present for eggless setuptools compat',
970 ),
970 ),
971 ]
971 ]
972
972
973 # Also helps setuptools not be sad while we refuse to create eggs.
973 # Also helps setuptools not be sad while we refuse to create eggs.
974 single_version_externally_managed = True
974 single_version_externally_managed = True
975
975
976 def get_sub_commands(self):
976 def get_sub_commands(self):
977 # Screen out egg related commands to prevent egg generation. But allow
977 # Screen out egg related commands to prevent egg generation. But allow
978 # mercurial.egg-info generation, since that is part of modern
978 # mercurial.egg-info generation, since that is part of modern
979 # packaging.
979 # packaging.
980 excl = set(['bdist_egg'])
980 excl = set(['bdist_egg'])
981 return filter(lambda x: x not in excl, install.get_sub_commands(self))
981 return filter(lambda x: x not in excl, install.get_sub_commands(self))
982
982
983
983
984 class hginstalllib(install_lib):
984 class hginstalllib(install_lib):
985 '''
985 '''
986 This is a specialization of install_lib that replaces the copy_file used
986 This is a specialization of install_lib that replaces the copy_file used
987 there so that it supports setting the mode of files after copying them,
987 there so that it supports setting the mode of files after copying them,
988 instead of just preserving the mode that the files originally had. If your
988 instead of just preserving the mode that the files originally had. If your
989 system has a umask of something like 027, preserving the permissions when
989 system has a umask of something like 027, preserving the permissions when
990 copying will lead to a broken install.
990 copying will lead to a broken install.
991
991
992 Note that just passing keep_permissions=False to copy_file would be
992 Note that just passing keep_permissions=False to copy_file would be
993 insufficient, as it might still be applying a umask.
993 insufficient, as it might still be applying a umask.
994 '''
994 '''
995
995
996 def run(self):
996 def run(self):
997 realcopyfile = file_util.copy_file
997 realcopyfile = file_util.copy_file
998
998
999 def copyfileandsetmode(*args, **kwargs):
999 def copyfileandsetmode(*args, **kwargs):
1000 src, dst = args[0], args[1]
1000 src, dst = args[0], args[1]
1001 dst, copied = realcopyfile(*args, **kwargs)
1001 dst, copied = realcopyfile(*args, **kwargs)
1002 if copied:
1002 if copied:
1003 st = os.stat(src)
1003 st = os.stat(src)
1004 # Persist executable bit (apply it to group and other if user
1004 # Persist executable bit (apply it to group and other if user
1005 # has it)
1005 # has it)
1006 if st[stat.ST_MODE] & stat.S_IXUSR:
1006 if st[stat.ST_MODE] & stat.S_IXUSR:
1007 setmode = int('0755', 8)
1007 setmode = int('0755', 8)
1008 else:
1008 else:
1009 setmode = int('0644', 8)
1009 setmode = int('0644', 8)
1010 m = stat.S_IMODE(st[stat.ST_MODE])
1010 m = stat.S_IMODE(st[stat.ST_MODE])
1011 m = (m & ~int('0777', 8)) | setmode
1011 m = (m & ~int('0777', 8)) | setmode
1012 os.chmod(dst, m)
1012 os.chmod(dst, m)
1013
1013
1014 file_util.copy_file = copyfileandsetmode
1014 file_util.copy_file = copyfileandsetmode
1015 try:
1015 try:
1016 install_lib.run(self)
1016 install_lib.run(self)
1017 finally:
1017 finally:
1018 file_util.copy_file = realcopyfile
1018 file_util.copy_file = realcopyfile
1019
1019
1020
1020
1021 class hginstallscripts(install_scripts):
1021 class hginstallscripts(install_scripts):
1022 '''
1022 '''
1023 This is a specialization of install_scripts that replaces the @LIBDIR@ with
1023 This is a specialization of install_scripts that replaces the @LIBDIR@ with
1024 the configured directory for modules. If possible, the path is made relative
1024 the configured directory for modules. If possible, the path is made relative
1025 to the directory for scripts.
1025 to the directory for scripts.
1026 '''
1026 '''
1027
1027
1028 def initialize_options(self):
1028 def initialize_options(self):
1029 install_scripts.initialize_options(self)
1029 install_scripts.initialize_options(self)
1030
1030
1031 self.install_lib = None
1031 self.install_lib = None
1032
1032
1033 def finalize_options(self):
1033 def finalize_options(self):
1034 install_scripts.finalize_options(self)
1034 install_scripts.finalize_options(self)
1035 self.set_undefined_options('install', ('install_lib', 'install_lib'))
1035 self.set_undefined_options('install', ('install_lib', 'install_lib'))
1036
1036
1037 def run(self):
1037 def run(self):
1038 install_scripts.run(self)
1038 install_scripts.run(self)
1039
1039
1040 # It only makes sense to replace @LIBDIR@ with the install path if
1040 # It only makes sense to replace @LIBDIR@ with the install path if
1041 # the install path is known. For wheels, the logic below calculates
1041 # the install path is known. For wheels, the logic below calculates
1042 # the libdir to be "../..". This is because the internal layout of a
1042 # the libdir to be "../..". This is because the internal layout of a
1043 # wheel archive looks like:
1043 # wheel archive looks like:
1044 #
1044 #
1045 # mercurial-3.6.1.data/scripts/hg
1045 # mercurial-3.6.1.data/scripts/hg
1046 # mercurial/__init__.py
1046 # mercurial/__init__.py
1047 #
1047 #
1048 # When installing wheels, the subdirectories of the "<pkg>.data"
1048 # When installing wheels, the subdirectories of the "<pkg>.data"
1049 # directory are translated to system local paths and files therein
1049 # directory are translated to system local paths and files therein
1050 # are copied in place. The mercurial/* files are installed into the
1050 # are copied in place. The mercurial/* files are installed into the
1051 # site-packages directory. However, the site-packages directory
1051 # site-packages directory. However, the site-packages directory
1052 # isn't known until wheel install time. This means we have no clue
1052 # isn't known until wheel install time. This means we have no clue
1053 # at wheel generation time what the installed site-packages directory
1053 # at wheel generation time what the installed site-packages directory
1054 # will be. And, wheels don't appear to provide the ability to register
1054 # will be. And, wheels don't appear to provide the ability to register
1055 # custom code to run during wheel installation. This all means that
1055 # custom code to run during wheel installation. This all means that
1056 # we can't reliably set the libdir in wheels: the default behavior
1056 # we can't reliably set the libdir in wheels: the default behavior
1057 # of looking in sys.path must do.
1057 # of looking in sys.path must do.
1058
1058
1059 if (
1059 if (
1060 os.path.splitdrive(self.install_dir)[0]
1060 os.path.splitdrive(self.install_dir)[0]
1061 != os.path.splitdrive(self.install_lib)[0]
1061 != os.path.splitdrive(self.install_lib)[0]
1062 ):
1062 ):
1063 # can't make relative paths from one drive to another, so use an
1063 # can't make relative paths from one drive to another, so use an
1064 # absolute path instead
1064 # absolute path instead
1065 libdir = self.install_lib
1065 libdir = self.install_lib
1066 else:
1066 else:
1067 common = os.path.commonprefix((self.install_dir, self.install_lib))
1067 common = os.path.commonprefix((self.install_dir, self.install_lib))
1068 rest = self.install_dir[len(common) :]
1068 rest = self.install_dir[len(common) :]
1069 uplevel = len([n for n in os.path.split(rest) if n])
1069 uplevel = len([n for n in os.path.split(rest) if n])
1070
1070
1071 libdir = uplevel * ('..' + os.sep) + self.install_lib[len(common) :]
1071 libdir = uplevel * ('..' + os.sep) + self.install_lib[len(common) :]
1072
1072
1073 for outfile in self.outfiles:
1073 for outfile in self.outfiles:
1074 with open(outfile, 'rb') as fp:
1074 with open(outfile, 'rb') as fp:
1075 data = fp.read()
1075 data = fp.read()
1076
1076
1077 # skip binary files
1077 # skip binary files
1078 if b'\0' in data:
1078 if b'\0' in data:
1079 continue
1079 continue
1080
1080
1081 # During local installs, the shebang will be rewritten to the final
1081 # During local installs, the shebang will be rewritten to the final
1082 # install path. During wheel packaging, the shebang has a special
1082 # install path. During wheel packaging, the shebang has a special
1083 # value.
1083 # value.
1084 if data.startswith(b'#!python'):
1084 if data.startswith(b'#!python'):
1085 log.info(
1085 log.info(
1086 'not rewriting @LIBDIR@ in %s because install path '
1086 'not rewriting @LIBDIR@ in %s because install path '
1087 'not known' % outfile
1087 'not known' % outfile
1088 )
1088 )
1089 continue
1089 continue
1090
1090
1091 data = data.replace(b'@LIBDIR@', libdir.encode(libdir_escape))
1091 data = data.replace(b'@LIBDIR@', libdir.encode(libdir_escape))
1092 with open(outfile, 'wb') as fp:
1092 with open(outfile, 'wb') as fp:
1093 fp.write(data)
1093 fp.write(data)
1094
1094
1095
1095
1096 # virtualenv installs custom distutils/__init__.py and
1096 # virtualenv installs custom distutils/__init__.py and
1097 # distutils/distutils.cfg files which essentially proxy back to the
1097 # distutils/distutils.cfg files which essentially proxy back to the
1098 # "real" distutils in the main Python install. The presence of this
1098 # "real" distutils in the main Python install. The presence of this
1099 # directory causes py2exe to pick up the "hacked" distutils package
1099 # directory causes py2exe to pick up the "hacked" distutils package
1100 # from the virtualenv and "import distutils" will fail from the py2exe
1100 # from the virtualenv and "import distutils" will fail from the py2exe
1101 # build because the "real" distutils files can't be located.
1101 # build because the "real" distutils files can't be located.
1102 #
1102 #
1103 # We work around this by monkeypatching the py2exe code finding Python
1103 # We work around this by monkeypatching the py2exe code finding Python
1104 # modules to replace the found virtualenv distutils modules with the
1104 # modules to replace the found virtualenv distutils modules with the
1105 # original versions via filesystem scanning. This is a bit hacky. But
1105 # original versions via filesystem scanning. This is a bit hacky. But
1106 # it allows us to use virtualenvs for py2exe packaging, which is more
1106 # it allows us to use virtualenvs for py2exe packaging, which is more
1107 # deterministic and reproducible.
1107 # deterministic and reproducible.
1108 #
1108 #
1109 # It's worth noting that the common StackOverflow suggestions for this
1109 # It's worth noting that the common StackOverflow suggestions for this
1110 # problem involve copying the original distutils files into the
1110 # problem involve copying the original distutils files into the
1111 # virtualenv or into the staging directory after setup() is invoked.
1111 # virtualenv or into the staging directory after setup() is invoked.
1112 # The former is very brittle and can easily break setup(). Our hacking
1112 # The former is very brittle and can easily break setup(). Our hacking
1113 # of the found modules routine has a similar result as copying the files
1113 # of the found modules routine has a similar result as copying the files
1114 # manually. But it makes fewer assumptions about how py2exe works and
1114 # manually. But it makes fewer assumptions about how py2exe works and
1115 # is less brittle.
1115 # is less brittle.
1116
1116
1117 # This only catches virtualenvs made with virtualenv (as opposed to
1117 # This only catches virtualenvs made with virtualenv (as opposed to
1118 # venv, which is likely what Python 3 uses).
1118 # venv, which is likely what Python 3 uses).
1119 py2exehacked = py2exeloaded and getattr(sys, 'real_prefix', None) is not None
1119 py2exehacked = py2exeloaded and getattr(sys, 'real_prefix', None) is not None
1120
1120
1121 if py2exehacked:
1121 if py2exehacked:
1122 from distutils.command.py2exe import py2exe as buildpy2exe
1122 from distutils.command.py2exe import py2exe as buildpy2exe
1123 from py2exe.mf import Module as py2exemodule
1123 from py2exe.mf import Module as py2exemodule
1124
1124
1125 class hgbuildpy2exe(buildpy2exe):
1125 class hgbuildpy2exe(buildpy2exe):
1126 def find_needed_modules(self, mf, files, modules):
1126 def find_needed_modules(self, mf, files, modules):
1127 res = buildpy2exe.find_needed_modules(self, mf, files, modules)
1127 res = buildpy2exe.find_needed_modules(self, mf, files, modules)
1128
1128
1129 # Replace virtualenv's distutils modules with the real ones.
1129 # Replace virtualenv's distutils modules with the real ones.
1130 modules = {}
1130 modules = {}
1131 for k, v in res.modules.items():
1131 for k, v in res.modules.items():
1132 if k != 'distutils' and not k.startswith('distutils.'):
1132 if k != 'distutils' and not k.startswith('distutils.'):
1133 modules[k] = v
1133 modules[k] = v
1134
1134
1135 res.modules = modules
1135 res.modules = modules
1136
1136
1137 import opcode
1137 import opcode
1138
1138
1139 distutilsreal = os.path.join(
1139 distutilsreal = os.path.join(
1140 os.path.dirname(opcode.__file__), 'distutils'
1140 os.path.dirname(opcode.__file__), 'distutils'
1141 )
1141 )
1142
1142
1143 for root, dirs, files in os.walk(distutilsreal):
1143 for root, dirs, files in os.walk(distutilsreal):
1144 for f in sorted(files):
1144 for f in sorted(files):
1145 if not f.endswith('.py'):
1145 if not f.endswith('.py'):
1146 continue
1146 continue
1147
1147
1148 full = os.path.join(root, f)
1148 full = os.path.join(root, f)
1149
1149
1150 parents = ['distutils']
1150 parents = ['distutils']
1151
1151
1152 if root != distutilsreal:
1152 if root != distutilsreal:
1153 rel = os.path.relpath(root, distutilsreal)
1153 rel = os.path.relpath(root, distutilsreal)
1154 parents.extend(p for p in rel.split(os.sep))
1154 parents.extend(p for p in rel.split(os.sep))
1155
1155
1156 modname = '%s.%s' % ('.'.join(parents), f[:-3])
1156 modname = '%s.%s' % ('.'.join(parents), f[:-3])
1157
1157
1158 if modname.startswith('distutils.tests.'):
1158 if modname.startswith('distutils.tests.'):
1159 continue
1159 continue
1160
1160
1161 if modname.endswith('.__init__'):
1161 if modname.endswith('.__init__'):
1162 modname = modname[: -len('.__init__')]
1162 modname = modname[: -len('.__init__')]
1163 path = os.path.dirname(full)
1163 path = os.path.dirname(full)
1164 else:
1164 else:
1165 path = None
1165 path = None
1166
1166
1167 res.modules[modname] = py2exemodule(
1167 res.modules[modname] = py2exemodule(
1168 modname, full, path=path
1168 modname, full, path=path
1169 )
1169 )
1170
1170
1171 if 'distutils' not in res.modules:
1171 if 'distutils' not in res.modules:
1172 raise SystemExit('could not find distutils modules')
1172 raise SystemExit('could not find distutils modules')
1173
1173
1174 return res
1174 return res
1175
1175
1176
1176
1177 cmdclass = {
1177 cmdclass = {
1178 'build': hgbuild,
1178 'build': hgbuild,
1179 'build_doc': hgbuilddoc,
1179 'build_doc': hgbuilddoc,
1180 'build_mo': hgbuildmo,
1180 'build_mo': hgbuildmo,
1181 'build_ext': hgbuildext,
1181 'build_ext': hgbuildext,
1182 'build_py': hgbuildpy,
1182 'build_py': hgbuildpy,
1183 'build_scripts': hgbuildscripts,
1183 'build_scripts': hgbuildscripts,
1184 'build_hgextindex': buildhgextindex,
1184 'build_hgextindex': buildhgextindex,
1185 'install': hginstall,
1185 'install': hginstall,
1186 'install_lib': hginstalllib,
1186 'install_lib': hginstalllib,
1187 'install_scripts': hginstallscripts,
1187 'install_scripts': hginstallscripts,
1188 'build_hgexe': buildhgexe,
1188 'build_hgexe': buildhgexe,
1189 }
1189 }
1190
1190
1191 if py2exehacked:
1191 if py2exehacked:
1192 cmdclass['py2exe'] = hgbuildpy2exe
1192 cmdclass['py2exe'] = hgbuildpy2exe
1193
1193
1194 packages = [
1194 packages = [
1195 'mercurial',
1195 'mercurial',
1196 'mercurial.cext',
1196 'mercurial.cext',
1197 'mercurial.cffi',
1197 'mercurial.cffi',
1198 'mercurial.defaultrc',
1198 'mercurial.defaultrc',
1199 'mercurial.helptext',
1199 'mercurial.helptext',
1200 'mercurial.helptext.internals',
1200 'mercurial.helptext.internals',
1201 'mercurial.hgweb',
1201 'mercurial.hgweb',
1202 'mercurial.interfaces',
1202 'mercurial.interfaces',
1203 'mercurial.pure',
1203 'mercurial.pure',
1204 'mercurial.thirdparty',
1204 'mercurial.thirdparty',
1205 'mercurial.thirdparty.attr',
1205 'mercurial.thirdparty.attr',
1206 'mercurial.thirdparty.zope',
1206 'mercurial.thirdparty.zope',
1207 'mercurial.thirdparty.zope.interface',
1207 'mercurial.thirdparty.zope.interface',
1208 'mercurial.utils',
1208 'mercurial.utils',
1209 'mercurial.revlogutils',
1209 'mercurial.revlogutils',
1210 'mercurial.testing',
1210 'mercurial.testing',
1211 'hgext',
1211 'hgext',
1212 'hgext.convert',
1212 'hgext.convert',
1213 'hgext.fsmonitor',
1213 'hgext.fsmonitor',
1214 'hgext.fastannotate',
1214 'hgext.fastannotate',
1215 'hgext.fsmonitor.pywatchman',
1215 'hgext.fsmonitor.pywatchman',
1216 'hgext.highlight',
1216 'hgext.highlight',
1217 'hgext.infinitepush',
1217 'hgext.infinitepush',
1218 'hgext.largefiles',
1218 'hgext.largefiles',
1219 'hgext.lfs',
1219 'hgext.lfs',
1220 'hgext.narrow',
1220 'hgext.narrow',
1221 'hgext.remotefilelog',
1221 'hgext.remotefilelog',
1222 'hgext.zeroconf',
1222 'hgext.zeroconf',
1223 'hgext3rd',
1223 'hgext3rd',
1224 'hgdemandimport',
1224 'hgdemandimport',
1225 ]
1225 ]
1226 if sys.version_info[0] == 2:
1226 if sys.version_info[0] == 2:
1227 packages.extend(
1227 packages.extend(
1228 [
1228 [
1229 'mercurial.thirdparty.concurrent',
1229 'mercurial.thirdparty.concurrent',
1230 'mercurial.thirdparty.concurrent.futures',
1230 'mercurial.thirdparty.concurrent.futures',
1231 ]
1231 ]
1232 )
1232 )
1233
1233
1234 if 'HG_PY2EXE_EXTRA_INSTALL_PACKAGES' in os.environ:
1234 if 'HG_PY2EXE_EXTRA_INSTALL_PACKAGES' in os.environ:
1235 # py2exe can't cope with namespace packages very well, so we have to
1235 # py2exe can't cope with namespace packages very well, so we have to
1236 # install any hgext3rd.* extensions that we want in the final py2exe
1236 # install any hgext3rd.* extensions that we want in the final py2exe
1237 # image here. This is gross, but you gotta do what you gotta do.
1237 # image here. This is gross, but you gotta do what you gotta do.
1238 packages.extend(os.environ['HG_PY2EXE_EXTRA_INSTALL_PACKAGES'].split(' '))
1238 packages.extend(os.environ['HG_PY2EXE_EXTRA_INSTALL_PACKAGES'].split(' '))
1239
1239
1240 common_depends = [
1240 common_depends = [
1241 'mercurial/bitmanipulation.h',
1241 'mercurial/bitmanipulation.h',
1242 'mercurial/compat.h',
1242 'mercurial/compat.h',
1243 'mercurial/cext/util.h',
1243 'mercurial/cext/util.h',
1244 ]
1244 ]
1245 common_include_dirs = ['mercurial']
1245 common_include_dirs = ['mercurial']
1246
1246
1247 osutil_cflags = []
1247 osutil_cflags = []
1248 osutil_ldflags = []
1248 osutil_ldflags = []
1249
1249
1250 # platform specific macros
1250 # platform specific macros
1251 for plat, func in [('bsd', 'setproctitle')]:
1251 for plat, func in [('bsd', 'setproctitle')]:
1252 if re.search(plat, sys.platform) and hasfunction(new_compiler(), func):
1252 if re.search(plat, sys.platform) and hasfunction(new_compiler(), func):
1253 osutil_cflags.append('-DHAVE_%s' % func.upper())
1253 osutil_cflags.append('-DHAVE_%s' % func.upper())
1254
1254
1255 for plat, macro, code in [
1255 for plat, macro, code in [
1256 (
1256 (
1257 'bsd|darwin',
1257 'bsd|darwin',
1258 'BSD_STATFS',
1258 'BSD_STATFS',
1259 '''
1259 '''
1260 #include <sys/param.h>
1260 #include <sys/param.h>
1261 #include <sys/mount.h>
1261 #include <sys/mount.h>
1262 int main() { struct statfs s; return sizeof(s.f_fstypename); }
1262 int main() { struct statfs s; return sizeof(s.f_fstypename); }
1263 ''',
1263 ''',
1264 ),
1264 ),
1265 (
1265 (
1266 'linux',
1266 'linux',
1267 'LINUX_STATFS',
1267 'LINUX_STATFS',
1268 '''
1268 '''
1269 #include <linux/magic.h>
1269 #include <linux/magic.h>
1270 #include <sys/vfs.h>
1270 #include <sys/vfs.h>
1271 int main() { struct statfs s; return sizeof(s.f_type); }
1271 int main() { struct statfs s; return sizeof(s.f_type); }
1272 ''',
1272 ''',
1273 ),
1273 ),
1274 ]:
1274 ]:
1275 if re.search(plat, sys.platform) and cancompile(new_compiler(), code):
1275 if re.search(plat, sys.platform) and cancompile(new_compiler(), code):
1276 osutil_cflags.append('-DHAVE_%s' % macro)
1276 osutil_cflags.append('-DHAVE_%s' % macro)
1277
1277
1278 if sys.platform == 'darwin':
1278 if sys.platform == 'darwin':
1279 osutil_ldflags += ['-framework', 'ApplicationServices']
1279 osutil_ldflags += ['-framework', 'ApplicationServices']
1280
1280
1281 xdiff_srcs = [
1281 xdiff_srcs = [
1282 'mercurial/thirdparty/xdiff/xdiffi.c',
1282 'mercurial/thirdparty/xdiff/xdiffi.c',
1283 'mercurial/thirdparty/xdiff/xprepare.c',
1283 'mercurial/thirdparty/xdiff/xprepare.c',
1284 'mercurial/thirdparty/xdiff/xutils.c',
1284 'mercurial/thirdparty/xdiff/xutils.c',
1285 ]
1285 ]
1286
1286
1287 xdiff_headers = [
1287 xdiff_headers = [
1288 'mercurial/thirdparty/xdiff/xdiff.h',
1288 'mercurial/thirdparty/xdiff/xdiff.h',
1289 'mercurial/thirdparty/xdiff/xdiffi.h',
1289 'mercurial/thirdparty/xdiff/xdiffi.h',
1290 'mercurial/thirdparty/xdiff/xinclude.h',
1290 'mercurial/thirdparty/xdiff/xinclude.h',
1291 'mercurial/thirdparty/xdiff/xmacros.h',
1291 'mercurial/thirdparty/xdiff/xmacros.h',
1292 'mercurial/thirdparty/xdiff/xprepare.h',
1292 'mercurial/thirdparty/xdiff/xprepare.h',
1293 'mercurial/thirdparty/xdiff/xtypes.h',
1293 'mercurial/thirdparty/xdiff/xtypes.h',
1294 'mercurial/thirdparty/xdiff/xutils.h',
1294 'mercurial/thirdparty/xdiff/xutils.h',
1295 ]
1295 ]
1296
1296
1297
1297
1298 class RustCompilationError(CCompilerError):
1298 class RustCompilationError(CCompilerError):
1299 """Exception class for Rust compilation errors."""
1299 """Exception class for Rust compilation errors."""
1300
1300
1301
1301
1302 class RustExtension(Extension):
1302 class RustExtension(Extension):
1303 """Base classes for concrete Rust Extension classes.
1303 """Base classes for concrete Rust Extension classes.
1304 """
1304 """
1305
1305
1306 rusttargetdir = os.path.join('rust', 'target', 'release')
1306 rusttargetdir = os.path.join('rust', 'target', 'release')
1307
1307
1308 def __init__(
1308 def __init__(
1309 self, mpath, sources, rustlibname, subcrate, py3_features=None, **kw
1309 self, mpath, sources, rustlibname, subcrate, py3_features=None, **kw
1310 ):
1310 ):
1311 Extension.__init__(self, mpath, sources, **kw)
1311 Extension.__init__(self, mpath, sources, **kw)
1312 srcdir = self.rustsrcdir = os.path.join('rust', subcrate)
1312 srcdir = self.rustsrcdir = os.path.join('rust', subcrate)
1313 self.py3_features = py3_features
1313 self.py3_features = py3_features
1314
1314
1315 # adding Rust source and control files to depends so that the extension
1315 # adding Rust source and control files to depends so that the extension
1316 # gets rebuilt if they've changed
1316 # gets rebuilt if they've changed
1317 self.depends.append(os.path.join(srcdir, 'Cargo.toml'))
1317 self.depends.append(os.path.join(srcdir, 'Cargo.toml'))
1318 cargo_lock = os.path.join(srcdir, 'Cargo.lock')
1318 cargo_lock = os.path.join(srcdir, 'Cargo.lock')
1319 if os.path.exists(cargo_lock):
1319 if os.path.exists(cargo_lock):
1320 self.depends.append(cargo_lock)
1320 self.depends.append(cargo_lock)
1321 for dirpath, subdir, fnames in os.walk(os.path.join(srcdir, 'src')):
1321 for dirpath, subdir, fnames in os.walk(os.path.join(srcdir, 'src')):
1322 self.depends.extend(
1322 self.depends.extend(
1323 os.path.join(dirpath, fname)
1323 os.path.join(dirpath, fname)
1324 for fname in fnames
1324 for fname in fnames
1325 if os.path.splitext(fname)[1] == '.rs'
1325 if os.path.splitext(fname)[1] == '.rs'
1326 )
1326 )
1327
1327
1328 @staticmethod
1328 @staticmethod
1329 def rustdylibsuffix():
1329 def rustdylibsuffix():
1330 """Return the suffix for shared libraries produced by rustc.
1330 """Return the suffix for shared libraries produced by rustc.
1331
1331
1332 See also: https://doc.rust-lang.org/reference/linkage.html
1332 See also: https://doc.rust-lang.org/reference/linkage.html
1333 """
1333 """
1334 if sys.platform == 'darwin':
1334 if sys.platform == 'darwin':
1335 return '.dylib'
1335 return '.dylib'
1336 elif os.name == 'nt':
1336 elif os.name == 'nt':
1337 return '.dll'
1337 return '.dll'
1338 else:
1338 else:
1339 return '.so'
1339 return '.so'
1340
1340
1341 def rustbuild(self):
1341 def rustbuild(self):
1342 env = os.environ.copy()
1342 env = os.environ.copy()
1343 if 'HGTEST_RESTOREENV' in env:
1343 if 'HGTEST_RESTOREENV' in env:
1344 # Mercurial tests change HOME to a temporary directory,
1344 # Mercurial tests change HOME to a temporary directory,
1345 # but, if installed with rustup, the Rust toolchain needs
1345 # but, if installed with rustup, the Rust toolchain needs
1346 # HOME to be correct (otherwise the 'no default toolchain'
1346 # HOME to be correct (otherwise the 'no default toolchain'
1347 # error message is issued and the build fails).
1347 # error message is issued and the build fails).
1348 # This happens currently with test-hghave.t, which does
1348 # This happens currently with test-hghave.t, which does
1349 # invoke this build.
1349 # invoke this build.
1350
1350
1351 # Unix only fix (os.path.expanduser not really reliable if
1351 # Unix only fix (os.path.expanduser not really reliable if
1352 # HOME is shadowed like this)
1352 # HOME is shadowed like this)
1353 import pwd
1353 import pwd
1354
1354
1355 env['HOME'] = pwd.getpwuid(os.getuid()).pw_dir
1355 env['HOME'] = pwd.getpwuid(os.getuid()).pw_dir
1356
1356
1357 cargocmd = ['cargo', 'rustc', '-vv', '--release']
1357 cargocmd = ['cargo', 'rustc', '-vv', '--release']
1358 if sys.version_info[0] == 3 and self.py3_features is not None:
1358 if sys.version_info[0] == 3 and self.py3_features is not None:
1359 cargocmd.extend(
1359 cargocmd.extend(
1360 ('--features', self.py3_features, '--no-default-features')
1360 ('--features', self.py3_features, '--no-default-features')
1361 )
1361 )
1362 cargocmd.append('--')
1362 cargocmd.append('--')
1363 if sys.platform == 'darwin':
1363 if sys.platform == 'darwin':
1364 cargocmd.extend(
1364 cargocmd.extend(
1365 ("-C", "link-arg=-undefined", "-C", "link-arg=dynamic_lookup")
1365 ("-C", "link-arg=-undefined", "-C", "link-arg=dynamic_lookup")
1366 )
1366 )
1367 try:
1367 try:
1368 subprocess.check_call(cargocmd, env=env, cwd=self.rustsrcdir)
1368 subprocess.check_call(cargocmd, env=env, cwd=self.rustsrcdir)
1369 except OSError as exc:
1369 except OSError as exc:
1370 if exc.errno == errno.ENOENT:
1370 if exc.errno == errno.ENOENT:
1371 raise RustCompilationError("Cargo not found")
1371 raise RustCompilationError("Cargo not found")
1372 elif exc.errno == errno.EACCES:
1372 elif exc.errno == errno.EACCES:
1373 raise RustCompilationError(
1373 raise RustCompilationError(
1374 "Cargo found, but permisssion to execute it is denied"
1374 "Cargo found, but permisssion to execute it is denied"
1375 )
1375 )
1376 else:
1376 else:
1377 raise
1377 raise
1378 except subprocess.CalledProcessError:
1378 except subprocess.CalledProcessError:
1379 raise RustCompilationError(
1379 raise RustCompilationError(
1380 "Cargo failed. Working directory: %r, "
1380 "Cargo failed. Working directory: %r, "
1381 "command: %r, environment: %r"
1381 "command: %r, environment: %r"
1382 % (self.rustsrcdir, cargocmd, env)
1382 % (self.rustsrcdir, cargocmd, env)
1383 )
1383 )
1384
1384
1385
1385
1386 class RustEnhancedExtension(RustExtension):
1386 class RustEnhancedExtension(RustExtension):
1387 """A C Extension, conditionally enhanced with Rust code.
1387 """A C Extension, conditionally enhanced with Rust code.
1388
1388
1389 If the HGRUSTEXT environment variable is set to something else
1389 If the HGRUSTEXT environment variable is set to something else
1390 than 'cpython', the Rust sources get compiled and linked within the
1390 than 'cpython', the Rust sources get compiled and linked within the
1391 C target shared library object.
1391 C target shared library object.
1392 """
1392 """
1393
1393
1394 def __init__(self, mpath, sources, rustlibname, subcrate, **kw):
1394 def __init__(self, mpath, sources, rustlibname, subcrate, **kw):
1395 RustExtension.__init__(
1395 RustExtension.__init__(
1396 self, mpath, sources, rustlibname, subcrate, **kw
1396 self, mpath, sources, rustlibname, subcrate, **kw
1397 )
1397 )
1398 if hgrustext != 'direct-ffi':
1398 if hgrustext != 'direct-ffi':
1399 return
1399 return
1400 self.extra_compile_args.append('-DWITH_RUST')
1400 self.extra_compile_args.append('-DWITH_RUST')
1401 self.libraries.append(rustlibname)
1401 self.libraries.append(rustlibname)
1402 self.library_dirs.append(self.rusttargetdir)
1402 self.library_dirs.append(self.rusttargetdir)
1403
1403
1404 def rustbuild(self):
1404 def rustbuild(self):
1405 if hgrustext == 'direct-ffi':
1405 if hgrustext == 'direct-ffi':
1406 RustExtension.rustbuild(self)
1406 RustExtension.rustbuild(self)
1407
1407
1408
1408
1409 class RustStandaloneExtension(RustExtension):
1409 class RustStandaloneExtension(RustExtension):
1410 def __init__(self, pydottedname, rustcrate, dylibname, **kw):
1410 def __init__(self, pydottedname, rustcrate, dylibname, **kw):
1411 RustExtension.__init__(
1411 RustExtension.__init__(
1412 self, pydottedname, [], dylibname, rustcrate, **kw
1412 self, pydottedname, [], dylibname, rustcrate, **kw
1413 )
1413 )
1414 self.dylibname = dylibname
1414 self.dylibname = dylibname
1415
1415
1416 def build(self, target_dir):
1416 def build(self, target_dir):
1417 self.rustbuild()
1417 self.rustbuild()
1418 target = [target_dir]
1418 target = [target_dir]
1419 target.extend(self.name.split('.'))
1419 target.extend(self.name.split('.'))
1420 target[-1] += DYLIB_SUFFIX
1420 target[-1] += DYLIB_SUFFIX
1421 shutil.copy2(
1421 shutil.copy2(
1422 os.path.join(
1422 os.path.join(
1423 self.rusttargetdir, self.dylibname + self.rustdylibsuffix()
1423 self.rusttargetdir, self.dylibname + self.rustdylibsuffix()
1424 ),
1424 ),
1425 os.path.join(*target),
1425 os.path.join(*target),
1426 )
1426 )
1427
1427
1428
1428
1429 extmodules = [
1429 extmodules = [
1430 Extension(
1430 Extension(
1431 'mercurial.cext.base85',
1431 'mercurial.cext.base85',
1432 ['mercurial/cext/base85.c'],
1432 ['mercurial/cext/base85.c'],
1433 include_dirs=common_include_dirs,
1433 include_dirs=common_include_dirs,
1434 depends=common_depends,
1434 depends=common_depends,
1435 ),
1435 ),
1436 Extension(
1436 Extension(
1437 'mercurial.cext.bdiff',
1437 'mercurial.cext.bdiff',
1438 ['mercurial/bdiff.c', 'mercurial/cext/bdiff.c'] + xdiff_srcs,
1438 ['mercurial/bdiff.c', 'mercurial/cext/bdiff.c'] + xdiff_srcs,
1439 include_dirs=common_include_dirs,
1439 include_dirs=common_include_dirs,
1440 depends=common_depends + ['mercurial/bdiff.h'] + xdiff_headers,
1440 depends=common_depends + ['mercurial/bdiff.h'] + xdiff_headers,
1441 ),
1441 ),
1442 Extension(
1442 Extension(
1443 'mercurial.cext.mpatch',
1443 'mercurial.cext.mpatch',
1444 ['mercurial/mpatch.c', 'mercurial/cext/mpatch.c'],
1444 ['mercurial/mpatch.c', 'mercurial/cext/mpatch.c'],
1445 include_dirs=common_include_dirs,
1445 include_dirs=common_include_dirs,
1446 depends=common_depends,
1446 depends=common_depends,
1447 ),
1447 ),
1448 RustEnhancedExtension(
1448 RustEnhancedExtension(
1449 'mercurial.cext.parsers',
1449 'mercurial.cext.parsers',
1450 [
1450 [
1451 'mercurial/cext/charencode.c',
1451 'mercurial/cext/charencode.c',
1452 'mercurial/cext/dirs.c',
1452 'mercurial/cext/dirs.c',
1453 'mercurial/cext/manifest.c',
1453 'mercurial/cext/manifest.c',
1454 'mercurial/cext/parsers.c',
1454 'mercurial/cext/parsers.c',
1455 'mercurial/cext/pathencode.c',
1455 'mercurial/cext/pathencode.c',
1456 'mercurial/cext/revlog.c',
1456 'mercurial/cext/revlog.c',
1457 ],
1457 ],
1458 'hgdirectffi',
1458 'hgdirectffi',
1459 'hg-direct-ffi',
1459 'hg-direct-ffi',
1460 include_dirs=common_include_dirs,
1460 include_dirs=common_include_dirs,
1461 depends=common_depends
1461 depends=common_depends
1462 + [
1462 + [
1463 'mercurial/cext/charencode.h',
1463 'mercurial/cext/charencode.h',
1464 'mercurial/cext/revlog.h',
1464 'mercurial/cext/revlog.h',
1465 'rust/hg-core/src/ancestors.rs',
1465 'rust/hg-core/src/ancestors.rs',
1466 'rust/hg-core/src/lib.rs',
1466 'rust/hg-core/src/lib.rs',
1467 ],
1467 ],
1468 ),
1468 ),
1469 Extension(
1469 Extension(
1470 'mercurial.cext.osutil',
1470 'mercurial.cext.osutil',
1471 ['mercurial/cext/osutil.c'],
1471 ['mercurial/cext/osutil.c'],
1472 include_dirs=common_include_dirs,
1472 include_dirs=common_include_dirs,
1473 extra_compile_args=osutil_cflags,
1473 extra_compile_args=osutil_cflags,
1474 extra_link_args=osutil_ldflags,
1474 extra_link_args=osutil_ldflags,
1475 depends=common_depends,
1475 depends=common_depends,
1476 ),
1476 ),
1477 Extension(
1477 Extension(
1478 'mercurial.thirdparty.zope.interface._zope_interface_coptimizations',
1478 'mercurial.thirdparty.zope.interface._zope_interface_coptimizations',
1479 [
1479 [
1480 'mercurial/thirdparty/zope/interface/_zope_interface_coptimizations.c',
1480 'mercurial/thirdparty/zope/interface/_zope_interface_coptimizations.c',
1481 ],
1481 ],
1482 ),
1482 ),
1483 Extension(
1483 Extension(
1484 'mercurial.thirdparty.sha1dc',
1485 [
1486 'mercurial/thirdparty/sha1dc/cext.c',
1487 'mercurial/thirdparty/sha1dc/lib/sha1.c',
1488 'mercurial/thirdparty/sha1dc/lib/ubc_check.c',
1489 ],
1490 ),
1491 Extension(
1484 'hgext.fsmonitor.pywatchman.bser', ['hgext/fsmonitor/pywatchman/bser.c']
1492 'hgext.fsmonitor.pywatchman.bser', ['hgext/fsmonitor/pywatchman/bser.c']
1485 ),
1493 ),
1486 RustStandaloneExtension(
1494 RustStandaloneExtension(
1487 'mercurial.rustext', 'hg-cpython', 'librusthg', py3_features='python3'
1495 'mercurial.rustext', 'hg-cpython', 'librusthg', py3_features='python3'
1488 ),
1496 ),
1489 ]
1497 ]
1490
1498
1491
1499
1492 sys.path.insert(0, 'contrib/python-zstandard')
1500 sys.path.insert(0, 'contrib/python-zstandard')
1493 import setup_zstd
1501 import setup_zstd
1494
1502
1495 extmodules.append(
1503 extmodules.append(
1496 setup_zstd.get_c_extension(
1504 setup_zstd.get_c_extension(
1497 name='mercurial.zstd', root=os.path.abspath(os.path.dirname(__file__))
1505 name='mercurial.zstd', root=os.path.abspath(os.path.dirname(__file__))
1498 )
1506 )
1499 )
1507 )
1500
1508
1501 try:
1509 try:
1502 from distutils import cygwinccompiler
1510 from distutils import cygwinccompiler
1503
1511
1504 # the -mno-cygwin option has been deprecated for years
1512 # the -mno-cygwin option has been deprecated for years
1505 mingw32compilerclass = cygwinccompiler.Mingw32CCompiler
1513 mingw32compilerclass = cygwinccompiler.Mingw32CCompiler
1506
1514
1507 class HackedMingw32CCompiler(cygwinccompiler.Mingw32CCompiler):
1515 class HackedMingw32CCompiler(cygwinccompiler.Mingw32CCompiler):
1508 def __init__(self, *args, **kwargs):
1516 def __init__(self, *args, **kwargs):
1509 mingw32compilerclass.__init__(self, *args, **kwargs)
1517 mingw32compilerclass.__init__(self, *args, **kwargs)
1510 for i in 'compiler compiler_so linker_exe linker_so'.split():
1518 for i in 'compiler compiler_so linker_exe linker_so'.split():
1511 try:
1519 try:
1512 getattr(self, i).remove('-mno-cygwin')
1520 getattr(self, i).remove('-mno-cygwin')
1513 except ValueError:
1521 except ValueError:
1514 pass
1522 pass
1515
1523
1516 cygwinccompiler.Mingw32CCompiler = HackedMingw32CCompiler
1524 cygwinccompiler.Mingw32CCompiler = HackedMingw32CCompiler
1517 except ImportError:
1525 except ImportError:
1518 # the cygwinccompiler package is not available on some Python
1526 # the cygwinccompiler package is not available on some Python
1519 # distributions like the ones from the optware project for Synology
1527 # distributions like the ones from the optware project for Synology
1520 # DiskStation boxes
1528 # DiskStation boxes
1521 class HackedMingw32CCompiler(object):
1529 class HackedMingw32CCompiler(object):
1522 pass
1530 pass
1523
1531
1524
1532
1525 if os.name == 'nt':
1533 if os.name == 'nt':
1526 # Allow compiler/linker flags to be added to Visual Studio builds. Passing
1534 # Allow compiler/linker flags to be added to Visual Studio builds. Passing
1527 # extra_link_args to distutils.extensions.Extension() doesn't have any
1535 # extra_link_args to distutils.extensions.Extension() doesn't have any
1528 # effect.
1536 # effect.
1529 from distutils import msvccompiler
1537 from distutils import msvccompiler
1530
1538
1531 msvccompilerclass = msvccompiler.MSVCCompiler
1539 msvccompilerclass = msvccompiler.MSVCCompiler
1532
1540
1533 class HackedMSVCCompiler(msvccompiler.MSVCCompiler):
1541 class HackedMSVCCompiler(msvccompiler.MSVCCompiler):
1534 def initialize(self):
1542 def initialize(self):
1535 msvccompilerclass.initialize(self)
1543 msvccompilerclass.initialize(self)
1536 # "warning LNK4197: export 'func' specified multiple times"
1544 # "warning LNK4197: export 'func' specified multiple times"
1537 self.ldflags_shared.append('/ignore:4197')
1545 self.ldflags_shared.append('/ignore:4197')
1538 self.ldflags_shared_debug.append('/ignore:4197')
1546 self.ldflags_shared_debug.append('/ignore:4197')
1539
1547
1540 msvccompiler.MSVCCompiler = HackedMSVCCompiler
1548 msvccompiler.MSVCCompiler = HackedMSVCCompiler
1541
1549
1542 packagedata = {
1550 packagedata = {
1543 'mercurial': [
1551 'mercurial': [
1544 'locale/*/LC_MESSAGES/hg.mo',
1552 'locale/*/LC_MESSAGES/hg.mo',
1545 'defaultrc/*.rc',
1553 'defaultrc/*.rc',
1546 'dummycert.pem',
1554 'dummycert.pem',
1547 ],
1555 ],
1548 'mercurial.helptext': ['*.txt',],
1556 'mercurial.helptext': ['*.txt',],
1549 'mercurial.helptext.internals': ['*.txt',],
1557 'mercurial.helptext.internals': ['*.txt',],
1550 }
1558 }
1551
1559
1552
1560
1553 def ordinarypath(p):
1561 def ordinarypath(p):
1554 return p and p[0] != '.' and p[-1] != '~'
1562 return p and p[0] != '.' and p[-1] != '~'
1555
1563
1556
1564
1557 for root in ('templates',):
1565 for root in ('templates',):
1558 for curdir, dirs, files in os.walk(os.path.join('mercurial', root)):
1566 for curdir, dirs, files in os.walk(os.path.join('mercurial', root)):
1559 curdir = curdir.split(os.sep, 1)[1]
1567 curdir = curdir.split(os.sep, 1)[1]
1560 dirs[:] = filter(ordinarypath, dirs)
1568 dirs[:] = filter(ordinarypath, dirs)
1561 for f in filter(ordinarypath, files):
1569 for f in filter(ordinarypath, files):
1562 f = os.path.join(curdir, f)
1570 f = os.path.join(curdir, f)
1563 packagedata['mercurial'].append(f)
1571 packagedata['mercurial'].append(f)
1564
1572
1565 datafiles = []
1573 datafiles = []
1566
1574
1567 # distutils expects version to be str/unicode. Converting it to
1575 # distutils expects version to be str/unicode. Converting it to
1568 # unicode on Python 2 still works because it won't contain any
1576 # unicode on Python 2 still works because it won't contain any
1569 # non-ascii bytes and will be implicitly converted back to bytes
1577 # non-ascii bytes and will be implicitly converted back to bytes
1570 # when operated on.
1578 # when operated on.
1571 assert isinstance(version, bytes)
1579 assert isinstance(version, bytes)
1572 setupversion = version.decode('ascii')
1580 setupversion = version.decode('ascii')
1573
1581
1574 extra = {}
1582 extra = {}
1575
1583
1576 py2exepackages = [
1584 py2exepackages = [
1577 'hgdemandimport',
1585 'hgdemandimport',
1578 'hgext3rd',
1586 'hgext3rd',
1579 'hgext',
1587 'hgext',
1580 'email',
1588 'email',
1581 # implicitly imported per module policy
1589 # implicitly imported per module policy
1582 # (cffi wouldn't be used as a frozen exe)
1590 # (cffi wouldn't be used as a frozen exe)
1583 'mercurial.cext',
1591 'mercurial.cext',
1584 #'mercurial.cffi',
1592 #'mercurial.cffi',
1585 'mercurial.pure',
1593 'mercurial.pure',
1586 ]
1594 ]
1587
1595
1588 py2exeexcludes = []
1596 py2exeexcludes = []
1589 py2exedllexcludes = ['crypt32.dll']
1597 py2exedllexcludes = ['crypt32.dll']
1590
1598
1591 if issetuptools:
1599 if issetuptools:
1592 extra['python_requires'] = supportedpy
1600 extra['python_requires'] = supportedpy
1593
1601
1594 if py2exeloaded:
1602 if py2exeloaded:
1595 extra['console'] = [
1603 extra['console'] = [
1596 {
1604 {
1597 'script': 'hg',
1605 'script': 'hg',
1598 'copyright': 'Copyright (C) 2005-2019 Matt Mackall and others',
1606 'copyright': 'Copyright (C) 2005-2019 Matt Mackall and others',
1599 'product_version': version,
1607 'product_version': version,
1600 }
1608 }
1601 ]
1609 ]
1602 # Sub command of 'build' because 'py2exe' does not handle sub_commands.
1610 # Sub command of 'build' because 'py2exe' does not handle sub_commands.
1603 # Need to override hgbuild because it has a private copy of
1611 # Need to override hgbuild because it has a private copy of
1604 # build.sub_commands.
1612 # build.sub_commands.
1605 hgbuild.sub_commands.insert(0, ('build_hgextindex', None))
1613 hgbuild.sub_commands.insert(0, ('build_hgextindex', None))
1606 # put dlls in sub directory so that they won't pollute PATH
1614 # put dlls in sub directory so that they won't pollute PATH
1607 extra['zipfile'] = 'lib/library.zip'
1615 extra['zipfile'] = 'lib/library.zip'
1608
1616
1609 # We allow some configuration to be supplemented via environment
1617 # We allow some configuration to be supplemented via environment
1610 # variables. This is better than setup.cfg files because it allows
1618 # variables. This is better than setup.cfg files because it allows
1611 # supplementing configs instead of replacing them.
1619 # supplementing configs instead of replacing them.
1612 extrapackages = os.environ.get('HG_PY2EXE_EXTRA_PACKAGES')
1620 extrapackages = os.environ.get('HG_PY2EXE_EXTRA_PACKAGES')
1613 if extrapackages:
1621 if extrapackages:
1614 py2exepackages.extend(extrapackages.split(' '))
1622 py2exepackages.extend(extrapackages.split(' '))
1615
1623
1616 excludes = os.environ.get('HG_PY2EXE_EXTRA_EXCLUDES')
1624 excludes = os.environ.get('HG_PY2EXE_EXTRA_EXCLUDES')
1617 if excludes:
1625 if excludes:
1618 py2exeexcludes.extend(excludes.split(' '))
1626 py2exeexcludes.extend(excludes.split(' '))
1619
1627
1620 dllexcludes = os.environ.get('HG_PY2EXE_EXTRA_DLL_EXCLUDES')
1628 dllexcludes = os.environ.get('HG_PY2EXE_EXTRA_DLL_EXCLUDES')
1621 if dllexcludes:
1629 if dllexcludes:
1622 py2exedllexcludes.extend(dllexcludes.split(' '))
1630 py2exedllexcludes.extend(dllexcludes.split(' '))
1623
1631
1624 if os.name == 'nt':
1632 if os.name == 'nt':
1625 # Windows binary file versions for exe/dll files must have the
1633 # Windows binary file versions for exe/dll files must have the
1626 # form W.X.Y.Z, where W,X,Y,Z are numbers in the range 0..65535
1634 # form W.X.Y.Z, where W,X,Y,Z are numbers in the range 0..65535
1627 setupversion = setupversion.split(r'+', 1)[0]
1635 setupversion = setupversion.split(r'+', 1)[0]
1628
1636
1629 if sys.platform == 'darwin' and os.path.exists('/usr/bin/xcodebuild'):
1637 if sys.platform == 'darwin' and os.path.exists('/usr/bin/xcodebuild'):
1630 version = runcmd(['/usr/bin/xcodebuild', '-version'], {})[1].splitlines()
1638 version = runcmd(['/usr/bin/xcodebuild', '-version'], {})[1].splitlines()
1631 if version:
1639 if version:
1632 version = version[0]
1640 version = version[0]
1633 if sys.version_info[0] == 3:
1641 if sys.version_info[0] == 3:
1634 version = version.decode('utf-8')
1642 version = version.decode('utf-8')
1635 xcode4 = version.startswith('Xcode') and StrictVersion(
1643 xcode4 = version.startswith('Xcode') and StrictVersion(
1636 version.split()[1]
1644 version.split()[1]
1637 ) >= StrictVersion('4.0')
1645 ) >= StrictVersion('4.0')
1638 xcode51 = re.match(r'^Xcode\s+5\.1', version) is not None
1646 xcode51 = re.match(r'^Xcode\s+5\.1', version) is not None
1639 else:
1647 else:
1640 # xcodebuild returns empty on OS X Lion with XCode 4.3 not
1648 # xcodebuild returns empty on OS X Lion with XCode 4.3 not
1641 # installed, but instead with only command-line tools. Assume
1649 # installed, but instead with only command-line tools. Assume
1642 # that only happens on >= Lion, thus no PPC support.
1650 # that only happens on >= Lion, thus no PPC support.
1643 xcode4 = True
1651 xcode4 = True
1644 xcode51 = False
1652 xcode51 = False
1645
1653
1646 # XCode 4.0 dropped support for ppc architecture, which is hardcoded in
1654 # XCode 4.0 dropped support for ppc architecture, which is hardcoded in
1647 # distutils.sysconfig
1655 # distutils.sysconfig
1648 if xcode4:
1656 if xcode4:
1649 os.environ['ARCHFLAGS'] = ''
1657 os.environ['ARCHFLAGS'] = ''
1650
1658
1651 # XCode 5.1 changes clang such that it now fails to compile if the
1659 # XCode 5.1 changes clang such that it now fails to compile if the
1652 # -mno-fused-madd flag is passed, but the version of Python shipped with
1660 # -mno-fused-madd flag is passed, but the version of Python shipped with
1653 # OS X 10.9 Mavericks includes this flag. This causes problems in all
1661 # OS X 10.9 Mavericks includes this flag. This causes problems in all
1654 # C extension modules, and a bug has been filed upstream at
1662 # C extension modules, and a bug has been filed upstream at
1655 # http://bugs.python.org/issue21244. We also need to patch this here
1663 # http://bugs.python.org/issue21244. We also need to patch this here
1656 # so Mercurial can continue to compile in the meantime.
1664 # so Mercurial can continue to compile in the meantime.
1657 if xcode51:
1665 if xcode51:
1658 cflags = get_config_var('CFLAGS')
1666 cflags = get_config_var('CFLAGS')
1659 if cflags and re.search(r'-mno-fused-madd\b', cflags) is not None:
1667 if cflags and re.search(r'-mno-fused-madd\b', cflags) is not None:
1660 os.environ['CFLAGS'] = (
1668 os.environ['CFLAGS'] = (
1661 os.environ.get('CFLAGS', '') + ' -Qunused-arguments'
1669 os.environ.get('CFLAGS', '') + ' -Qunused-arguments'
1662 )
1670 )
1663
1671
1664 setup(
1672 setup(
1665 name='mercurial',
1673 name='mercurial',
1666 version=setupversion,
1674 version=setupversion,
1667 author='Matt Mackall and many others',
1675 author='Matt Mackall and many others',
1668 author_email='mercurial@mercurial-scm.org',
1676 author_email='mercurial@mercurial-scm.org',
1669 url='https://mercurial-scm.org/',
1677 url='https://mercurial-scm.org/',
1670 download_url='https://mercurial-scm.org/release/',
1678 download_url='https://mercurial-scm.org/release/',
1671 description=(
1679 description=(
1672 'Fast scalable distributed SCM (revision control, version '
1680 'Fast scalable distributed SCM (revision control, version '
1673 'control) system'
1681 'control) system'
1674 ),
1682 ),
1675 long_description=(
1683 long_description=(
1676 'Mercurial is a distributed SCM tool written in Python.'
1684 'Mercurial is a distributed SCM tool written in Python.'
1677 ' It is used by a number of large projects that require'
1685 ' It is used by a number of large projects that require'
1678 ' fast, reliable distributed revision control, such as '
1686 ' fast, reliable distributed revision control, such as '
1679 'Mozilla.'
1687 'Mozilla.'
1680 ),
1688 ),
1681 license='GNU GPLv2 or any later version',
1689 license='GNU GPLv2 or any later version',
1682 classifiers=[
1690 classifiers=[
1683 'Development Status :: 6 - Mature',
1691 'Development Status :: 6 - Mature',
1684 'Environment :: Console',
1692 'Environment :: Console',
1685 'Intended Audience :: Developers',
1693 'Intended Audience :: Developers',
1686 'Intended Audience :: System Administrators',
1694 'Intended Audience :: System Administrators',
1687 'License :: OSI Approved :: GNU General Public License (GPL)',
1695 'License :: OSI Approved :: GNU General Public License (GPL)',
1688 'Natural Language :: Danish',
1696 'Natural Language :: Danish',
1689 'Natural Language :: English',
1697 'Natural Language :: English',
1690 'Natural Language :: German',
1698 'Natural Language :: German',
1691 'Natural Language :: Italian',
1699 'Natural Language :: Italian',
1692 'Natural Language :: Japanese',
1700 'Natural Language :: Japanese',
1693 'Natural Language :: Portuguese (Brazilian)',
1701 'Natural Language :: Portuguese (Brazilian)',
1694 'Operating System :: Microsoft :: Windows',
1702 'Operating System :: Microsoft :: Windows',
1695 'Operating System :: OS Independent',
1703 'Operating System :: OS Independent',
1696 'Operating System :: POSIX',
1704 'Operating System :: POSIX',
1697 'Programming Language :: C',
1705 'Programming Language :: C',
1698 'Programming Language :: Python',
1706 'Programming Language :: Python',
1699 'Topic :: Software Development :: Version Control',
1707 'Topic :: Software Development :: Version Control',
1700 ],
1708 ],
1701 scripts=scripts,
1709 scripts=scripts,
1702 packages=packages,
1710 packages=packages,
1703 ext_modules=extmodules,
1711 ext_modules=extmodules,
1704 data_files=datafiles,
1712 data_files=datafiles,
1705 package_data=packagedata,
1713 package_data=packagedata,
1706 cmdclass=cmdclass,
1714 cmdclass=cmdclass,
1707 distclass=hgdist,
1715 distclass=hgdist,
1708 options={
1716 options={
1709 'py2exe': {
1717 'py2exe': {
1710 'bundle_files': 3,
1718 'bundle_files': 3,
1711 'dll_excludes': py2exedllexcludes,
1719 'dll_excludes': py2exedllexcludes,
1712 'excludes': py2exeexcludes,
1720 'excludes': py2exeexcludes,
1713 'packages': py2exepackages,
1721 'packages': py2exepackages,
1714 },
1722 },
1715 'bdist_mpkg': {
1723 'bdist_mpkg': {
1716 'zipdist': False,
1724 'zipdist': False,
1717 'license': 'COPYING',
1725 'license': 'COPYING',
1718 'readme': 'contrib/packaging/macosx/Readme.html',
1726 'readme': 'contrib/packaging/macosx/Readme.html',
1719 'welcome': 'contrib/packaging/macosx/Welcome.html',
1727 'welcome': 'contrib/packaging/macosx/Welcome.html',
1720 },
1728 },
1721 },
1729 },
1722 **extra
1730 **extra
1723 )
1731 )
General Comments 0
You need to be logged in to leave comments. Login now