Show More
@@ -0,0 +1,132 b'' | |||
|
1 | /* | |
|
2 | base85 codec | |
|
3 | ||
|
4 | Copyright 2006 Brendan Cully <brendan@kublai.com> | |
|
5 | ||
|
6 | This software may be used and distributed according to the terms of | |
|
7 | the GNU General Public License, incorporated herein by reference. | |
|
8 | ||
|
9 | Largely based on git's implementation | |
|
10 | */ | |
|
11 | ||
|
12 | #include <Python.h> | |
|
13 | ||
|
14 | static const char b85chars[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ" | |
|
15 | "abcdefghijklmnopqrstuvwxyz!#$%&()*+-;<=>?@^_`{|}~"; | |
|
16 | static char b85dec[256]; | |
|
17 | ||
|
18 | static void | |
|
19 | b85prep(void) | |
|
20 | { | |
|
21 | int i; | |
|
22 | ||
|
23 | memset(b85dec, 0, sizeof(b85dec)); | |
|
24 | for (i = 0; i < sizeof(b85chars); i++) | |
|
25 | b85dec[(int)(b85chars[i])] = i + 1; | |
|
26 | } | |
|
27 | ||
|
28 | static PyObject * | |
|
29 | b85encode(PyObject *self, PyObject *args) | |
|
30 | { | |
|
31 | const unsigned char *text; | |
|
32 | PyObject *out; | |
|
33 | char *dst; | |
|
34 | int len, olen, i; | |
|
35 | unsigned int acc, val, ch; | |
|
36 | ||
|
37 | if (!PyArg_ParseTuple(args, "s#", &text, &len)) | |
|
38 | return NULL; | |
|
39 | ||
|
40 | olen = (len + 3) / 4 * 5; | |
|
41 | if (!(out = PyString_FromStringAndSize(NULL, olen))) | |
|
42 | return NULL; | |
|
43 | ||
|
44 | dst = PyString_AS_STRING(out); | |
|
45 | ||
|
46 | while (len) | |
|
47 | { | |
|
48 | acc = 0; | |
|
49 | for (i = 24; i >= 0; i -= 8) { | |
|
50 | ch = *text++; | |
|
51 | acc |= ch << i; | |
|
52 | if (--len == 0) | |
|
53 | break; | |
|
54 | } | |
|
55 | for (i = 4; i >= 0; i--) { | |
|
56 | val = acc % 85; | |
|
57 | acc /= 85; | |
|
58 | dst[i] = b85chars[val]; | |
|
59 | } | |
|
60 | dst += 5; | |
|
61 | } | |
|
62 | ||
|
63 | return out; | |
|
64 | } | |
|
65 | ||
|
66 | static PyObject * | |
|
67 | b85decode(PyObject *self, PyObject *args) | |
|
68 | { | |
|
69 | PyObject *out; | |
|
70 | const char *text; | |
|
71 | char *dst; | |
|
72 | int len, i, j, olen, c; | |
|
73 | unsigned int acc; | |
|
74 | ||
|
75 | if (!PyArg_ParseTuple(args, "s#", &text, &len)) | |
|
76 | return NULL; | |
|
77 | ||
|
78 | olen = (len + 4) / 5 * 4; | |
|
79 | if (!(out = PyString_FromStringAndSize(NULL, olen))) | |
|
80 | return NULL; | |
|
81 | ||
|
82 | dst = PyString_AS_STRING(out); | |
|
83 | ||
|
84 | for (i = 1; len; i++) | |
|
85 | { | |
|
86 | acc = 0; | |
|
87 | for (j = 0; j < 4 && --len; j++) | |
|
88 | { | |
|
89 | c = b85dec[(int)*text++] - 1; | |
|
90 | if (c < 0) | |
|
91 | return PyErr_Format(PyExc_ValueError, "Bad base85 character at position %d", i); | |
|
92 | acc = acc * 85 + c; | |
|
93 | } | |
|
94 | if (len--) | |
|
95 | { | |
|
96 | c = b85dec[(int)*text++] - 1; | |
|
97 | if (c < 0) | |
|
98 | return PyErr_Format(PyExc_ValueError, "Bad base85 character at position %d", i); | |
|
99 | } | |
|
100 | else | |
|
101 | c = 0; | |
|
102 | /* overflow detection: 0xffffffff == "|NsC0", | |
|
103 | * "|NsC" == 0x03030303 */ | |
|
104 | if (acc > 0x03030303 || (acc *= 85) > 0xffffffff - c) | |
|
105 | return PyErr_Format(PyExc_ValueError, "Bad base85 sequence at position %d", i); | |
|
106 | ||
|
107 | acc += c; | |
|
108 | ||
|
109 | for (j = 0; j < 4; j++) | |
|
110 | { | |
|
111 | acc = (acc << 8) | (acc >> 24); | |
|
112 | *dst++ = (char)acc; | |
|
113 | } | |
|
114 | } | |
|
115 | ||
|
116 | return out; | |
|
117 | } | |
|
118 | ||
|
119 | static char base85_doc[] = "Base85 Data Encoding"; | |
|
120 | ||
|
121 | static PyMethodDef methods[] = { | |
|
122 | {"b85encode", b85encode, METH_VARARGS, "encode text in base85\n"}, | |
|
123 | {"b85decode", b85decode, METH_VARARGS, "decode base85 text\n"}, | |
|
124 | {NULL, NULL} | |
|
125 | }; | |
|
126 | ||
|
127 | PyMODINIT_FUNC initbase85(void) | |
|
128 | { | |
|
129 | Py_InitModule3("base85", methods, base85_doc); | |
|
130 | ||
|
131 | b85prep(); | |
|
132 | } |
@@ -1,102 +1,103 b'' | |||
|
1 | 1 | #!/usr/bin/env python |
|
2 | 2 | # |
|
3 | 3 | # This is the mercurial setup script. |
|
4 | 4 | # |
|
5 | 5 | # './setup.py install', or |
|
6 | 6 | # './setup.py --help' for more options |
|
7 | 7 | |
|
8 | 8 | import sys |
|
9 | 9 | if not hasattr(sys, 'version_info') or sys.version_info < (2, 3): |
|
10 | 10 | raise SystemExit, "Mercurial requires python 2.3 or later." |
|
11 | 11 | |
|
12 | 12 | import os |
|
13 | 13 | from distutils.core import setup, Extension |
|
14 | 14 | from distutils.command.install_data import install_data |
|
15 | 15 | |
|
16 | 16 | # mercurial.packagescan must be the first mercurial module imported |
|
17 | 17 | import mercurial.packagescan |
|
18 | 18 | import mercurial.version |
|
19 | 19 | |
|
20 | 20 | # py2exe needs to be installed to work |
|
21 | 21 | try: |
|
22 | 22 | import py2exe |
|
23 | 23 | |
|
24 | 24 | # Help py2exe to find win32com.shell |
|
25 | 25 | try: |
|
26 | 26 | import modulefinder |
|
27 | 27 | import win32com |
|
28 | 28 | for p in win32com.__path__[1:]: # Take the path to win32comext |
|
29 | 29 | modulefinder.AddPackagePath("win32com", p) |
|
30 | 30 | pn = "win32com.shell" |
|
31 | 31 | __import__(pn) |
|
32 | 32 | m = sys.modules[pn] |
|
33 | 33 | for p in m.__path__[1:]: |
|
34 | 34 | modulefinder.AddPackagePath(pn, p) |
|
35 | 35 | except ImportError: |
|
36 | 36 | pass |
|
37 | 37 | |
|
38 | 38 | # Due to the use of demandload py2exe is not finding the modules. |
|
39 | 39 | # packagescan.getmodules creates a list of modules included in |
|
40 | 40 | # the mercurial package plus depdent modules. |
|
41 | 41 | from py2exe.build_exe import py2exe as build_exe |
|
42 | 42 | |
|
43 | 43 | class py2exe_for_demandload(build_exe): |
|
44 | 44 | """ overwrites the py2exe command class for getting the build |
|
45 | 45 | directory and for setting the 'includes' option.""" |
|
46 | 46 | def initialize_options(self): |
|
47 | 47 | self.build_lib = None |
|
48 | 48 | build_exe.initialize_options(self) |
|
49 | 49 | def finalize_options(self): |
|
50 | 50 | # Get the build directory, ie. where to search for modules. |
|
51 | 51 | self.set_undefined_options('build', |
|
52 | 52 | ('build_lib', 'build_lib')) |
|
53 | 53 | # Sets the 'includes' option with the list of needed modules |
|
54 | 54 | if not self.includes: |
|
55 | 55 | self.includes = [] |
|
56 | 56 | else: |
|
57 | 57 | self.includes = self.includes.split(',') |
|
58 | 58 | mercurial.packagescan.scan(self.build_lib, 'mercurial') |
|
59 | 59 | mercurial.packagescan.scan(self.build_lib, 'mercurial.hgweb') |
|
60 | 60 | mercurial.packagescan.scan(self.build_lib, 'hgext') |
|
61 | 61 | self.includes += mercurial.packagescan.getmodules() |
|
62 | 62 | build_exe.finalize_options(self) |
|
63 | 63 | except ImportError: |
|
64 | 64 | py2exe_for_demandload = None |
|
65 | 65 | |
|
66 | 66 | |
|
67 | 67 | # specify version string, otherwise 'hg identify' will be used: |
|
68 | 68 | version = '' |
|
69 | 69 | |
|
70 | 70 | class install_package_data(install_data): |
|
71 | 71 | def finalize_options(self): |
|
72 | 72 | self.set_undefined_options('install', |
|
73 | 73 | ('install_lib', 'install_dir')) |
|
74 | 74 | install_data.finalize_options(self) |
|
75 | 75 | |
|
76 | 76 | mercurial.version.remember_version(version) |
|
77 | 77 | cmdclass = {'install_data': install_package_data} |
|
78 | 78 | py2exe_opts = {} |
|
79 | 79 | if py2exe_for_demandload is not None: |
|
80 | 80 | cmdclass['py2exe'] = py2exe_for_demandload |
|
81 | 81 | py2exe_opts['console'] = ['hg'] |
|
82 | 82 | |
|
83 | 83 | setup(name='mercurial', |
|
84 | 84 | version=mercurial.version.get_version(), |
|
85 | 85 | author='Matt Mackall', |
|
86 | 86 | author_email='mpm@selenic.com', |
|
87 | 87 | url='http://selenic.com/mercurial', |
|
88 | 88 | description='Scalable distributed SCM', |
|
89 | 89 | license='GNU GPL', |
|
90 | 90 | packages=['mercurial', 'mercurial.hgweb', 'hgext'], |
|
91 | 91 | ext_modules=[Extension('mercurial.mpatch', ['mercurial/mpatch.c']), |
|
92 |
Extension('mercurial.bdiff', ['mercurial/bdiff.c']) |
|
|
92 | Extension('mercurial.bdiff', ['mercurial/bdiff.c']), | |
|
93 | Extension('mercurial.base85', ['mercurial/base85.c'])], | |
|
93 | 94 | data_files=[(os.path.join('mercurial', root), |
|
94 | 95 | [os.path.join(root, file_) for file_ in files]) |
|
95 | 96 | for root, dirs, files in os.walk('templates')], |
|
96 | 97 | cmdclass=cmdclass, |
|
97 | 98 | scripts=['hg', 'hgmerge'], |
|
98 | 99 | options=dict(bdist_mpkg=dict(zipdist=True, |
|
99 | 100 | license='COPYING', |
|
100 | 101 | readme='contrib/macosx/Readme.html', |
|
101 | 102 | welcome='contrib/macosx/Welcome.html')), |
|
102 | 103 | **py2exe_opts) |
General Comments 0
You need to be logged in to leave comments.
Login now