##// END OF EJS Templates
pushvars: add a coreconfigitem for push.pushvars.server...
Pulkit Goyal -
r33834:057d31ce default
parent child Browse files
Show More
@@ -1,1894 +1,1894
1 # bundle2.py - generic container format to transmit arbitrary data.
1 # bundle2.py - generic container format to transmit arbitrary data.
2 #
2 #
3 # Copyright 2013 Facebook, Inc.
3 # Copyright 2013 Facebook, Inc.
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7 """Handling of the new bundle2 format
7 """Handling of the new bundle2 format
8
8
9 The goal of bundle2 is to act as an atomically packet to transmit a set of
9 The goal of bundle2 is to act as an atomically packet to transmit a set of
10 payloads in an application agnostic way. It consist in a sequence of "parts"
10 payloads in an application agnostic way. It consist in a sequence of "parts"
11 that will be handed to and processed by the application layer.
11 that will be handed to and processed by the application layer.
12
12
13
13
14 General format architecture
14 General format architecture
15 ===========================
15 ===========================
16
16
17 The format is architectured as follow
17 The format is architectured as follow
18
18
19 - magic string
19 - magic string
20 - stream level parameters
20 - stream level parameters
21 - payload parts (any number)
21 - payload parts (any number)
22 - end of stream marker.
22 - end of stream marker.
23
23
24 the Binary format
24 the Binary format
25 ============================
25 ============================
26
26
27 All numbers are unsigned and big-endian.
27 All numbers are unsigned and big-endian.
28
28
29 stream level parameters
29 stream level parameters
30 ------------------------
30 ------------------------
31
31
32 Binary format is as follow
32 Binary format is as follow
33
33
34 :params size: int32
34 :params size: int32
35
35
36 The total number of Bytes used by the parameters
36 The total number of Bytes used by the parameters
37
37
38 :params value: arbitrary number of Bytes
38 :params value: arbitrary number of Bytes
39
39
40 A blob of `params size` containing the serialized version of all stream level
40 A blob of `params size` containing the serialized version of all stream level
41 parameters.
41 parameters.
42
42
43 The blob contains a space separated list of parameters. Parameters with value
43 The blob contains a space separated list of parameters. Parameters with value
44 are stored in the form `<name>=<value>`. Both name and value are urlquoted.
44 are stored in the form `<name>=<value>`. Both name and value are urlquoted.
45
45
46 Empty name are obviously forbidden.
46 Empty name are obviously forbidden.
47
47
48 Name MUST start with a letter. If this first letter is lower case, the
48 Name MUST start with a letter. If this first letter is lower case, the
49 parameter is advisory and can be safely ignored. However when the first
49 parameter is advisory and can be safely ignored. However when the first
50 letter is capital, the parameter is mandatory and the bundling process MUST
50 letter is capital, the parameter is mandatory and the bundling process MUST
51 stop if he is not able to proceed it.
51 stop if he is not able to proceed it.
52
52
53 Stream parameters use a simple textual format for two main reasons:
53 Stream parameters use a simple textual format for two main reasons:
54
54
55 - Stream level parameters should remain simple and we want to discourage any
55 - Stream level parameters should remain simple and we want to discourage any
56 crazy usage.
56 crazy usage.
57 - Textual data allow easy human inspection of a bundle2 header in case of
57 - Textual data allow easy human inspection of a bundle2 header in case of
58 troubles.
58 troubles.
59
59
60 Any Applicative level options MUST go into a bundle2 part instead.
60 Any Applicative level options MUST go into a bundle2 part instead.
61
61
62 Payload part
62 Payload part
63 ------------------------
63 ------------------------
64
64
65 Binary format is as follow
65 Binary format is as follow
66
66
67 :header size: int32
67 :header size: int32
68
68
69 The total number of Bytes used by the part header. When the header is empty
69 The total number of Bytes used by the part header. When the header is empty
70 (size = 0) this is interpreted as the end of stream marker.
70 (size = 0) this is interpreted as the end of stream marker.
71
71
72 :header:
72 :header:
73
73
74 The header defines how to interpret the part. It contains two piece of
74 The header defines how to interpret the part. It contains two piece of
75 data: the part type, and the part parameters.
75 data: the part type, and the part parameters.
76
76
77 The part type is used to route an application level handler, that can
77 The part type is used to route an application level handler, that can
78 interpret payload.
78 interpret payload.
79
79
80 Part parameters are passed to the application level handler. They are
80 Part parameters are passed to the application level handler. They are
81 meant to convey information that will help the application level object to
81 meant to convey information that will help the application level object to
82 interpret the part payload.
82 interpret the part payload.
83
83
84 The binary format of the header is has follow
84 The binary format of the header is has follow
85
85
86 :typesize: (one byte)
86 :typesize: (one byte)
87
87
88 :parttype: alphanumerical part name (restricted to [a-zA-Z0-9_:-]*)
88 :parttype: alphanumerical part name (restricted to [a-zA-Z0-9_:-]*)
89
89
90 :partid: A 32bits integer (unique in the bundle) that can be used to refer
90 :partid: A 32bits integer (unique in the bundle) that can be used to refer
91 to this part.
91 to this part.
92
92
93 :parameters:
93 :parameters:
94
94
95 Part's parameter may have arbitrary content, the binary structure is::
95 Part's parameter may have arbitrary content, the binary structure is::
96
96
97 <mandatory-count><advisory-count><param-sizes><param-data>
97 <mandatory-count><advisory-count><param-sizes><param-data>
98
98
99 :mandatory-count: 1 byte, number of mandatory parameters
99 :mandatory-count: 1 byte, number of mandatory parameters
100
100
101 :advisory-count: 1 byte, number of advisory parameters
101 :advisory-count: 1 byte, number of advisory parameters
102
102
103 :param-sizes:
103 :param-sizes:
104
104
105 N couple of bytes, where N is the total number of parameters. Each
105 N couple of bytes, where N is the total number of parameters. Each
106 couple contains (<size-of-key>, <size-of-value) for one parameter.
106 couple contains (<size-of-key>, <size-of-value) for one parameter.
107
107
108 :param-data:
108 :param-data:
109
109
110 A blob of bytes from which each parameter key and value can be
110 A blob of bytes from which each parameter key and value can be
111 retrieved using the list of size couples stored in the previous
111 retrieved using the list of size couples stored in the previous
112 field.
112 field.
113
113
114 Mandatory parameters comes first, then the advisory ones.
114 Mandatory parameters comes first, then the advisory ones.
115
115
116 Each parameter's key MUST be unique within the part.
116 Each parameter's key MUST be unique within the part.
117
117
118 :payload:
118 :payload:
119
119
120 payload is a series of `<chunksize><chunkdata>`.
120 payload is a series of `<chunksize><chunkdata>`.
121
121
122 `chunksize` is an int32, `chunkdata` are plain bytes (as much as
122 `chunksize` is an int32, `chunkdata` are plain bytes (as much as
123 `chunksize` says)` The payload part is concluded by a zero size chunk.
123 `chunksize` says)` The payload part is concluded by a zero size chunk.
124
124
125 The current implementation always produces either zero or one chunk.
125 The current implementation always produces either zero or one chunk.
126 This is an implementation limitation that will ultimately be lifted.
126 This is an implementation limitation that will ultimately be lifted.
127
127
128 `chunksize` can be negative to trigger special case processing. No such
128 `chunksize` can be negative to trigger special case processing. No such
129 processing is in place yet.
129 processing is in place yet.
130
130
131 Bundle processing
131 Bundle processing
132 ============================
132 ============================
133
133
134 Each part is processed in order using a "part handler". Handler are registered
134 Each part is processed in order using a "part handler". Handler are registered
135 for a certain part type.
135 for a certain part type.
136
136
137 The matching of a part to its handler is case insensitive. The case of the
137 The matching of a part to its handler is case insensitive. The case of the
138 part type is used to know if a part is mandatory or advisory. If the Part type
138 part type is used to know if a part is mandatory or advisory. If the Part type
139 contains any uppercase char it is considered mandatory. When no handler is
139 contains any uppercase char it is considered mandatory. When no handler is
140 known for a Mandatory part, the process is aborted and an exception is raised.
140 known for a Mandatory part, the process is aborted and an exception is raised.
141 If the part is advisory and no handler is known, the part is ignored. When the
141 If the part is advisory and no handler is known, the part is ignored. When the
142 process is aborted, the full bundle is still read from the stream to keep the
142 process is aborted, the full bundle is still read from the stream to keep the
143 channel usable. But none of the part read from an abort are processed. In the
143 channel usable. But none of the part read from an abort are processed. In the
144 future, dropping the stream may become an option for channel we do not care to
144 future, dropping the stream may become an option for channel we do not care to
145 preserve.
145 preserve.
146 """
146 """
147
147
148 from __future__ import absolute_import, division
148 from __future__ import absolute_import, division
149
149
150 import errno
150 import errno
151 import re
151 import re
152 import string
152 import string
153 import struct
153 import struct
154 import sys
154 import sys
155
155
156 from .i18n import _
156 from .i18n import _
157 from . import (
157 from . import (
158 changegroup,
158 changegroup,
159 error,
159 error,
160 obsolete,
160 obsolete,
161 phases,
161 phases,
162 pushkey,
162 pushkey,
163 pycompat,
163 pycompat,
164 tags,
164 tags,
165 url,
165 url,
166 util,
166 util,
167 )
167 )
168
168
169 urlerr = util.urlerr
169 urlerr = util.urlerr
170 urlreq = util.urlreq
170 urlreq = util.urlreq
171
171
172 _pack = struct.pack
172 _pack = struct.pack
173 _unpack = struct.unpack
173 _unpack = struct.unpack
174
174
175 _fstreamparamsize = '>i'
175 _fstreamparamsize = '>i'
176 _fpartheadersize = '>i'
176 _fpartheadersize = '>i'
177 _fparttypesize = '>B'
177 _fparttypesize = '>B'
178 _fpartid = '>I'
178 _fpartid = '>I'
179 _fpayloadsize = '>i'
179 _fpayloadsize = '>i'
180 _fpartparamcount = '>BB'
180 _fpartparamcount = '>BB'
181
181
182 _fphasesentry = '>i20s'
182 _fphasesentry = '>i20s'
183
183
184 preferedchunksize = 4096
184 preferedchunksize = 4096
185
185
186 _parttypeforbidden = re.compile('[^a-zA-Z0-9_:-]')
186 _parttypeforbidden = re.compile('[^a-zA-Z0-9_:-]')
187
187
188 def outdebug(ui, message):
188 def outdebug(ui, message):
189 """debug regarding output stream (bundling)"""
189 """debug regarding output stream (bundling)"""
190 if ui.configbool('devel', 'bundle2.debug'):
190 if ui.configbool('devel', 'bundle2.debug'):
191 ui.debug('bundle2-output: %s\n' % message)
191 ui.debug('bundle2-output: %s\n' % message)
192
192
193 def indebug(ui, message):
193 def indebug(ui, message):
194 """debug on input stream (unbundling)"""
194 """debug on input stream (unbundling)"""
195 if ui.configbool('devel', 'bundle2.debug'):
195 if ui.configbool('devel', 'bundle2.debug'):
196 ui.debug('bundle2-input: %s\n' % message)
196 ui.debug('bundle2-input: %s\n' % message)
197
197
198 def validateparttype(parttype):
198 def validateparttype(parttype):
199 """raise ValueError if a parttype contains invalid character"""
199 """raise ValueError if a parttype contains invalid character"""
200 if _parttypeforbidden.search(parttype):
200 if _parttypeforbidden.search(parttype):
201 raise ValueError(parttype)
201 raise ValueError(parttype)
202
202
203 def _makefpartparamsizes(nbparams):
203 def _makefpartparamsizes(nbparams):
204 """return a struct format to read part parameter sizes
204 """return a struct format to read part parameter sizes
205
205
206 The number parameters is variable so we need to build that format
206 The number parameters is variable so we need to build that format
207 dynamically.
207 dynamically.
208 """
208 """
209 return '>'+('BB'*nbparams)
209 return '>'+('BB'*nbparams)
210
210
211 parthandlermapping = {}
211 parthandlermapping = {}
212
212
213 def parthandler(parttype, params=()):
213 def parthandler(parttype, params=()):
214 """decorator that register a function as a bundle2 part handler
214 """decorator that register a function as a bundle2 part handler
215
215
216 eg::
216 eg::
217
217
218 @parthandler('myparttype', ('mandatory', 'param', 'handled'))
218 @parthandler('myparttype', ('mandatory', 'param', 'handled'))
219 def myparttypehandler(...):
219 def myparttypehandler(...):
220 '''process a part of type "my part".'''
220 '''process a part of type "my part".'''
221 ...
221 ...
222 """
222 """
223 validateparttype(parttype)
223 validateparttype(parttype)
224 def _decorator(func):
224 def _decorator(func):
225 lparttype = parttype.lower() # enforce lower case matching.
225 lparttype = parttype.lower() # enforce lower case matching.
226 assert lparttype not in parthandlermapping
226 assert lparttype not in parthandlermapping
227 parthandlermapping[lparttype] = func
227 parthandlermapping[lparttype] = func
228 func.params = frozenset(params)
228 func.params = frozenset(params)
229 return func
229 return func
230 return _decorator
230 return _decorator
231
231
232 class unbundlerecords(object):
232 class unbundlerecords(object):
233 """keep record of what happens during and unbundle
233 """keep record of what happens during and unbundle
234
234
235 New records are added using `records.add('cat', obj)`. Where 'cat' is a
235 New records are added using `records.add('cat', obj)`. Where 'cat' is a
236 category of record and obj is an arbitrary object.
236 category of record and obj is an arbitrary object.
237
237
238 `records['cat']` will return all entries of this category 'cat'.
238 `records['cat']` will return all entries of this category 'cat'.
239
239
240 Iterating on the object itself will yield `('category', obj)` tuples
240 Iterating on the object itself will yield `('category', obj)` tuples
241 for all entries.
241 for all entries.
242
242
243 All iterations happens in chronological order.
243 All iterations happens in chronological order.
244 """
244 """
245
245
246 def __init__(self):
246 def __init__(self):
247 self._categories = {}
247 self._categories = {}
248 self._sequences = []
248 self._sequences = []
249 self._replies = {}
249 self._replies = {}
250
250
251 def add(self, category, entry, inreplyto=None):
251 def add(self, category, entry, inreplyto=None):
252 """add a new record of a given category.
252 """add a new record of a given category.
253
253
254 The entry can then be retrieved in the list returned by
254 The entry can then be retrieved in the list returned by
255 self['category']."""
255 self['category']."""
256 self._categories.setdefault(category, []).append(entry)
256 self._categories.setdefault(category, []).append(entry)
257 self._sequences.append((category, entry))
257 self._sequences.append((category, entry))
258 if inreplyto is not None:
258 if inreplyto is not None:
259 self.getreplies(inreplyto).add(category, entry)
259 self.getreplies(inreplyto).add(category, entry)
260
260
261 def getreplies(self, partid):
261 def getreplies(self, partid):
262 """get the records that are replies to a specific part"""
262 """get the records that are replies to a specific part"""
263 return self._replies.setdefault(partid, unbundlerecords())
263 return self._replies.setdefault(partid, unbundlerecords())
264
264
265 def __getitem__(self, cat):
265 def __getitem__(self, cat):
266 return tuple(self._categories.get(cat, ()))
266 return tuple(self._categories.get(cat, ()))
267
267
268 def __iter__(self):
268 def __iter__(self):
269 return iter(self._sequences)
269 return iter(self._sequences)
270
270
271 def __len__(self):
271 def __len__(self):
272 return len(self._sequences)
272 return len(self._sequences)
273
273
274 def __nonzero__(self):
274 def __nonzero__(self):
275 return bool(self._sequences)
275 return bool(self._sequences)
276
276
277 __bool__ = __nonzero__
277 __bool__ = __nonzero__
278
278
279 class bundleoperation(object):
279 class bundleoperation(object):
280 """an object that represents a single bundling process
280 """an object that represents a single bundling process
281
281
282 Its purpose is to carry unbundle-related objects and states.
282 Its purpose is to carry unbundle-related objects and states.
283
283
284 A new object should be created at the beginning of each bundle processing.
284 A new object should be created at the beginning of each bundle processing.
285 The object is to be returned by the processing function.
285 The object is to be returned by the processing function.
286
286
287 The object has very little content now it will ultimately contain:
287 The object has very little content now it will ultimately contain:
288 * an access to the repo the bundle is applied to,
288 * an access to the repo the bundle is applied to,
289 * a ui object,
289 * a ui object,
290 * a way to retrieve a transaction to add changes to the repo,
290 * a way to retrieve a transaction to add changes to the repo,
291 * a way to record the result of processing each part,
291 * a way to record the result of processing each part,
292 * a way to construct a bundle response when applicable.
292 * a way to construct a bundle response when applicable.
293 """
293 """
294
294
295 def __init__(self, repo, transactiongetter, captureoutput=True):
295 def __init__(self, repo, transactiongetter, captureoutput=True):
296 self.repo = repo
296 self.repo = repo
297 self.ui = repo.ui
297 self.ui = repo.ui
298 self.records = unbundlerecords()
298 self.records = unbundlerecords()
299 self.reply = None
299 self.reply = None
300 self.captureoutput = captureoutput
300 self.captureoutput = captureoutput
301 self.hookargs = {}
301 self.hookargs = {}
302 self._gettransaction = transactiongetter
302 self._gettransaction = transactiongetter
303
303
304 def gettransaction(self):
304 def gettransaction(self):
305 transaction = self._gettransaction()
305 transaction = self._gettransaction()
306
306
307 if self.hookargs:
307 if self.hookargs:
308 # the ones added to the transaction supercede those added
308 # the ones added to the transaction supercede those added
309 # to the operation.
309 # to the operation.
310 self.hookargs.update(transaction.hookargs)
310 self.hookargs.update(transaction.hookargs)
311 transaction.hookargs = self.hookargs
311 transaction.hookargs = self.hookargs
312
312
313 # mark the hookargs as flushed. further attempts to add to
313 # mark the hookargs as flushed. further attempts to add to
314 # hookargs will result in an abort.
314 # hookargs will result in an abort.
315 self.hookargs = None
315 self.hookargs = None
316
316
317 return transaction
317 return transaction
318
318
319 def addhookargs(self, hookargs):
319 def addhookargs(self, hookargs):
320 if self.hookargs is None:
320 if self.hookargs is None:
321 raise error.ProgrammingError('attempted to add hookargs to '
321 raise error.ProgrammingError('attempted to add hookargs to '
322 'operation after transaction started')
322 'operation after transaction started')
323 self.hookargs.update(hookargs)
323 self.hookargs.update(hookargs)
324
324
325 class TransactionUnavailable(RuntimeError):
325 class TransactionUnavailable(RuntimeError):
326 pass
326 pass
327
327
328 def _notransaction():
328 def _notransaction():
329 """default method to get a transaction while processing a bundle
329 """default method to get a transaction while processing a bundle
330
330
331 Raise an exception to highlight the fact that no transaction was expected
331 Raise an exception to highlight the fact that no transaction was expected
332 to be created"""
332 to be created"""
333 raise TransactionUnavailable()
333 raise TransactionUnavailable()
334
334
335 def applybundle(repo, unbundler, tr, source=None, url=None, **kwargs):
335 def applybundle(repo, unbundler, tr, source=None, url=None, **kwargs):
336 # transform me into unbundler.apply() as soon as the freeze is lifted
336 # transform me into unbundler.apply() as soon as the freeze is lifted
337 if isinstance(unbundler, unbundle20):
337 if isinstance(unbundler, unbundle20):
338 tr.hookargs['bundle2'] = '1'
338 tr.hookargs['bundle2'] = '1'
339 if source is not None and 'source' not in tr.hookargs:
339 if source is not None and 'source' not in tr.hookargs:
340 tr.hookargs['source'] = source
340 tr.hookargs['source'] = source
341 if url is not None and 'url' not in tr.hookargs:
341 if url is not None and 'url' not in tr.hookargs:
342 tr.hookargs['url'] = url
342 tr.hookargs['url'] = url
343 return processbundle(repo, unbundler, lambda: tr)
343 return processbundle(repo, unbundler, lambda: tr)
344 else:
344 else:
345 # the transactiongetter won't be used, but we might as well set it
345 # the transactiongetter won't be used, but we might as well set it
346 op = bundleoperation(repo, lambda: tr)
346 op = bundleoperation(repo, lambda: tr)
347 _processchangegroup(op, unbundler, tr, source, url, **kwargs)
347 _processchangegroup(op, unbundler, tr, source, url, **kwargs)
348 return op
348 return op
349
349
350 def processbundle(repo, unbundler, transactiongetter=None, op=None):
350 def processbundle(repo, unbundler, transactiongetter=None, op=None):
351 """This function process a bundle, apply effect to/from a repo
351 """This function process a bundle, apply effect to/from a repo
352
352
353 It iterates over each part then searches for and uses the proper handling
353 It iterates over each part then searches for and uses the proper handling
354 code to process the part. Parts are processed in order.
354 code to process the part. Parts are processed in order.
355
355
356 Unknown Mandatory part will abort the process.
356 Unknown Mandatory part will abort the process.
357
357
358 It is temporarily possible to provide a prebuilt bundleoperation to the
358 It is temporarily possible to provide a prebuilt bundleoperation to the
359 function. This is used to ensure output is properly propagated in case of
359 function. This is used to ensure output is properly propagated in case of
360 an error during the unbundling. This output capturing part will likely be
360 an error during the unbundling. This output capturing part will likely be
361 reworked and this ability will probably go away in the process.
361 reworked and this ability will probably go away in the process.
362 """
362 """
363 if op is None:
363 if op is None:
364 if transactiongetter is None:
364 if transactiongetter is None:
365 transactiongetter = _notransaction
365 transactiongetter = _notransaction
366 op = bundleoperation(repo, transactiongetter)
366 op = bundleoperation(repo, transactiongetter)
367 # todo:
367 # todo:
368 # - replace this is a init function soon.
368 # - replace this is a init function soon.
369 # - exception catching
369 # - exception catching
370 unbundler.params
370 unbundler.params
371 if repo.ui.debugflag:
371 if repo.ui.debugflag:
372 msg = ['bundle2-input-bundle:']
372 msg = ['bundle2-input-bundle:']
373 if unbundler.params:
373 if unbundler.params:
374 msg.append(' %i params' % len(unbundler.params))
374 msg.append(' %i params' % len(unbundler.params))
375 if op._gettransaction is None or op._gettransaction is _notransaction:
375 if op._gettransaction is None or op._gettransaction is _notransaction:
376 msg.append(' no-transaction')
376 msg.append(' no-transaction')
377 else:
377 else:
378 msg.append(' with-transaction')
378 msg.append(' with-transaction')
379 msg.append('\n')
379 msg.append('\n')
380 repo.ui.debug(''.join(msg))
380 repo.ui.debug(''.join(msg))
381 iterparts = enumerate(unbundler.iterparts())
381 iterparts = enumerate(unbundler.iterparts())
382 part = None
382 part = None
383 nbpart = 0
383 nbpart = 0
384 try:
384 try:
385 for nbpart, part in iterparts:
385 for nbpart, part in iterparts:
386 _processpart(op, part)
386 _processpart(op, part)
387 except Exception as exc:
387 except Exception as exc:
388 # Any exceptions seeking to the end of the bundle at this point are
388 # Any exceptions seeking to the end of the bundle at this point are
389 # almost certainly related to the underlying stream being bad.
389 # almost certainly related to the underlying stream being bad.
390 # And, chances are that the exception we're handling is related to
390 # And, chances are that the exception we're handling is related to
391 # getting in that bad state. So, we swallow the seeking error and
391 # getting in that bad state. So, we swallow the seeking error and
392 # re-raise the original error.
392 # re-raise the original error.
393 seekerror = False
393 seekerror = False
394 try:
394 try:
395 for nbpart, part in iterparts:
395 for nbpart, part in iterparts:
396 # consume the bundle content
396 # consume the bundle content
397 part.seek(0, 2)
397 part.seek(0, 2)
398 except Exception:
398 except Exception:
399 seekerror = True
399 seekerror = True
400
400
401 # Small hack to let caller code distinguish exceptions from bundle2
401 # Small hack to let caller code distinguish exceptions from bundle2
402 # processing from processing the old format. This is mostly
402 # processing from processing the old format. This is mostly
403 # needed to handle different return codes to unbundle according to the
403 # needed to handle different return codes to unbundle according to the
404 # type of bundle. We should probably clean up or drop this return code
404 # type of bundle. We should probably clean up or drop this return code
405 # craziness in a future version.
405 # craziness in a future version.
406 exc.duringunbundle2 = True
406 exc.duringunbundle2 = True
407 salvaged = []
407 salvaged = []
408 replycaps = None
408 replycaps = None
409 if op.reply is not None:
409 if op.reply is not None:
410 salvaged = op.reply.salvageoutput()
410 salvaged = op.reply.salvageoutput()
411 replycaps = op.reply.capabilities
411 replycaps = op.reply.capabilities
412 exc._replycaps = replycaps
412 exc._replycaps = replycaps
413 exc._bundle2salvagedoutput = salvaged
413 exc._bundle2salvagedoutput = salvaged
414
414
415 # Re-raising from a variable loses the original stack. So only use
415 # Re-raising from a variable loses the original stack. So only use
416 # that form if we need to.
416 # that form if we need to.
417 if seekerror:
417 if seekerror:
418 raise exc
418 raise exc
419 else:
419 else:
420 raise
420 raise
421 finally:
421 finally:
422 repo.ui.debug('bundle2-input-bundle: %i parts total\n' % nbpart)
422 repo.ui.debug('bundle2-input-bundle: %i parts total\n' % nbpart)
423
423
424 return op
424 return op
425
425
426 def _processchangegroup(op, cg, tr, source, url, **kwargs):
426 def _processchangegroup(op, cg, tr, source, url, **kwargs):
427 ret = cg.apply(op.repo, tr, source, url, **kwargs)
427 ret = cg.apply(op.repo, tr, source, url, **kwargs)
428 op.records.add('changegroup', {
428 op.records.add('changegroup', {
429 'return': ret,
429 'return': ret,
430 })
430 })
431 return ret
431 return ret
432
432
433 def _processpart(op, part):
433 def _processpart(op, part):
434 """process a single part from a bundle
434 """process a single part from a bundle
435
435
436 The part is guaranteed to have been fully consumed when the function exits
436 The part is guaranteed to have been fully consumed when the function exits
437 (even if an exception is raised)."""
437 (even if an exception is raised)."""
438 status = 'unknown' # used by debug output
438 status = 'unknown' # used by debug output
439 hardabort = False
439 hardabort = False
440 try:
440 try:
441 try:
441 try:
442 handler = parthandlermapping.get(part.type)
442 handler = parthandlermapping.get(part.type)
443 if handler is None:
443 if handler is None:
444 status = 'unsupported-type'
444 status = 'unsupported-type'
445 raise error.BundleUnknownFeatureError(parttype=part.type)
445 raise error.BundleUnknownFeatureError(parttype=part.type)
446 indebug(op.ui, 'found a handler for part %r' % part.type)
446 indebug(op.ui, 'found a handler for part %r' % part.type)
447 unknownparams = part.mandatorykeys - handler.params
447 unknownparams = part.mandatorykeys - handler.params
448 if unknownparams:
448 if unknownparams:
449 unknownparams = list(unknownparams)
449 unknownparams = list(unknownparams)
450 unknownparams.sort()
450 unknownparams.sort()
451 status = 'unsupported-params (%s)' % unknownparams
451 status = 'unsupported-params (%s)' % unknownparams
452 raise error.BundleUnknownFeatureError(parttype=part.type,
452 raise error.BundleUnknownFeatureError(parttype=part.type,
453 params=unknownparams)
453 params=unknownparams)
454 status = 'supported'
454 status = 'supported'
455 except error.BundleUnknownFeatureError as exc:
455 except error.BundleUnknownFeatureError as exc:
456 if part.mandatory: # mandatory parts
456 if part.mandatory: # mandatory parts
457 raise
457 raise
458 indebug(op.ui, 'ignoring unsupported advisory part %s' % exc)
458 indebug(op.ui, 'ignoring unsupported advisory part %s' % exc)
459 return # skip to part processing
459 return # skip to part processing
460 finally:
460 finally:
461 if op.ui.debugflag:
461 if op.ui.debugflag:
462 msg = ['bundle2-input-part: "%s"' % part.type]
462 msg = ['bundle2-input-part: "%s"' % part.type]
463 if not part.mandatory:
463 if not part.mandatory:
464 msg.append(' (advisory)')
464 msg.append(' (advisory)')
465 nbmp = len(part.mandatorykeys)
465 nbmp = len(part.mandatorykeys)
466 nbap = len(part.params) - nbmp
466 nbap = len(part.params) - nbmp
467 if nbmp or nbap:
467 if nbmp or nbap:
468 msg.append(' (params:')
468 msg.append(' (params:')
469 if nbmp:
469 if nbmp:
470 msg.append(' %i mandatory' % nbmp)
470 msg.append(' %i mandatory' % nbmp)
471 if nbap:
471 if nbap:
472 msg.append(' %i advisory' % nbmp)
472 msg.append(' %i advisory' % nbmp)
473 msg.append(')')
473 msg.append(')')
474 msg.append(' %s\n' % status)
474 msg.append(' %s\n' % status)
475 op.ui.debug(''.join(msg))
475 op.ui.debug(''.join(msg))
476
476
477 # handler is called outside the above try block so that we don't
477 # handler is called outside the above try block so that we don't
478 # risk catching KeyErrors from anything other than the
478 # risk catching KeyErrors from anything other than the
479 # parthandlermapping lookup (any KeyError raised by handler()
479 # parthandlermapping lookup (any KeyError raised by handler()
480 # itself represents a defect of a different variety).
480 # itself represents a defect of a different variety).
481 output = None
481 output = None
482 if op.captureoutput and op.reply is not None:
482 if op.captureoutput and op.reply is not None:
483 op.ui.pushbuffer(error=True, subproc=True)
483 op.ui.pushbuffer(error=True, subproc=True)
484 output = ''
484 output = ''
485 try:
485 try:
486 handler(op, part)
486 handler(op, part)
487 finally:
487 finally:
488 if output is not None:
488 if output is not None:
489 output = op.ui.popbuffer()
489 output = op.ui.popbuffer()
490 if output:
490 if output:
491 outpart = op.reply.newpart('output', data=output,
491 outpart = op.reply.newpart('output', data=output,
492 mandatory=False)
492 mandatory=False)
493 outpart.addparam(
493 outpart.addparam(
494 'in-reply-to', pycompat.bytestr(part.id), mandatory=False)
494 'in-reply-to', pycompat.bytestr(part.id), mandatory=False)
495 # If exiting or interrupted, do not attempt to seek the stream in the
495 # If exiting or interrupted, do not attempt to seek the stream in the
496 # finally block below. This makes abort faster.
496 # finally block below. This makes abort faster.
497 except (SystemExit, KeyboardInterrupt):
497 except (SystemExit, KeyboardInterrupt):
498 hardabort = True
498 hardabort = True
499 raise
499 raise
500 finally:
500 finally:
501 # consume the part content to not corrupt the stream.
501 # consume the part content to not corrupt the stream.
502 if not hardabort:
502 if not hardabort:
503 part.seek(0, 2)
503 part.seek(0, 2)
504
504
505
505
506 def decodecaps(blob):
506 def decodecaps(blob):
507 """decode a bundle2 caps bytes blob into a dictionary
507 """decode a bundle2 caps bytes blob into a dictionary
508
508
509 The blob is a list of capabilities (one per line)
509 The blob is a list of capabilities (one per line)
510 Capabilities may have values using a line of the form::
510 Capabilities may have values using a line of the form::
511
511
512 capability=value1,value2,value3
512 capability=value1,value2,value3
513
513
514 The values are always a list."""
514 The values are always a list."""
515 caps = {}
515 caps = {}
516 for line in blob.splitlines():
516 for line in blob.splitlines():
517 if not line:
517 if not line:
518 continue
518 continue
519 if '=' not in line:
519 if '=' not in line:
520 key, vals = line, ()
520 key, vals = line, ()
521 else:
521 else:
522 key, vals = line.split('=', 1)
522 key, vals = line.split('=', 1)
523 vals = vals.split(',')
523 vals = vals.split(',')
524 key = urlreq.unquote(key)
524 key = urlreq.unquote(key)
525 vals = [urlreq.unquote(v) for v in vals]
525 vals = [urlreq.unquote(v) for v in vals]
526 caps[key] = vals
526 caps[key] = vals
527 return caps
527 return caps
528
528
529 def encodecaps(caps):
529 def encodecaps(caps):
530 """encode a bundle2 caps dictionary into a bytes blob"""
530 """encode a bundle2 caps dictionary into a bytes blob"""
531 chunks = []
531 chunks = []
532 for ca in sorted(caps):
532 for ca in sorted(caps):
533 vals = caps[ca]
533 vals = caps[ca]
534 ca = urlreq.quote(ca)
534 ca = urlreq.quote(ca)
535 vals = [urlreq.quote(v) for v in vals]
535 vals = [urlreq.quote(v) for v in vals]
536 if vals:
536 if vals:
537 ca = "%s=%s" % (ca, ','.join(vals))
537 ca = "%s=%s" % (ca, ','.join(vals))
538 chunks.append(ca)
538 chunks.append(ca)
539 return '\n'.join(chunks)
539 return '\n'.join(chunks)
540
540
541 bundletypes = {
541 bundletypes = {
542 "": ("", 'UN'), # only when using unbundle on ssh and old http servers
542 "": ("", 'UN'), # only when using unbundle on ssh and old http servers
543 # since the unification ssh accepts a header but there
543 # since the unification ssh accepts a header but there
544 # is no capability signaling it.
544 # is no capability signaling it.
545 "HG20": (), # special-cased below
545 "HG20": (), # special-cased below
546 "HG10UN": ("HG10UN", 'UN'),
546 "HG10UN": ("HG10UN", 'UN'),
547 "HG10BZ": ("HG10", 'BZ'),
547 "HG10BZ": ("HG10", 'BZ'),
548 "HG10GZ": ("HG10GZ", 'GZ'),
548 "HG10GZ": ("HG10GZ", 'GZ'),
549 }
549 }
550
550
551 # hgweb uses this list to communicate its preferred type
551 # hgweb uses this list to communicate its preferred type
552 bundlepriority = ['HG10GZ', 'HG10BZ', 'HG10UN']
552 bundlepriority = ['HG10GZ', 'HG10BZ', 'HG10UN']
553
553
554 class bundle20(object):
554 class bundle20(object):
555 """represent an outgoing bundle2 container
555 """represent an outgoing bundle2 container
556
556
557 Use the `addparam` method to add stream level parameter. and `newpart` to
557 Use the `addparam` method to add stream level parameter. and `newpart` to
558 populate it. Then call `getchunks` to retrieve all the binary chunks of
558 populate it. Then call `getchunks` to retrieve all the binary chunks of
559 data that compose the bundle2 container."""
559 data that compose the bundle2 container."""
560
560
561 _magicstring = 'HG20'
561 _magicstring = 'HG20'
562
562
563 def __init__(self, ui, capabilities=()):
563 def __init__(self, ui, capabilities=()):
564 self.ui = ui
564 self.ui = ui
565 self._params = []
565 self._params = []
566 self._parts = []
566 self._parts = []
567 self.capabilities = dict(capabilities)
567 self.capabilities = dict(capabilities)
568 self._compengine = util.compengines.forbundletype('UN')
568 self._compengine = util.compengines.forbundletype('UN')
569 self._compopts = None
569 self._compopts = None
570
570
571 def setcompression(self, alg, compopts=None):
571 def setcompression(self, alg, compopts=None):
572 """setup core part compression to <alg>"""
572 """setup core part compression to <alg>"""
573 if alg in (None, 'UN'):
573 if alg in (None, 'UN'):
574 return
574 return
575 assert not any(n.lower() == 'compression' for n, v in self._params)
575 assert not any(n.lower() == 'compression' for n, v in self._params)
576 self.addparam('Compression', alg)
576 self.addparam('Compression', alg)
577 self._compengine = util.compengines.forbundletype(alg)
577 self._compengine = util.compengines.forbundletype(alg)
578 self._compopts = compopts
578 self._compopts = compopts
579
579
580 @property
580 @property
581 def nbparts(self):
581 def nbparts(self):
582 """total number of parts added to the bundler"""
582 """total number of parts added to the bundler"""
583 return len(self._parts)
583 return len(self._parts)
584
584
585 # methods used to defines the bundle2 content
585 # methods used to defines the bundle2 content
586 def addparam(self, name, value=None):
586 def addparam(self, name, value=None):
587 """add a stream level parameter"""
587 """add a stream level parameter"""
588 if not name:
588 if not name:
589 raise ValueError('empty parameter name')
589 raise ValueError('empty parameter name')
590 if name[0] not in string.letters:
590 if name[0] not in string.letters:
591 raise ValueError('non letter first character: %r' % name)
591 raise ValueError('non letter first character: %r' % name)
592 self._params.append((name, value))
592 self._params.append((name, value))
593
593
594 def addpart(self, part):
594 def addpart(self, part):
595 """add a new part to the bundle2 container
595 """add a new part to the bundle2 container
596
596
597 Parts contains the actual applicative payload."""
597 Parts contains the actual applicative payload."""
598 assert part.id is None
598 assert part.id is None
599 part.id = len(self._parts) # very cheap counter
599 part.id = len(self._parts) # very cheap counter
600 self._parts.append(part)
600 self._parts.append(part)
601
601
602 def newpart(self, typeid, *args, **kwargs):
602 def newpart(self, typeid, *args, **kwargs):
603 """create a new part and add it to the containers
603 """create a new part and add it to the containers
604
604
605 As the part is directly added to the containers. For now, this means
605 As the part is directly added to the containers. For now, this means
606 that any failure to properly initialize the part after calling
606 that any failure to properly initialize the part after calling
607 ``newpart`` should result in a failure of the whole bundling process.
607 ``newpart`` should result in a failure of the whole bundling process.
608
608
609 You can still fall back to manually create and add if you need better
609 You can still fall back to manually create and add if you need better
610 control."""
610 control."""
611 part = bundlepart(typeid, *args, **kwargs)
611 part = bundlepart(typeid, *args, **kwargs)
612 self.addpart(part)
612 self.addpart(part)
613 return part
613 return part
614
614
615 # methods used to generate the bundle2 stream
615 # methods used to generate the bundle2 stream
616 def getchunks(self):
616 def getchunks(self):
617 if self.ui.debugflag:
617 if self.ui.debugflag:
618 msg = ['bundle2-output-bundle: "%s",' % self._magicstring]
618 msg = ['bundle2-output-bundle: "%s",' % self._magicstring]
619 if self._params:
619 if self._params:
620 msg.append(' (%i params)' % len(self._params))
620 msg.append(' (%i params)' % len(self._params))
621 msg.append(' %i parts total\n' % len(self._parts))
621 msg.append(' %i parts total\n' % len(self._parts))
622 self.ui.debug(''.join(msg))
622 self.ui.debug(''.join(msg))
623 outdebug(self.ui, 'start emission of %s stream' % self._magicstring)
623 outdebug(self.ui, 'start emission of %s stream' % self._magicstring)
624 yield self._magicstring
624 yield self._magicstring
625 param = self._paramchunk()
625 param = self._paramchunk()
626 outdebug(self.ui, 'bundle parameter: %s' % param)
626 outdebug(self.ui, 'bundle parameter: %s' % param)
627 yield _pack(_fstreamparamsize, len(param))
627 yield _pack(_fstreamparamsize, len(param))
628 if param:
628 if param:
629 yield param
629 yield param
630 for chunk in self._compengine.compressstream(self._getcorechunk(),
630 for chunk in self._compengine.compressstream(self._getcorechunk(),
631 self._compopts):
631 self._compopts):
632 yield chunk
632 yield chunk
633
633
634 def _paramchunk(self):
634 def _paramchunk(self):
635 """return a encoded version of all stream parameters"""
635 """return a encoded version of all stream parameters"""
636 blocks = []
636 blocks = []
637 for par, value in self._params:
637 for par, value in self._params:
638 par = urlreq.quote(par)
638 par = urlreq.quote(par)
639 if value is not None:
639 if value is not None:
640 value = urlreq.quote(value)
640 value = urlreq.quote(value)
641 par = '%s=%s' % (par, value)
641 par = '%s=%s' % (par, value)
642 blocks.append(par)
642 blocks.append(par)
643 return ' '.join(blocks)
643 return ' '.join(blocks)
644
644
645 def _getcorechunk(self):
645 def _getcorechunk(self):
646 """yield chunk for the core part of the bundle
646 """yield chunk for the core part of the bundle
647
647
648 (all but headers and parameters)"""
648 (all but headers and parameters)"""
649 outdebug(self.ui, 'start of parts')
649 outdebug(self.ui, 'start of parts')
650 for part in self._parts:
650 for part in self._parts:
651 outdebug(self.ui, 'bundle part: "%s"' % part.type)
651 outdebug(self.ui, 'bundle part: "%s"' % part.type)
652 for chunk in part.getchunks(ui=self.ui):
652 for chunk in part.getchunks(ui=self.ui):
653 yield chunk
653 yield chunk
654 outdebug(self.ui, 'end of bundle')
654 outdebug(self.ui, 'end of bundle')
655 yield _pack(_fpartheadersize, 0)
655 yield _pack(_fpartheadersize, 0)
656
656
657
657
658 def salvageoutput(self):
658 def salvageoutput(self):
659 """return a list with a copy of all output parts in the bundle
659 """return a list with a copy of all output parts in the bundle
660
660
661 This is meant to be used during error handling to make sure we preserve
661 This is meant to be used during error handling to make sure we preserve
662 server output"""
662 server output"""
663 salvaged = []
663 salvaged = []
664 for part in self._parts:
664 for part in self._parts:
665 if part.type.startswith('output'):
665 if part.type.startswith('output'):
666 salvaged.append(part.copy())
666 salvaged.append(part.copy())
667 return salvaged
667 return salvaged
668
668
669
669
670 class unpackermixin(object):
670 class unpackermixin(object):
671 """A mixin to extract bytes and struct data from a stream"""
671 """A mixin to extract bytes and struct data from a stream"""
672
672
673 def __init__(self, fp):
673 def __init__(self, fp):
674 self._fp = fp
674 self._fp = fp
675
675
676 def _unpack(self, format):
676 def _unpack(self, format):
677 """unpack this struct format from the stream
677 """unpack this struct format from the stream
678
678
679 This method is meant for internal usage by the bundle2 protocol only.
679 This method is meant for internal usage by the bundle2 protocol only.
680 They directly manipulate the low level stream including bundle2 level
680 They directly manipulate the low level stream including bundle2 level
681 instruction.
681 instruction.
682
682
683 Do not use it to implement higher-level logic or methods."""
683 Do not use it to implement higher-level logic or methods."""
684 data = self._readexact(struct.calcsize(format))
684 data = self._readexact(struct.calcsize(format))
685 return _unpack(format, data)
685 return _unpack(format, data)
686
686
687 def _readexact(self, size):
687 def _readexact(self, size):
688 """read exactly <size> bytes from the stream
688 """read exactly <size> bytes from the stream
689
689
690 This method is meant for internal usage by the bundle2 protocol only.
690 This method is meant for internal usage by the bundle2 protocol only.
691 They directly manipulate the low level stream including bundle2 level
691 They directly manipulate the low level stream including bundle2 level
692 instruction.
692 instruction.
693
693
694 Do not use it to implement higher-level logic or methods."""
694 Do not use it to implement higher-level logic or methods."""
695 return changegroup.readexactly(self._fp, size)
695 return changegroup.readexactly(self._fp, size)
696
696
697 def getunbundler(ui, fp, magicstring=None):
697 def getunbundler(ui, fp, magicstring=None):
698 """return a valid unbundler object for a given magicstring"""
698 """return a valid unbundler object for a given magicstring"""
699 if magicstring is None:
699 if magicstring is None:
700 magicstring = changegroup.readexactly(fp, 4)
700 magicstring = changegroup.readexactly(fp, 4)
701 magic, version = magicstring[0:2], magicstring[2:4]
701 magic, version = magicstring[0:2], magicstring[2:4]
702 if magic != 'HG':
702 if magic != 'HG':
703 ui.debug(
703 ui.debug(
704 "error: invalid magic: %r (version %r), should be 'HG'\n"
704 "error: invalid magic: %r (version %r), should be 'HG'\n"
705 % (magic, version))
705 % (magic, version))
706 raise error.Abort(_('not a Mercurial bundle'))
706 raise error.Abort(_('not a Mercurial bundle'))
707 unbundlerclass = formatmap.get(version)
707 unbundlerclass = formatmap.get(version)
708 if unbundlerclass is None:
708 if unbundlerclass is None:
709 raise error.Abort(_('unknown bundle version %s') % version)
709 raise error.Abort(_('unknown bundle version %s') % version)
710 unbundler = unbundlerclass(ui, fp)
710 unbundler = unbundlerclass(ui, fp)
711 indebug(ui, 'start processing of %s stream' % magicstring)
711 indebug(ui, 'start processing of %s stream' % magicstring)
712 return unbundler
712 return unbundler
713
713
714 class unbundle20(unpackermixin):
714 class unbundle20(unpackermixin):
715 """interpret a bundle2 stream
715 """interpret a bundle2 stream
716
716
717 This class is fed with a binary stream and yields parts through its
717 This class is fed with a binary stream and yields parts through its
718 `iterparts` methods."""
718 `iterparts` methods."""
719
719
720 _magicstring = 'HG20'
720 _magicstring = 'HG20'
721
721
722 def __init__(self, ui, fp):
722 def __init__(self, ui, fp):
723 """If header is specified, we do not read it out of the stream."""
723 """If header is specified, we do not read it out of the stream."""
724 self.ui = ui
724 self.ui = ui
725 self._compengine = util.compengines.forbundletype('UN')
725 self._compengine = util.compengines.forbundletype('UN')
726 self._compressed = None
726 self._compressed = None
727 super(unbundle20, self).__init__(fp)
727 super(unbundle20, self).__init__(fp)
728
728
729 @util.propertycache
729 @util.propertycache
730 def params(self):
730 def params(self):
731 """dictionary of stream level parameters"""
731 """dictionary of stream level parameters"""
732 indebug(self.ui, 'reading bundle2 stream parameters')
732 indebug(self.ui, 'reading bundle2 stream parameters')
733 params = {}
733 params = {}
734 paramssize = self._unpack(_fstreamparamsize)[0]
734 paramssize = self._unpack(_fstreamparamsize)[0]
735 if paramssize < 0:
735 if paramssize < 0:
736 raise error.BundleValueError('negative bundle param size: %i'
736 raise error.BundleValueError('negative bundle param size: %i'
737 % paramssize)
737 % paramssize)
738 if paramssize:
738 if paramssize:
739 params = self._readexact(paramssize)
739 params = self._readexact(paramssize)
740 params = self._processallparams(params)
740 params = self._processallparams(params)
741 return params
741 return params
742
742
743 def _processallparams(self, paramsblock):
743 def _processallparams(self, paramsblock):
744 """"""
744 """"""
745 params = util.sortdict()
745 params = util.sortdict()
746 for p in paramsblock.split(' '):
746 for p in paramsblock.split(' '):
747 p = p.split('=', 1)
747 p = p.split('=', 1)
748 p = [urlreq.unquote(i) for i in p]
748 p = [urlreq.unquote(i) for i in p]
749 if len(p) < 2:
749 if len(p) < 2:
750 p.append(None)
750 p.append(None)
751 self._processparam(*p)
751 self._processparam(*p)
752 params[p[0]] = p[1]
752 params[p[0]] = p[1]
753 return params
753 return params
754
754
755
755
756 def _processparam(self, name, value):
756 def _processparam(self, name, value):
757 """process a parameter, applying its effect if needed
757 """process a parameter, applying its effect if needed
758
758
759 Parameter starting with a lower case letter are advisory and will be
759 Parameter starting with a lower case letter are advisory and will be
760 ignored when unknown. Those starting with an upper case letter are
760 ignored when unknown. Those starting with an upper case letter are
761 mandatory and will this function will raise a KeyError when unknown.
761 mandatory and will this function will raise a KeyError when unknown.
762
762
763 Note: no option are currently supported. Any input will be either
763 Note: no option are currently supported. Any input will be either
764 ignored or failing.
764 ignored or failing.
765 """
765 """
766 if not name:
766 if not name:
767 raise ValueError('empty parameter name')
767 raise ValueError('empty parameter name')
768 if name[0] not in string.letters:
768 if name[0] not in string.letters:
769 raise ValueError('non letter first character: %r' % name)
769 raise ValueError('non letter first character: %r' % name)
770 try:
770 try:
771 handler = b2streamparamsmap[name.lower()]
771 handler = b2streamparamsmap[name.lower()]
772 except KeyError:
772 except KeyError:
773 if name[0].islower():
773 if name[0].islower():
774 indebug(self.ui, "ignoring unknown parameter %r" % name)
774 indebug(self.ui, "ignoring unknown parameter %r" % name)
775 else:
775 else:
776 raise error.BundleUnknownFeatureError(params=(name,))
776 raise error.BundleUnknownFeatureError(params=(name,))
777 else:
777 else:
778 handler(self, name, value)
778 handler(self, name, value)
779
779
780 def _forwardchunks(self):
780 def _forwardchunks(self):
781 """utility to transfer a bundle2 as binary
781 """utility to transfer a bundle2 as binary
782
782
783 This is made necessary by the fact the 'getbundle' command over 'ssh'
783 This is made necessary by the fact the 'getbundle' command over 'ssh'
784 have no way to know then the reply end, relying on the bundle to be
784 have no way to know then the reply end, relying on the bundle to be
785 interpreted to know its end. This is terrible and we are sorry, but we
785 interpreted to know its end. This is terrible and we are sorry, but we
786 needed to move forward to get general delta enabled.
786 needed to move forward to get general delta enabled.
787 """
787 """
788 yield self._magicstring
788 yield self._magicstring
789 assert 'params' not in vars(self)
789 assert 'params' not in vars(self)
790 paramssize = self._unpack(_fstreamparamsize)[0]
790 paramssize = self._unpack(_fstreamparamsize)[0]
791 if paramssize < 0:
791 if paramssize < 0:
792 raise error.BundleValueError('negative bundle param size: %i'
792 raise error.BundleValueError('negative bundle param size: %i'
793 % paramssize)
793 % paramssize)
794 yield _pack(_fstreamparamsize, paramssize)
794 yield _pack(_fstreamparamsize, paramssize)
795 if paramssize:
795 if paramssize:
796 params = self._readexact(paramssize)
796 params = self._readexact(paramssize)
797 self._processallparams(params)
797 self._processallparams(params)
798 yield params
798 yield params
799 assert self._compengine.bundletype == 'UN'
799 assert self._compengine.bundletype == 'UN'
800 # From there, payload might need to be decompressed
800 # From there, payload might need to be decompressed
801 self._fp = self._compengine.decompressorreader(self._fp)
801 self._fp = self._compengine.decompressorreader(self._fp)
802 emptycount = 0
802 emptycount = 0
803 while emptycount < 2:
803 while emptycount < 2:
804 # so we can brainlessly loop
804 # so we can brainlessly loop
805 assert _fpartheadersize == _fpayloadsize
805 assert _fpartheadersize == _fpayloadsize
806 size = self._unpack(_fpartheadersize)[0]
806 size = self._unpack(_fpartheadersize)[0]
807 yield _pack(_fpartheadersize, size)
807 yield _pack(_fpartheadersize, size)
808 if size:
808 if size:
809 emptycount = 0
809 emptycount = 0
810 else:
810 else:
811 emptycount += 1
811 emptycount += 1
812 continue
812 continue
813 if size == flaginterrupt:
813 if size == flaginterrupt:
814 continue
814 continue
815 elif size < 0:
815 elif size < 0:
816 raise error.BundleValueError('negative chunk size: %i')
816 raise error.BundleValueError('negative chunk size: %i')
817 yield self._readexact(size)
817 yield self._readexact(size)
818
818
819
819
820 def iterparts(self):
820 def iterparts(self):
821 """yield all parts contained in the stream"""
821 """yield all parts contained in the stream"""
822 # make sure param have been loaded
822 # make sure param have been loaded
823 self.params
823 self.params
824 # From there, payload need to be decompressed
824 # From there, payload need to be decompressed
825 self._fp = self._compengine.decompressorreader(self._fp)
825 self._fp = self._compengine.decompressorreader(self._fp)
826 indebug(self.ui, 'start extraction of bundle2 parts')
826 indebug(self.ui, 'start extraction of bundle2 parts')
827 headerblock = self._readpartheader()
827 headerblock = self._readpartheader()
828 while headerblock is not None:
828 while headerblock is not None:
829 part = unbundlepart(self.ui, headerblock, self._fp)
829 part = unbundlepart(self.ui, headerblock, self._fp)
830 yield part
830 yield part
831 part.seek(0, 2)
831 part.seek(0, 2)
832 headerblock = self._readpartheader()
832 headerblock = self._readpartheader()
833 indebug(self.ui, 'end of bundle2 stream')
833 indebug(self.ui, 'end of bundle2 stream')
834
834
835 def _readpartheader(self):
835 def _readpartheader(self):
836 """reads a part header size and return the bytes blob
836 """reads a part header size and return the bytes blob
837
837
838 returns None if empty"""
838 returns None if empty"""
839 headersize = self._unpack(_fpartheadersize)[0]
839 headersize = self._unpack(_fpartheadersize)[0]
840 if headersize < 0:
840 if headersize < 0:
841 raise error.BundleValueError('negative part header size: %i'
841 raise error.BundleValueError('negative part header size: %i'
842 % headersize)
842 % headersize)
843 indebug(self.ui, 'part header size: %i' % headersize)
843 indebug(self.ui, 'part header size: %i' % headersize)
844 if headersize:
844 if headersize:
845 return self._readexact(headersize)
845 return self._readexact(headersize)
846 return None
846 return None
847
847
848 def compressed(self):
848 def compressed(self):
849 self.params # load params
849 self.params # load params
850 return self._compressed
850 return self._compressed
851
851
852 def close(self):
852 def close(self):
853 """close underlying file"""
853 """close underlying file"""
854 if util.safehasattr(self._fp, 'close'):
854 if util.safehasattr(self._fp, 'close'):
855 return self._fp.close()
855 return self._fp.close()
856
856
857 formatmap = {'20': unbundle20}
857 formatmap = {'20': unbundle20}
858
858
859 b2streamparamsmap = {}
859 b2streamparamsmap = {}
860
860
861 def b2streamparamhandler(name):
861 def b2streamparamhandler(name):
862 """register a handler for a stream level parameter"""
862 """register a handler for a stream level parameter"""
863 def decorator(func):
863 def decorator(func):
864 assert name not in formatmap
864 assert name not in formatmap
865 b2streamparamsmap[name] = func
865 b2streamparamsmap[name] = func
866 return func
866 return func
867 return decorator
867 return decorator
868
868
869 @b2streamparamhandler('compression')
869 @b2streamparamhandler('compression')
870 def processcompression(unbundler, param, value):
870 def processcompression(unbundler, param, value):
871 """read compression parameter and install payload decompression"""
871 """read compression parameter and install payload decompression"""
872 if value not in util.compengines.supportedbundletypes:
872 if value not in util.compengines.supportedbundletypes:
873 raise error.BundleUnknownFeatureError(params=(param,),
873 raise error.BundleUnknownFeatureError(params=(param,),
874 values=(value,))
874 values=(value,))
875 unbundler._compengine = util.compengines.forbundletype(value)
875 unbundler._compengine = util.compengines.forbundletype(value)
876 if value is not None:
876 if value is not None:
877 unbundler._compressed = True
877 unbundler._compressed = True
878
878
879 class bundlepart(object):
879 class bundlepart(object):
880 """A bundle2 part contains application level payload
880 """A bundle2 part contains application level payload
881
881
882 The part `type` is used to route the part to the application level
882 The part `type` is used to route the part to the application level
883 handler.
883 handler.
884
884
885 The part payload is contained in ``part.data``. It could be raw bytes or a
885 The part payload is contained in ``part.data``. It could be raw bytes or a
886 generator of byte chunks.
886 generator of byte chunks.
887
887
888 You can add parameters to the part using the ``addparam`` method.
888 You can add parameters to the part using the ``addparam`` method.
889 Parameters can be either mandatory (default) or advisory. Remote side
889 Parameters can be either mandatory (default) or advisory. Remote side
890 should be able to safely ignore the advisory ones.
890 should be able to safely ignore the advisory ones.
891
891
892 Both data and parameters cannot be modified after the generation has begun.
892 Both data and parameters cannot be modified after the generation has begun.
893 """
893 """
894
894
895 def __init__(self, parttype, mandatoryparams=(), advisoryparams=(),
895 def __init__(self, parttype, mandatoryparams=(), advisoryparams=(),
896 data='', mandatory=True):
896 data='', mandatory=True):
897 validateparttype(parttype)
897 validateparttype(parttype)
898 self.id = None
898 self.id = None
899 self.type = parttype
899 self.type = parttype
900 self._data = data
900 self._data = data
901 self._mandatoryparams = list(mandatoryparams)
901 self._mandatoryparams = list(mandatoryparams)
902 self._advisoryparams = list(advisoryparams)
902 self._advisoryparams = list(advisoryparams)
903 # checking for duplicated entries
903 # checking for duplicated entries
904 self._seenparams = set()
904 self._seenparams = set()
905 for pname, __ in self._mandatoryparams + self._advisoryparams:
905 for pname, __ in self._mandatoryparams + self._advisoryparams:
906 if pname in self._seenparams:
906 if pname in self._seenparams:
907 raise error.ProgrammingError('duplicated params: %s' % pname)
907 raise error.ProgrammingError('duplicated params: %s' % pname)
908 self._seenparams.add(pname)
908 self._seenparams.add(pname)
909 # status of the part's generation:
909 # status of the part's generation:
910 # - None: not started,
910 # - None: not started,
911 # - False: currently generated,
911 # - False: currently generated,
912 # - True: generation done.
912 # - True: generation done.
913 self._generated = None
913 self._generated = None
914 self.mandatory = mandatory
914 self.mandatory = mandatory
915
915
916 def __repr__(self):
916 def __repr__(self):
917 cls = "%s.%s" % (self.__class__.__module__, self.__class__.__name__)
917 cls = "%s.%s" % (self.__class__.__module__, self.__class__.__name__)
918 return ('<%s object at %x; id: %s; type: %s; mandatory: %s>'
918 return ('<%s object at %x; id: %s; type: %s; mandatory: %s>'
919 % (cls, id(self), self.id, self.type, self.mandatory))
919 % (cls, id(self), self.id, self.type, self.mandatory))
920
920
921 def copy(self):
921 def copy(self):
922 """return a copy of the part
922 """return a copy of the part
923
923
924 The new part have the very same content but no partid assigned yet.
924 The new part have the very same content but no partid assigned yet.
925 Parts with generated data cannot be copied."""
925 Parts with generated data cannot be copied."""
926 assert not util.safehasattr(self.data, 'next')
926 assert not util.safehasattr(self.data, 'next')
927 return self.__class__(self.type, self._mandatoryparams,
927 return self.__class__(self.type, self._mandatoryparams,
928 self._advisoryparams, self._data, self.mandatory)
928 self._advisoryparams, self._data, self.mandatory)
929
929
930 # methods used to defines the part content
930 # methods used to defines the part content
931 @property
931 @property
932 def data(self):
932 def data(self):
933 return self._data
933 return self._data
934
934
935 @data.setter
935 @data.setter
936 def data(self, data):
936 def data(self, data):
937 if self._generated is not None:
937 if self._generated is not None:
938 raise error.ReadOnlyPartError('part is being generated')
938 raise error.ReadOnlyPartError('part is being generated')
939 self._data = data
939 self._data = data
940
940
941 @property
941 @property
942 def mandatoryparams(self):
942 def mandatoryparams(self):
943 # make it an immutable tuple to force people through ``addparam``
943 # make it an immutable tuple to force people through ``addparam``
944 return tuple(self._mandatoryparams)
944 return tuple(self._mandatoryparams)
945
945
946 @property
946 @property
947 def advisoryparams(self):
947 def advisoryparams(self):
948 # make it an immutable tuple to force people through ``addparam``
948 # make it an immutable tuple to force people through ``addparam``
949 return tuple(self._advisoryparams)
949 return tuple(self._advisoryparams)
950
950
951 def addparam(self, name, value='', mandatory=True):
951 def addparam(self, name, value='', mandatory=True):
952 """add a parameter to the part
952 """add a parameter to the part
953
953
954 If 'mandatory' is set to True, the remote handler must claim support
954 If 'mandatory' is set to True, the remote handler must claim support
955 for this parameter or the unbundling will be aborted.
955 for this parameter or the unbundling will be aborted.
956
956
957 The 'name' and 'value' cannot exceed 255 bytes each.
957 The 'name' and 'value' cannot exceed 255 bytes each.
958 """
958 """
959 if self._generated is not None:
959 if self._generated is not None:
960 raise error.ReadOnlyPartError('part is being generated')
960 raise error.ReadOnlyPartError('part is being generated')
961 if name in self._seenparams:
961 if name in self._seenparams:
962 raise ValueError('duplicated params: %s' % name)
962 raise ValueError('duplicated params: %s' % name)
963 self._seenparams.add(name)
963 self._seenparams.add(name)
964 params = self._advisoryparams
964 params = self._advisoryparams
965 if mandatory:
965 if mandatory:
966 params = self._mandatoryparams
966 params = self._mandatoryparams
967 params.append((name, value))
967 params.append((name, value))
968
968
969 # methods used to generates the bundle2 stream
969 # methods used to generates the bundle2 stream
970 def getchunks(self, ui):
970 def getchunks(self, ui):
971 if self._generated is not None:
971 if self._generated is not None:
972 raise error.ProgrammingError('part can only be consumed once')
972 raise error.ProgrammingError('part can only be consumed once')
973 self._generated = False
973 self._generated = False
974
974
975 if ui.debugflag:
975 if ui.debugflag:
976 msg = ['bundle2-output-part: "%s"' % self.type]
976 msg = ['bundle2-output-part: "%s"' % self.type]
977 if not self.mandatory:
977 if not self.mandatory:
978 msg.append(' (advisory)')
978 msg.append(' (advisory)')
979 nbmp = len(self.mandatoryparams)
979 nbmp = len(self.mandatoryparams)
980 nbap = len(self.advisoryparams)
980 nbap = len(self.advisoryparams)
981 if nbmp or nbap:
981 if nbmp or nbap:
982 msg.append(' (params:')
982 msg.append(' (params:')
983 if nbmp:
983 if nbmp:
984 msg.append(' %i mandatory' % nbmp)
984 msg.append(' %i mandatory' % nbmp)
985 if nbap:
985 if nbap:
986 msg.append(' %i advisory' % nbmp)
986 msg.append(' %i advisory' % nbmp)
987 msg.append(')')
987 msg.append(')')
988 if not self.data:
988 if not self.data:
989 msg.append(' empty payload')
989 msg.append(' empty payload')
990 elif util.safehasattr(self.data, 'next'):
990 elif util.safehasattr(self.data, 'next'):
991 msg.append(' streamed payload')
991 msg.append(' streamed payload')
992 else:
992 else:
993 msg.append(' %i bytes payload' % len(self.data))
993 msg.append(' %i bytes payload' % len(self.data))
994 msg.append('\n')
994 msg.append('\n')
995 ui.debug(''.join(msg))
995 ui.debug(''.join(msg))
996
996
997 #### header
997 #### header
998 if self.mandatory:
998 if self.mandatory:
999 parttype = self.type.upper()
999 parttype = self.type.upper()
1000 else:
1000 else:
1001 parttype = self.type.lower()
1001 parttype = self.type.lower()
1002 outdebug(ui, 'part %s: "%s"' % (pycompat.bytestr(self.id), parttype))
1002 outdebug(ui, 'part %s: "%s"' % (pycompat.bytestr(self.id), parttype))
1003 ## parttype
1003 ## parttype
1004 header = [_pack(_fparttypesize, len(parttype)),
1004 header = [_pack(_fparttypesize, len(parttype)),
1005 parttype, _pack(_fpartid, self.id),
1005 parttype, _pack(_fpartid, self.id),
1006 ]
1006 ]
1007 ## parameters
1007 ## parameters
1008 # count
1008 # count
1009 manpar = self.mandatoryparams
1009 manpar = self.mandatoryparams
1010 advpar = self.advisoryparams
1010 advpar = self.advisoryparams
1011 header.append(_pack(_fpartparamcount, len(manpar), len(advpar)))
1011 header.append(_pack(_fpartparamcount, len(manpar), len(advpar)))
1012 # size
1012 # size
1013 parsizes = []
1013 parsizes = []
1014 for key, value in manpar:
1014 for key, value in manpar:
1015 parsizes.append(len(key))
1015 parsizes.append(len(key))
1016 parsizes.append(len(value))
1016 parsizes.append(len(value))
1017 for key, value in advpar:
1017 for key, value in advpar:
1018 parsizes.append(len(key))
1018 parsizes.append(len(key))
1019 parsizes.append(len(value))
1019 parsizes.append(len(value))
1020 paramsizes = _pack(_makefpartparamsizes(len(parsizes) // 2), *parsizes)
1020 paramsizes = _pack(_makefpartparamsizes(len(parsizes) // 2), *parsizes)
1021 header.append(paramsizes)
1021 header.append(paramsizes)
1022 # key, value
1022 # key, value
1023 for key, value in manpar:
1023 for key, value in manpar:
1024 header.append(key)
1024 header.append(key)
1025 header.append(value)
1025 header.append(value)
1026 for key, value in advpar:
1026 for key, value in advpar:
1027 header.append(key)
1027 header.append(key)
1028 header.append(value)
1028 header.append(value)
1029 ## finalize header
1029 ## finalize header
1030 headerchunk = ''.join(header)
1030 headerchunk = ''.join(header)
1031 outdebug(ui, 'header chunk size: %i' % len(headerchunk))
1031 outdebug(ui, 'header chunk size: %i' % len(headerchunk))
1032 yield _pack(_fpartheadersize, len(headerchunk))
1032 yield _pack(_fpartheadersize, len(headerchunk))
1033 yield headerchunk
1033 yield headerchunk
1034 ## payload
1034 ## payload
1035 try:
1035 try:
1036 for chunk in self._payloadchunks():
1036 for chunk in self._payloadchunks():
1037 outdebug(ui, 'payload chunk size: %i' % len(chunk))
1037 outdebug(ui, 'payload chunk size: %i' % len(chunk))
1038 yield _pack(_fpayloadsize, len(chunk))
1038 yield _pack(_fpayloadsize, len(chunk))
1039 yield chunk
1039 yield chunk
1040 except GeneratorExit:
1040 except GeneratorExit:
1041 # GeneratorExit means that nobody is listening for our
1041 # GeneratorExit means that nobody is listening for our
1042 # results anyway, so just bail quickly rather than trying
1042 # results anyway, so just bail quickly rather than trying
1043 # to produce an error part.
1043 # to produce an error part.
1044 ui.debug('bundle2-generatorexit\n')
1044 ui.debug('bundle2-generatorexit\n')
1045 raise
1045 raise
1046 except BaseException as exc:
1046 except BaseException as exc:
1047 bexc = util.forcebytestr(exc)
1047 bexc = util.forcebytestr(exc)
1048 # backup exception data for later
1048 # backup exception data for later
1049 ui.debug('bundle2-input-stream-interrupt: encoding exception %s'
1049 ui.debug('bundle2-input-stream-interrupt: encoding exception %s'
1050 % bexc)
1050 % bexc)
1051 tb = sys.exc_info()[2]
1051 tb = sys.exc_info()[2]
1052 msg = 'unexpected error: %s' % bexc
1052 msg = 'unexpected error: %s' % bexc
1053 interpart = bundlepart('error:abort', [('message', msg)],
1053 interpart = bundlepart('error:abort', [('message', msg)],
1054 mandatory=False)
1054 mandatory=False)
1055 interpart.id = 0
1055 interpart.id = 0
1056 yield _pack(_fpayloadsize, -1)
1056 yield _pack(_fpayloadsize, -1)
1057 for chunk in interpart.getchunks(ui=ui):
1057 for chunk in interpart.getchunks(ui=ui):
1058 yield chunk
1058 yield chunk
1059 outdebug(ui, 'closing payload chunk')
1059 outdebug(ui, 'closing payload chunk')
1060 # abort current part payload
1060 # abort current part payload
1061 yield _pack(_fpayloadsize, 0)
1061 yield _pack(_fpayloadsize, 0)
1062 pycompat.raisewithtb(exc, tb)
1062 pycompat.raisewithtb(exc, tb)
1063 # end of payload
1063 # end of payload
1064 outdebug(ui, 'closing payload chunk')
1064 outdebug(ui, 'closing payload chunk')
1065 yield _pack(_fpayloadsize, 0)
1065 yield _pack(_fpayloadsize, 0)
1066 self._generated = True
1066 self._generated = True
1067
1067
1068 def _payloadchunks(self):
1068 def _payloadchunks(self):
1069 """yield chunks of a the part payload
1069 """yield chunks of a the part payload
1070
1070
1071 Exists to handle the different methods to provide data to a part."""
1071 Exists to handle the different methods to provide data to a part."""
1072 # we only support fixed size data now.
1072 # we only support fixed size data now.
1073 # This will be improved in the future.
1073 # This will be improved in the future.
1074 if (util.safehasattr(self.data, 'next')
1074 if (util.safehasattr(self.data, 'next')
1075 or util.safehasattr(self.data, '__next__')):
1075 or util.safehasattr(self.data, '__next__')):
1076 buff = util.chunkbuffer(self.data)
1076 buff = util.chunkbuffer(self.data)
1077 chunk = buff.read(preferedchunksize)
1077 chunk = buff.read(preferedchunksize)
1078 while chunk:
1078 while chunk:
1079 yield chunk
1079 yield chunk
1080 chunk = buff.read(preferedchunksize)
1080 chunk = buff.read(preferedchunksize)
1081 elif len(self.data):
1081 elif len(self.data):
1082 yield self.data
1082 yield self.data
1083
1083
1084
1084
1085 flaginterrupt = -1
1085 flaginterrupt = -1
1086
1086
1087 class interrupthandler(unpackermixin):
1087 class interrupthandler(unpackermixin):
1088 """read one part and process it with restricted capability
1088 """read one part and process it with restricted capability
1089
1089
1090 This allows to transmit exception raised on the producer size during part
1090 This allows to transmit exception raised on the producer size during part
1091 iteration while the consumer is reading a part.
1091 iteration while the consumer is reading a part.
1092
1092
1093 Part processed in this manner only have access to a ui object,"""
1093 Part processed in this manner only have access to a ui object,"""
1094
1094
1095 def __init__(self, ui, fp):
1095 def __init__(self, ui, fp):
1096 super(interrupthandler, self).__init__(fp)
1096 super(interrupthandler, self).__init__(fp)
1097 self.ui = ui
1097 self.ui = ui
1098
1098
1099 def _readpartheader(self):
1099 def _readpartheader(self):
1100 """reads a part header size and return the bytes blob
1100 """reads a part header size and return the bytes blob
1101
1101
1102 returns None if empty"""
1102 returns None if empty"""
1103 headersize = self._unpack(_fpartheadersize)[0]
1103 headersize = self._unpack(_fpartheadersize)[0]
1104 if headersize < 0:
1104 if headersize < 0:
1105 raise error.BundleValueError('negative part header size: %i'
1105 raise error.BundleValueError('negative part header size: %i'
1106 % headersize)
1106 % headersize)
1107 indebug(self.ui, 'part header size: %i\n' % headersize)
1107 indebug(self.ui, 'part header size: %i\n' % headersize)
1108 if headersize:
1108 if headersize:
1109 return self._readexact(headersize)
1109 return self._readexact(headersize)
1110 return None
1110 return None
1111
1111
1112 def __call__(self):
1112 def __call__(self):
1113
1113
1114 self.ui.debug('bundle2-input-stream-interrupt:'
1114 self.ui.debug('bundle2-input-stream-interrupt:'
1115 ' opening out of band context\n')
1115 ' opening out of band context\n')
1116 indebug(self.ui, 'bundle2 stream interruption, looking for a part.')
1116 indebug(self.ui, 'bundle2 stream interruption, looking for a part.')
1117 headerblock = self._readpartheader()
1117 headerblock = self._readpartheader()
1118 if headerblock is None:
1118 if headerblock is None:
1119 indebug(self.ui, 'no part found during interruption.')
1119 indebug(self.ui, 'no part found during interruption.')
1120 return
1120 return
1121 part = unbundlepart(self.ui, headerblock, self._fp)
1121 part = unbundlepart(self.ui, headerblock, self._fp)
1122 op = interruptoperation(self.ui)
1122 op = interruptoperation(self.ui)
1123 _processpart(op, part)
1123 _processpart(op, part)
1124 self.ui.debug('bundle2-input-stream-interrupt:'
1124 self.ui.debug('bundle2-input-stream-interrupt:'
1125 ' closing out of band context\n')
1125 ' closing out of band context\n')
1126
1126
1127 class interruptoperation(object):
1127 class interruptoperation(object):
1128 """A limited operation to be use by part handler during interruption
1128 """A limited operation to be use by part handler during interruption
1129
1129
1130 It only have access to an ui object.
1130 It only have access to an ui object.
1131 """
1131 """
1132
1132
1133 def __init__(self, ui):
1133 def __init__(self, ui):
1134 self.ui = ui
1134 self.ui = ui
1135 self.reply = None
1135 self.reply = None
1136 self.captureoutput = False
1136 self.captureoutput = False
1137
1137
1138 @property
1138 @property
1139 def repo(self):
1139 def repo(self):
1140 raise error.ProgrammingError('no repo access from stream interruption')
1140 raise error.ProgrammingError('no repo access from stream interruption')
1141
1141
1142 def gettransaction(self):
1142 def gettransaction(self):
1143 raise TransactionUnavailable('no repo access from stream interruption')
1143 raise TransactionUnavailable('no repo access from stream interruption')
1144
1144
1145 class unbundlepart(unpackermixin):
1145 class unbundlepart(unpackermixin):
1146 """a bundle part read from a bundle"""
1146 """a bundle part read from a bundle"""
1147
1147
1148 def __init__(self, ui, header, fp):
1148 def __init__(self, ui, header, fp):
1149 super(unbundlepart, self).__init__(fp)
1149 super(unbundlepart, self).__init__(fp)
1150 self._seekable = (util.safehasattr(fp, 'seek') and
1150 self._seekable = (util.safehasattr(fp, 'seek') and
1151 util.safehasattr(fp, 'tell'))
1151 util.safehasattr(fp, 'tell'))
1152 self.ui = ui
1152 self.ui = ui
1153 # unbundle state attr
1153 # unbundle state attr
1154 self._headerdata = header
1154 self._headerdata = header
1155 self._headeroffset = 0
1155 self._headeroffset = 0
1156 self._initialized = False
1156 self._initialized = False
1157 self.consumed = False
1157 self.consumed = False
1158 # part data
1158 # part data
1159 self.id = None
1159 self.id = None
1160 self.type = None
1160 self.type = None
1161 self.mandatoryparams = None
1161 self.mandatoryparams = None
1162 self.advisoryparams = None
1162 self.advisoryparams = None
1163 self.params = None
1163 self.params = None
1164 self.mandatorykeys = ()
1164 self.mandatorykeys = ()
1165 self._payloadstream = None
1165 self._payloadstream = None
1166 self._readheader()
1166 self._readheader()
1167 self._mandatory = None
1167 self._mandatory = None
1168 self._chunkindex = [] #(payload, file) position tuples for chunk starts
1168 self._chunkindex = [] #(payload, file) position tuples for chunk starts
1169 self._pos = 0
1169 self._pos = 0
1170
1170
1171 def _fromheader(self, size):
1171 def _fromheader(self, size):
1172 """return the next <size> byte from the header"""
1172 """return the next <size> byte from the header"""
1173 offset = self._headeroffset
1173 offset = self._headeroffset
1174 data = self._headerdata[offset:(offset + size)]
1174 data = self._headerdata[offset:(offset + size)]
1175 self._headeroffset = offset + size
1175 self._headeroffset = offset + size
1176 return data
1176 return data
1177
1177
1178 def _unpackheader(self, format):
1178 def _unpackheader(self, format):
1179 """read given format from header
1179 """read given format from header
1180
1180
1181 This automatically compute the size of the format to read."""
1181 This automatically compute the size of the format to read."""
1182 data = self._fromheader(struct.calcsize(format))
1182 data = self._fromheader(struct.calcsize(format))
1183 return _unpack(format, data)
1183 return _unpack(format, data)
1184
1184
1185 def _initparams(self, mandatoryparams, advisoryparams):
1185 def _initparams(self, mandatoryparams, advisoryparams):
1186 """internal function to setup all logic related parameters"""
1186 """internal function to setup all logic related parameters"""
1187 # make it read only to prevent people touching it by mistake.
1187 # make it read only to prevent people touching it by mistake.
1188 self.mandatoryparams = tuple(mandatoryparams)
1188 self.mandatoryparams = tuple(mandatoryparams)
1189 self.advisoryparams = tuple(advisoryparams)
1189 self.advisoryparams = tuple(advisoryparams)
1190 # user friendly UI
1190 # user friendly UI
1191 self.params = util.sortdict(self.mandatoryparams)
1191 self.params = util.sortdict(self.mandatoryparams)
1192 self.params.update(self.advisoryparams)
1192 self.params.update(self.advisoryparams)
1193 self.mandatorykeys = frozenset(p[0] for p in mandatoryparams)
1193 self.mandatorykeys = frozenset(p[0] for p in mandatoryparams)
1194
1194
1195 def _payloadchunks(self, chunknum=0):
1195 def _payloadchunks(self, chunknum=0):
1196 '''seek to specified chunk and start yielding data'''
1196 '''seek to specified chunk and start yielding data'''
1197 if len(self._chunkindex) == 0:
1197 if len(self._chunkindex) == 0:
1198 assert chunknum == 0, 'Must start with chunk 0'
1198 assert chunknum == 0, 'Must start with chunk 0'
1199 self._chunkindex.append((0, self._tellfp()))
1199 self._chunkindex.append((0, self._tellfp()))
1200 else:
1200 else:
1201 assert chunknum < len(self._chunkindex), \
1201 assert chunknum < len(self._chunkindex), \
1202 'Unknown chunk %d' % chunknum
1202 'Unknown chunk %d' % chunknum
1203 self._seekfp(self._chunkindex[chunknum][1])
1203 self._seekfp(self._chunkindex[chunknum][1])
1204
1204
1205 pos = self._chunkindex[chunknum][0]
1205 pos = self._chunkindex[chunknum][0]
1206 payloadsize = self._unpack(_fpayloadsize)[0]
1206 payloadsize = self._unpack(_fpayloadsize)[0]
1207 indebug(self.ui, 'payload chunk size: %i' % payloadsize)
1207 indebug(self.ui, 'payload chunk size: %i' % payloadsize)
1208 while payloadsize:
1208 while payloadsize:
1209 if payloadsize == flaginterrupt:
1209 if payloadsize == flaginterrupt:
1210 # interruption detection, the handler will now read a
1210 # interruption detection, the handler will now read a
1211 # single part and process it.
1211 # single part and process it.
1212 interrupthandler(self.ui, self._fp)()
1212 interrupthandler(self.ui, self._fp)()
1213 elif payloadsize < 0:
1213 elif payloadsize < 0:
1214 msg = 'negative payload chunk size: %i' % payloadsize
1214 msg = 'negative payload chunk size: %i' % payloadsize
1215 raise error.BundleValueError(msg)
1215 raise error.BundleValueError(msg)
1216 else:
1216 else:
1217 result = self._readexact(payloadsize)
1217 result = self._readexact(payloadsize)
1218 chunknum += 1
1218 chunknum += 1
1219 pos += payloadsize
1219 pos += payloadsize
1220 if chunknum == len(self._chunkindex):
1220 if chunknum == len(self._chunkindex):
1221 self._chunkindex.append((pos, self._tellfp()))
1221 self._chunkindex.append((pos, self._tellfp()))
1222 yield result
1222 yield result
1223 payloadsize = self._unpack(_fpayloadsize)[0]
1223 payloadsize = self._unpack(_fpayloadsize)[0]
1224 indebug(self.ui, 'payload chunk size: %i' % payloadsize)
1224 indebug(self.ui, 'payload chunk size: %i' % payloadsize)
1225
1225
1226 def _findchunk(self, pos):
1226 def _findchunk(self, pos):
1227 '''for a given payload position, return a chunk number and offset'''
1227 '''for a given payload position, return a chunk number and offset'''
1228 for chunk, (ppos, fpos) in enumerate(self._chunkindex):
1228 for chunk, (ppos, fpos) in enumerate(self._chunkindex):
1229 if ppos == pos:
1229 if ppos == pos:
1230 return chunk, 0
1230 return chunk, 0
1231 elif ppos > pos:
1231 elif ppos > pos:
1232 return chunk - 1, pos - self._chunkindex[chunk - 1][0]
1232 return chunk - 1, pos - self._chunkindex[chunk - 1][0]
1233 raise ValueError('Unknown chunk')
1233 raise ValueError('Unknown chunk')
1234
1234
1235 def _readheader(self):
1235 def _readheader(self):
1236 """read the header and setup the object"""
1236 """read the header and setup the object"""
1237 typesize = self._unpackheader(_fparttypesize)[0]
1237 typesize = self._unpackheader(_fparttypesize)[0]
1238 self.type = self._fromheader(typesize)
1238 self.type = self._fromheader(typesize)
1239 indebug(self.ui, 'part type: "%s"' % self.type)
1239 indebug(self.ui, 'part type: "%s"' % self.type)
1240 self.id = self._unpackheader(_fpartid)[0]
1240 self.id = self._unpackheader(_fpartid)[0]
1241 indebug(self.ui, 'part id: "%s"' % pycompat.bytestr(self.id))
1241 indebug(self.ui, 'part id: "%s"' % pycompat.bytestr(self.id))
1242 # extract mandatory bit from type
1242 # extract mandatory bit from type
1243 self.mandatory = (self.type != self.type.lower())
1243 self.mandatory = (self.type != self.type.lower())
1244 self.type = self.type.lower()
1244 self.type = self.type.lower()
1245 ## reading parameters
1245 ## reading parameters
1246 # param count
1246 # param count
1247 mancount, advcount = self._unpackheader(_fpartparamcount)
1247 mancount, advcount = self._unpackheader(_fpartparamcount)
1248 indebug(self.ui, 'part parameters: %i' % (mancount + advcount))
1248 indebug(self.ui, 'part parameters: %i' % (mancount + advcount))
1249 # param size
1249 # param size
1250 fparamsizes = _makefpartparamsizes(mancount + advcount)
1250 fparamsizes = _makefpartparamsizes(mancount + advcount)
1251 paramsizes = self._unpackheader(fparamsizes)
1251 paramsizes = self._unpackheader(fparamsizes)
1252 # make it a list of couple again
1252 # make it a list of couple again
1253 paramsizes = list(zip(paramsizes[::2], paramsizes[1::2]))
1253 paramsizes = list(zip(paramsizes[::2], paramsizes[1::2]))
1254 # split mandatory from advisory
1254 # split mandatory from advisory
1255 mansizes = paramsizes[:mancount]
1255 mansizes = paramsizes[:mancount]
1256 advsizes = paramsizes[mancount:]
1256 advsizes = paramsizes[mancount:]
1257 # retrieve param value
1257 # retrieve param value
1258 manparams = []
1258 manparams = []
1259 for key, value in mansizes:
1259 for key, value in mansizes:
1260 manparams.append((self._fromheader(key), self._fromheader(value)))
1260 manparams.append((self._fromheader(key), self._fromheader(value)))
1261 advparams = []
1261 advparams = []
1262 for key, value in advsizes:
1262 for key, value in advsizes:
1263 advparams.append((self._fromheader(key), self._fromheader(value)))
1263 advparams.append((self._fromheader(key), self._fromheader(value)))
1264 self._initparams(manparams, advparams)
1264 self._initparams(manparams, advparams)
1265 ## part payload
1265 ## part payload
1266 self._payloadstream = util.chunkbuffer(self._payloadchunks())
1266 self._payloadstream = util.chunkbuffer(self._payloadchunks())
1267 # we read the data, tell it
1267 # we read the data, tell it
1268 self._initialized = True
1268 self._initialized = True
1269
1269
1270 def read(self, size=None):
1270 def read(self, size=None):
1271 """read payload data"""
1271 """read payload data"""
1272 if not self._initialized:
1272 if not self._initialized:
1273 self._readheader()
1273 self._readheader()
1274 if size is None:
1274 if size is None:
1275 data = self._payloadstream.read()
1275 data = self._payloadstream.read()
1276 else:
1276 else:
1277 data = self._payloadstream.read(size)
1277 data = self._payloadstream.read(size)
1278 self._pos += len(data)
1278 self._pos += len(data)
1279 if size is None or len(data) < size:
1279 if size is None or len(data) < size:
1280 if not self.consumed and self._pos:
1280 if not self.consumed and self._pos:
1281 self.ui.debug('bundle2-input-part: total payload size %i\n'
1281 self.ui.debug('bundle2-input-part: total payload size %i\n'
1282 % self._pos)
1282 % self._pos)
1283 self.consumed = True
1283 self.consumed = True
1284 return data
1284 return data
1285
1285
1286 def tell(self):
1286 def tell(self):
1287 return self._pos
1287 return self._pos
1288
1288
1289 def seek(self, offset, whence=0):
1289 def seek(self, offset, whence=0):
1290 if whence == 0:
1290 if whence == 0:
1291 newpos = offset
1291 newpos = offset
1292 elif whence == 1:
1292 elif whence == 1:
1293 newpos = self._pos + offset
1293 newpos = self._pos + offset
1294 elif whence == 2:
1294 elif whence == 2:
1295 if not self.consumed:
1295 if not self.consumed:
1296 self.read()
1296 self.read()
1297 newpos = self._chunkindex[-1][0] - offset
1297 newpos = self._chunkindex[-1][0] - offset
1298 else:
1298 else:
1299 raise ValueError('Unknown whence value: %r' % (whence,))
1299 raise ValueError('Unknown whence value: %r' % (whence,))
1300
1300
1301 if newpos > self._chunkindex[-1][0] and not self.consumed:
1301 if newpos > self._chunkindex[-1][0] and not self.consumed:
1302 self.read()
1302 self.read()
1303 if not 0 <= newpos <= self._chunkindex[-1][0]:
1303 if not 0 <= newpos <= self._chunkindex[-1][0]:
1304 raise ValueError('Offset out of range')
1304 raise ValueError('Offset out of range')
1305
1305
1306 if self._pos != newpos:
1306 if self._pos != newpos:
1307 chunk, internaloffset = self._findchunk(newpos)
1307 chunk, internaloffset = self._findchunk(newpos)
1308 self._payloadstream = util.chunkbuffer(self._payloadchunks(chunk))
1308 self._payloadstream = util.chunkbuffer(self._payloadchunks(chunk))
1309 adjust = self.read(internaloffset)
1309 adjust = self.read(internaloffset)
1310 if len(adjust) != internaloffset:
1310 if len(adjust) != internaloffset:
1311 raise error.Abort(_('Seek failed\n'))
1311 raise error.Abort(_('Seek failed\n'))
1312 self._pos = newpos
1312 self._pos = newpos
1313
1313
1314 def _seekfp(self, offset, whence=0):
1314 def _seekfp(self, offset, whence=0):
1315 """move the underlying file pointer
1315 """move the underlying file pointer
1316
1316
1317 This method is meant for internal usage by the bundle2 protocol only.
1317 This method is meant for internal usage by the bundle2 protocol only.
1318 They directly manipulate the low level stream including bundle2 level
1318 They directly manipulate the low level stream including bundle2 level
1319 instruction.
1319 instruction.
1320
1320
1321 Do not use it to implement higher-level logic or methods."""
1321 Do not use it to implement higher-level logic or methods."""
1322 if self._seekable:
1322 if self._seekable:
1323 return self._fp.seek(offset, whence)
1323 return self._fp.seek(offset, whence)
1324 else:
1324 else:
1325 raise NotImplementedError(_('File pointer is not seekable'))
1325 raise NotImplementedError(_('File pointer is not seekable'))
1326
1326
1327 def _tellfp(self):
1327 def _tellfp(self):
1328 """return the file offset, or None if file is not seekable
1328 """return the file offset, or None if file is not seekable
1329
1329
1330 This method is meant for internal usage by the bundle2 protocol only.
1330 This method is meant for internal usage by the bundle2 protocol only.
1331 They directly manipulate the low level stream including bundle2 level
1331 They directly manipulate the low level stream including bundle2 level
1332 instruction.
1332 instruction.
1333
1333
1334 Do not use it to implement higher-level logic or methods."""
1334 Do not use it to implement higher-level logic or methods."""
1335 if self._seekable:
1335 if self._seekable:
1336 try:
1336 try:
1337 return self._fp.tell()
1337 return self._fp.tell()
1338 except IOError as e:
1338 except IOError as e:
1339 if e.errno == errno.ESPIPE:
1339 if e.errno == errno.ESPIPE:
1340 self._seekable = False
1340 self._seekable = False
1341 else:
1341 else:
1342 raise
1342 raise
1343 return None
1343 return None
1344
1344
1345 # These are only the static capabilities.
1345 # These are only the static capabilities.
1346 # Check the 'getrepocaps' function for the rest.
1346 # Check the 'getrepocaps' function for the rest.
1347 capabilities = {'HG20': (),
1347 capabilities = {'HG20': (),
1348 'error': ('abort', 'unsupportedcontent', 'pushraced',
1348 'error': ('abort', 'unsupportedcontent', 'pushraced',
1349 'pushkey'),
1349 'pushkey'),
1350 'listkeys': (),
1350 'listkeys': (),
1351 'pushkey': (),
1351 'pushkey': (),
1352 'digests': tuple(sorted(util.DIGESTS.keys())),
1352 'digests': tuple(sorted(util.DIGESTS.keys())),
1353 'remote-changegroup': ('http', 'https'),
1353 'remote-changegroup': ('http', 'https'),
1354 'hgtagsfnodes': (),
1354 'hgtagsfnodes': (),
1355 }
1355 }
1356
1356
1357 def getrepocaps(repo, allowpushback=False):
1357 def getrepocaps(repo, allowpushback=False):
1358 """return the bundle2 capabilities for a given repo
1358 """return the bundle2 capabilities for a given repo
1359
1359
1360 Exists to allow extensions (like evolution) to mutate the capabilities.
1360 Exists to allow extensions (like evolution) to mutate the capabilities.
1361 """
1361 """
1362 caps = capabilities.copy()
1362 caps = capabilities.copy()
1363 caps['changegroup'] = tuple(sorted(
1363 caps['changegroup'] = tuple(sorted(
1364 changegroup.supportedincomingversions(repo)))
1364 changegroup.supportedincomingversions(repo)))
1365 if obsolete.isenabled(repo, obsolete.exchangeopt):
1365 if obsolete.isenabled(repo, obsolete.exchangeopt):
1366 supportedformat = tuple('V%i' % v for v in obsolete.formats)
1366 supportedformat = tuple('V%i' % v for v in obsolete.formats)
1367 caps['obsmarkers'] = supportedformat
1367 caps['obsmarkers'] = supportedformat
1368 if allowpushback:
1368 if allowpushback:
1369 caps['pushback'] = ()
1369 caps['pushback'] = ()
1370 cpmode = repo.ui.config('server', 'concurrent-push-mode')
1370 cpmode = repo.ui.config('server', 'concurrent-push-mode')
1371 if cpmode == 'check-related':
1371 if cpmode == 'check-related':
1372 caps['checkheads'] = ('related',)
1372 caps['checkheads'] = ('related',)
1373 return caps
1373 return caps
1374
1374
1375 def bundle2caps(remote):
1375 def bundle2caps(remote):
1376 """return the bundle capabilities of a peer as dict"""
1376 """return the bundle capabilities of a peer as dict"""
1377 raw = remote.capable('bundle2')
1377 raw = remote.capable('bundle2')
1378 if not raw and raw != '':
1378 if not raw and raw != '':
1379 return {}
1379 return {}
1380 capsblob = urlreq.unquote(remote.capable('bundle2'))
1380 capsblob = urlreq.unquote(remote.capable('bundle2'))
1381 return decodecaps(capsblob)
1381 return decodecaps(capsblob)
1382
1382
1383 def obsmarkersversion(caps):
1383 def obsmarkersversion(caps):
1384 """extract the list of supported obsmarkers versions from a bundle2caps dict
1384 """extract the list of supported obsmarkers versions from a bundle2caps dict
1385 """
1385 """
1386 obscaps = caps.get('obsmarkers', ())
1386 obscaps = caps.get('obsmarkers', ())
1387 return [int(c[1:]) for c in obscaps if c.startswith('V')]
1387 return [int(c[1:]) for c in obscaps if c.startswith('V')]
1388
1388
1389 def writenewbundle(ui, repo, source, filename, bundletype, outgoing, opts,
1389 def writenewbundle(ui, repo, source, filename, bundletype, outgoing, opts,
1390 vfs=None, compression=None, compopts=None):
1390 vfs=None, compression=None, compopts=None):
1391 if bundletype.startswith('HG10'):
1391 if bundletype.startswith('HG10'):
1392 cg = changegroup.getchangegroup(repo, source, outgoing, version='01')
1392 cg = changegroup.getchangegroup(repo, source, outgoing, version='01')
1393 return writebundle(ui, cg, filename, bundletype, vfs=vfs,
1393 return writebundle(ui, cg, filename, bundletype, vfs=vfs,
1394 compression=compression, compopts=compopts)
1394 compression=compression, compopts=compopts)
1395 elif not bundletype.startswith('HG20'):
1395 elif not bundletype.startswith('HG20'):
1396 raise error.ProgrammingError('unknown bundle type: %s' % bundletype)
1396 raise error.ProgrammingError('unknown bundle type: %s' % bundletype)
1397
1397
1398 caps = {}
1398 caps = {}
1399 if 'obsolescence' in opts:
1399 if 'obsolescence' in opts:
1400 caps['obsmarkers'] = ('V1',)
1400 caps['obsmarkers'] = ('V1',)
1401 bundle = bundle20(ui, caps)
1401 bundle = bundle20(ui, caps)
1402 bundle.setcompression(compression, compopts)
1402 bundle.setcompression(compression, compopts)
1403 _addpartsfromopts(ui, repo, bundle, source, outgoing, opts)
1403 _addpartsfromopts(ui, repo, bundle, source, outgoing, opts)
1404 chunkiter = bundle.getchunks()
1404 chunkiter = bundle.getchunks()
1405
1405
1406 return changegroup.writechunks(ui, chunkiter, filename, vfs=vfs)
1406 return changegroup.writechunks(ui, chunkiter, filename, vfs=vfs)
1407
1407
1408 def _addpartsfromopts(ui, repo, bundler, source, outgoing, opts):
1408 def _addpartsfromopts(ui, repo, bundler, source, outgoing, opts):
1409 # We should eventually reconcile this logic with the one behind
1409 # We should eventually reconcile this logic with the one behind
1410 # 'exchange.getbundle2partsgenerator'.
1410 # 'exchange.getbundle2partsgenerator'.
1411 #
1411 #
1412 # The type of input from 'getbundle' and 'writenewbundle' are a bit
1412 # The type of input from 'getbundle' and 'writenewbundle' are a bit
1413 # different right now. So we keep them separated for now for the sake of
1413 # different right now. So we keep them separated for now for the sake of
1414 # simplicity.
1414 # simplicity.
1415
1415
1416 # we always want a changegroup in such bundle
1416 # we always want a changegroup in such bundle
1417 cgversion = opts.get('cg.version')
1417 cgversion = opts.get('cg.version')
1418 if cgversion is None:
1418 if cgversion is None:
1419 cgversion = changegroup.safeversion(repo)
1419 cgversion = changegroup.safeversion(repo)
1420 cg = changegroup.getchangegroup(repo, source, outgoing,
1420 cg = changegroup.getchangegroup(repo, source, outgoing,
1421 version=cgversion)
1421 version=cgversion)
1422 part = bundler.newpart('changegroup', data=cg.getchunks())
1422 part = bundler.newpart('changegroup', data=cg.getchunks())
1423 part.addparam('version', cg.version)
1423 part.addparam('version', cg.version)
1424 if 'clcount' in cg.extras:
1424 if 'clcount' in cg.extras:
1425 part.addparam('nbchanges', str(cg.extras['clcount']),
1425 part.addparam('nbchanges', str(cg.extras['clcount']),
1426 mandatory=False)
1426 mandatory=False)
1427 if opts.get('phases') and repo.revs('%ln and secret()',
1427 if opts.get('phases') and repo.revs('%ln and secret()',
1428 outgoing.missingheads):
1428 outgoing.missingheads):
1429 part.addparam('targetphase', '%d' % phases.secret, mandatory=False)
1429 part.addparam('targetphase', '%d' % phases.secret, mandatory=False)
1430
1430
1431 addparttagsfnodescache(repo, bundler, outgoing)
1431 addparttagsfnodescache(repo, bundler, outgoing)
1432
1432
1433 if opts.get('obsolescence', False):
1433 if opts.get('obsolescence', False):
1434 obsmarkers = repo.obsstore.relevantmarkers(outgoing.missing)
1434 obsmarkers = repo.obsstore.relevantmarkers(outgoing.missing)
1435 buildobsmarkerspart(bundler, obsmarkers)
1435 buildobsmarkerspart(bundler, obsmarkers)
1436
1436
1437 if opts.get('phases', False):
1437 if opts.get('phases', False):
1438 headsbyphase = phases.subsetphaseheads(repo, outgoing.missing)
1438 headsbyphase = phases.subsetphaseheads(repo, outgoing.missing)
1439 phasedata = []
1439 phasedata = []
1440 for phase in phases.allphases:
1440 for phase in phases.allphases:
1441 for head in headsbyphase[phase]:
1441 for head in headsbyphase[phase]:
1442 phasedata.append(_pack(_fphasesentry, phase, head))
1442 phasedata.append(_pack(_fphasesentry, phase, head))
1443 bundler.newpart('phase-heads', data=''.join(phasedata))
1443 bundler.newpart('phase-heads', data=''.join(phasedata))
1444
1444
1445 def addparttagsfnodescache(repo, bundler, outgoing):
1445 def addparttagsfnodescache(repo, bundler, outgoing):
1446 # we include the tags fnode cache for the bundle changeset
1446 # we include the tags fnode cache for the bundle changeset
1447 # (as an optional parts)
1447 # (as an optional parts)
1448 cache = tags.hgtagsfnodescache(repo.unfiltered())
1448 cache = tags.hgtagsfnodescache(repo.unfiltered())
1449 chunks = []
1449 chunks = []
1450
1450
1451 # .hgtags fnodes are only relevant for head changesets. While we could
1451 # .hgtags fnodes are only relevant for head changesets. While we could
1452 # transfer values for all known nodes, there will likely be little to
1452 # transfer values for all known nodes, there will likely be little to
1453 # no benefit.
1453 # no benefit.
1454 #
1454 #
1455 # We don't bother using a generator to produce output data because
1455 # We don't bother using a generator to produce output data because
1456 # a) we only have 40 bytes per head and even esoteric numbers of heads
1456 # a) we only have 40 bytes per head and even esoteric numbers of heads
1457 # consume little memory (1M heads is 40MB) b) we don't want to send the
1457 # consume little memory (1M heads is 40MB) b) we don't want to send the
1458 # part if we don't have entries and knowing if we have entries requires
1458 # part if we don't have entries and knowing if we have entries requires
1459 # cache lookups.
1459 # cache lookups.
1460 for node in outgoing.missingheads:
1460 for node in outgoing.missingheads:
1461 # Don't compute missing, as this may slow down serving.
1461 # Don't compute missing, as this may slow down serving.
1462 fnode = cache.getfnode(node, computemissing=False)
1462 fnode = cache.getfnode(node, computemissing=False)
1463 if fnode is not None:
1463 if fnode is not None:
1464 chunks.extend([node, fnode])
1464 chunks.extend([node, fnode])
1465
1465
1466 if chunks:
1466 if chunks:
1467 bundler.newpart('hgtagsfnodes', data=''.join(chunks))
1467 bundler.newpart('hgtagsfnodes', data=''.join(chunks))
1468
1468
1469 def buildobsmarkerspart(bundler, markers):
1469 def buildobsmarkerspart(bundler, markers):
1470 """add an obsmarker part to the bundler with <markers>
1470 """add an obsmarker part to the bundler with <markers>
1471
1471
1472 No part is created if markers is empty.
1472 No part is created if markers is empty.
1473 Raises ValueError if the bundler doesn't support any known obsmarker format.
1473 Raises ValueError if the bundler doesn't support any known obsmarker format.
1474 """
1474 """
1475 if not markers:
1475 if not markers:
1476 return None
1476 return None
1477
1477
1478 remoteversions = obsmarkersversion(bundler.capabilities)
1478 remoteversions = obsmarkersversion(bundler.capabilities)
1479 version = obsolete.commonversion(remoteversions)
1479 version = obsolete.commonversion(remoteversions)
1480 if version is None:
1480 if version is None:
1481 raise ValueError('bundler does not support common obsmarker format')
1481 raise ValueError('bundler does not support common obsmarker format')
1482 stream = obsolete.encodemarkers(markers, True, version=version)
1482 stream = obsolete.encodemarkers(markers, True, version=version)
1483 return bundler.newpart('obsmarkers', data=stream)
1483 return bundler.newpart('obsmarkers', data=stream)
1484
1484
1485 def writebundle(ui, cg, filename, bundletype, vfs=None, compression=None,
1485 def writebundle(ui, cg, filename, bundletype, vfs=None, compression=None,
1486 compopts=None):
1486 compopts=None):
1487 """Write a bundle file and return its filename.
1487 """Write a bundle file and return its filename.
1488
1488
1489 Existing files will not be overwritten.
1489 Existing files will not be overwritten.
1490 If no filename is specified, a temporary file is created.
1490 If no filename is specified, a temporary file is created.
1491 bz2 compression can be turned off.
1491 bz2 compression can be turned off.
1492 The bundle file will be deleted in case of errors.
1492 The bundle file will be deleted in case of errors.
1493 """
1493 """
1494
1494
1495 if bundletype == "HG20":
1495 if bundletype == "HG20":
1496 bundle = bundle20(ui)
1496 bundle = bundle20(ui)
1497 bundle.setcompression(compression, compopts)
1497 bundle.setcompression(compression, compopts)
1498 part = bundle.newpart('changegroup', data=cg.getchunks())
1498 part = bundle.newpart('changegroup', data=cg.getchunks())
1499 part.addparam('version', cg.version)
1499 part.addparam('version', cg.version)
1500 if 'clcount' in cg.extras:
1500 if 'clcount' in cg.extras:
1501 part.addparam('nbchanges', str(cg.extras['clcount']),
1501 part.addparam('nbchanges', str(cg.extras['clcount']),
1502 mandatory=False)
1502 mandatory=False)
1503 chunkiter = bundle.getchunks()
1503 chunkiter = bundle.getchunks()
1504 else:
1504 else:
1505 # compression argument is only for the bundle2 case
1505 # compression argument is only for the bundle2 case
1506 assert compression is None
1506 assert compression is None
1507 if cg.version != '01':
1507 if cg.version != '01':
1508 raise error.Abort(_('old bundle types only supports v1 '
1508 raise error.Abort(_('old bundle types only supports v1 '
1509 'changegroups'))
1509 'changegroups'))
1510 header, comp = bundletypes[bundletype]
1510 header, comp = bundletypes[bundletype]
1511 if comp not in util.compengines.supportedbundletypes:
1511 if comp not in util.compengines.supportedbundletypes:
1512 raise error.Abort(_('unknown stream compression type: %s')
1512 raise error.Abort(_('unknown stream compression type: %s')
1513 % comp)
1513 % comp)
1514 compengine = util.compengines.forbundletype(comp)
1514 compengine = util.compengines.forbundletype(comp)
1515 def chunkiter():
1515 def chunkiter():
1516 yield header
1516 yield header
1517 for chunk in compengine.compressstream(cg.getchunks(), compopts):
1517 for chunk in compengine.compressstream(cg.getchunks(), compopts):
1518 yield chunk
1518 yield chunk
1519 chunkiter = chunkiter()
1519 chunkiter = chunkiter()
1520
1520
1521 # parse the changegroup data, otherwise we will block
1521 # parse the changegroup data, otherwise we will block
1522 # in case of sshrepo because we don't know the end of the stream
1522 # in case of sshrepo because we don't know the end of the stream
1523 return changegroup.writechunks(ui, chunkiter, filename, vfs=vfs)
1523 return changegroup.writechunks(ui, chunkiter, filename, vfs=vfs)
1524
1524
1525 def combinechangegroupresults(op):
1525 def combinechangegroupresults(op):
1526 """logic to combine 0 or more addchangegroup results into one"""
1526 """logic to combine 0 or more addchangegroup results into one"""
1527 results = [r.get('return', 0)
1527 results = [r.get('return', 0)
1528 for r in op.records['changegroup']]
1528 for r in op.records['changegroup']]
1529 changedheads = 0
1529 changedheads = 0
1530 result = 1
1530 result = 1
1531 for ret in results:
1531 for ret in results:
1532 # If any changegroup result is 0, return 0
1532 # If any changegroup result is 0, return 0
1533 if ret == 0:
1533 if ret == 0:
1534 result = 0
1534 result = 0
1535 break
1535 break
1536 if ret < -1:
1536 if ret < -1:
1537 changedheads += ret + 1
1537 changedheads += ret + 1
1538 elif ret > 1:
1538 elif ret > 1:
1539 changedheads += ret - 1
1539 changedheads += ret - 1
1540 if changedheads > 0:
1540 if changedheads > 0:
1541 result = 1 + changedheads
1541 result = 1 + changedheads
1542 elif changedheads < 0:
1542 elif changedheads < 0:
1543 result = -1 + changedheads
1543 result = -1 + changedheads
1544 return result
1544 return result
1545
1545
1546 @parthandler('changegroup', ('version', 'nbchanges', 'treemanifest',
1546 @parthandler('changegroup', ('version', 'nbchanges', 'treemanifest',
1547 'targetphase'))
1547 'targetphase'))
1548 def handlechangegroup(op, inpart):
1548 def handlechangegroup(op, inpart):
1549 """apply a changegroup part on the repo
1549 """apply a changegroup part on the repo
1550
1550
1551 This is a very early implementation that will massive rework before being
1551 This is a very early implementation that will massive rework before being
1552 inflicted to any end-user.
1552 inflicted to any end-user.
1553 """
1553 """
1554 tr = op.gettransaction()
1554 tr = op.gettransaction()
1555 unpackerversion = inpart.params.get('version', '01')
1555 unpackerversion = inpart.params.get('version', '01')
1556 # We should raise an appropriate exception here
1556 # We should raise an appropriate exception here
1557 cg = changegroup.getunbundler(unpackerversion, inpart, None)
1557 cg = changegroup.getunbundler(unpackerversion, inpart, None)
1558 # the source and url passed here are overwritten by the one contained in
1558 # the source and url passed here are overwritten by the one contained in
1559 # the transaction.hookargs argument. So 'bundle2' is a placeholder
1559 # the transaction.hookargs argument. So 'bundle2' is a placeholder
1560 nbchangesets = None
1560 nbchangesets = None
1561 if 'nbchanges' in inpart.params:
1561 if 'nbchanges' in inpart.params:
1562 nbchangesets = int(inpart.params.get('nbchanges'))
1562 nbchangesets = int(inpart.params.get('nbchanges'))
1563 if ('treemanifest' in inpart.params and
1563 if ('treemanifest' in inpart.params and
1564 'treemanifest' not in op.repo.requirements):
1564 'treemanifest' not in op.repo.requirements):
1565 if len(op.repo.changelog) != 0:
1565 if len(op.repo.changelog) != 0:
1566 raise error.Abort(_(
1566 raise error.Abort(_(
1567 "bundle contains tree manifests, but local repo is "
1567 "bundle contains tree manifests, but local repo is "
1568 "non-empty and does not use tree manifests"))
1568 "non-empty and does not use tree manifests"))
1569 op.repo.requirements.add('treemanifest')
1569 op.repo.requirements.add('treemanifest')
1570 op.repo._applyopenerreqs()
1570 op.repo._applyopenerreqs()
1571 op.repo._writerequirements()
1571 op.repo._writerequirements()
1572 extrakwargs = {}
1572 extrakwargs = {}
1573 targetphase = inpart.params.get('targetphase')
1573 targetphase = inpart.params.get('targetphase')
1574 if targetphase is not None:
1574 if targetphase is not None:
1575 extrakwargs['targetphase'] = int(targetphase)
1575 extrakwargs['targetphase'] = int(targetphase)
1576 ret = _processchangegroup(op, cg, tr, 'bundle2', 'bundle2',
1576 ret = _processchangegroup(op, cg, tr, 'bundle2', 'bundle2',
1577 expectedtotal=nbchangesets, **extrakwargs)
1577 expectedtotal=nbchangesets, **extrakwargs)
1578 if op.reply is not None:
1578 if op.reply is not None:
1579 # This is definitely not the final form of this
1579 # This is definitely not the final form of this
1580 # return. But one need to start somewhere.
1580 # return. But one need to start somewhere.
1581 part = op.reply.newpart('reply:changegroup', mandatory=False)
1581 part = op.reply.newpart('reply:changegroup', mandatory=False)
1582 part.addparam(
1582 part.addparam(
1583 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1583 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1584 part.addparam('return', '%i' % ret, mandatory=False)
1584 part.addparam('return', '%i' % ret, mandatory=False)
1585 assert not inpart.read()
1585 assert not inpart.read()
1586
1586
1587 _remotechangegroupparams = tuple(['url', 'size', 'digests'] +
1587 _remotechangegroupparams = tuple(['url', 'size', 'digests'] +
1588 ['digest:%s' % k for k in util.DIGESTS.keys()])
1588 ['digest:%s' % k for k in util.DIGESTS.keys()])
1589 @parthandler('remote-changegroup', _remotechangegroupparams)
1589 @parthandler('remote-changegroup', _remotechangegroupparams)
1590 def handleremotechangegroup(op, inpart):
1590 def handleremotechangegroup(op, inpart):
1591 """apply a bundle10 on the repo, given an url and validation information
1591 """apply a bundle10 on the repo, given an url and validation information
1592
1592
1593 All the information about the remote bundle to import are given as
1593 All the information about the remote bundle to import are given as
1594 parameters. The parameters include:
1594 parameters. The parameters include:
1595 - url: the url to the bundle10.
1595 - url: the url to the bundle10.
1596 - size: the bundle10 file size. It is used to validate what was
1596 - size: the bundle10 file size. It is used to validate what was
1597 retrieved by the client matches the server knowledge about the bundle.
1597 retrieved by the client matches the server knowledge about the bundle.
1598 - digests: a space separated list of the digest types provided as
1598 - digests: a space separated list of the digest types provided as
1599 parameters.
1599 parameters.
1600 - digest:<digest-type>: the hexadecimal representation of the digest with
1600 - digest:<digest-type>: the hexadecimal representation of the digest with
1601 that name. Like the size, it is used to validate what was retrieved by
1601 that name. Like the size, it is used to validate what was retrieved by
1602 the client matches what the server knows about the bundle.
1602 the client matches what the server knows about the bundle.
1603
1603
1604 When multiple digest types are given, all of them are checked.
1604 When multiple digest types are given, all of them are checked.
1605 """
1605 """
1606 try:
1606 try:
1607 raw_url = inpart.params['url']
1607 raw_url = inpart.params['url']
1608 except KeyError:
1608 except KeyError:
1609 raise error.Abort(_('remote-changegroup: missing "%s" param') % 'url')
1609 raise error.Abort(_('remote-changegroup: missing "%s" param') % 'url')
1610 parsed_url = util.url(raw_url)
1610 parsed_url = util.url(raw_url)
1611 if parsed_url.scheme not in capabilities['remote-changegroup']:
1611 if parsed_url.scheme not in capabilities['remote-changegroup']:
1612 raise error.Abort(_('remote-changegroup does not support %s urls') %
1612 raise error.Abort(_('remote-changegroup does not support %s urls') %
1613 parsed_url.scheme)
1613 parsed_url.scheme)
1614
1614
1615 try:
1615 try:
1616 size = int(inpart.params['size'])
1616 size = int(inpart.params['size'])
1617 except ValueError:
1617 except ValueError:
1618 raise error.Abort(_('remote-changegroup: invalid value for param "%s"')
1618 raise error.Abort(_('remote-changegroup: invalid value for param "%s"')
1619 % 'size')
1619 % 'size')
1620 except KeyError:
1620 except KeyError:
1621 raise error.Abort(_('remote-changegroup: missing "%s" param') % 'size')
1621 raise error.Abort(_('remote-changegroup: missing "%s" param') % 'size')
1622
1622
1623 digests = {}
1623 digests = {}
1624 for typ in inpart.params.get('digests', '').split():
1624 for typ in inpart.params.get('digests', '').split():
1625 param = 'digest:%s' % typ
1625 param = 'digest:%s' % typ
1626 try:
1626 try:
1627 value = inpart.params[param]
1627 value = inpart.params[param]
1628 except KeyError:
1628 except KeyError:
1629 raise error.Abort(_('remote-changegroup: missing "%s" param') %
1629 raise error.Abort(_('remote-changegroup: missing "%s" param') %
1630 param)
1630 param)
1631 digests[typ] = value
1631 digests[typ] = value
1632
1632
1633 real_part = util.digestchecker(url.open(op.ui, raw_url), size, digests)
1633 real_part = util.digestchecker(url.open(op.ui, raw_url), size, digests)
1634
1634
1635 tr = op.gettransaction()
1635 tr = op.gettransaction()
1636 from . import exchange
1636 from . import exchange
1637 cg = exchange.readbundle(op.repo.ui, real_part, raw_url)
1637 cg = exchange.readbundle(op.repo.ui, real_part, raw_url)
1638 if not isinstance(cg, changegroup.cg1unpacker):
1638 if not isinstance(cg, changegroup.cg1unpacker):
1639 raise error.Abort(_('%s: not a bundle version 1.0') %
1639 raise error.Abort(_('%s: not a bundle version 1.0') %
1640 util.hidepassword(raw_url))
1640 util.hidepassword(raw_url))
1641 ret = _processchangegroup(op, cg, tr, 'bundle2', 'bundle2')
1641 ret = _processchangegroup(op, cg, tr, 'bundle2', 'bundle2')
1642 if op.reply is not None:
1642 if op.reply is not None:
1643 # This is definitely not the final form of this
1643 # This is definitely not the final form of this
1644 # return. But one need to start somewhere.
1644 # return. But one need to start somewhere.
1645 part = op.reply.newpart('reply:changegroup')
1645 part = op.reply.newpart('reply:changegroup')
1646 part.addparam(
1646 part.addparam(
1647 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1647 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1648 part.addparam('return', '%i' % ret, mandatory=False)
1648 part.addparam('return', '%i' % ret, mandatory=False)
1649 try:
1649 try:
1650 real_part.validate()
1650 real_part.validate()
1651 except error.Abort as e:
1651 except error.Abort as e:
1652 raise error.Abort(_('bundle at %s is corrupted:\n%s') %
1652 raise error.Abort(_('bundle at %s is corrupted:\n%s') %
1653 (util.hidepassword(raw_url), str(e)))
1653 (util.hidepassword(raw_url), str(e)))
1654 assert not inpart.read()
1654 assert not inpart.read()
1655
1655
1656 @parthandler('reply:changegroup', ('return', 'in-reply-to'))
1656 @parthandler('reply:changegroup', ('return', 'in-reply-to'))
1657 def handlereplychangegroup(op, inpart):
1657 def handlereplychangegroup(op, inpart):
1658 ret = int(inpart.params['return'])
1658 ret = int(inpart.params['return'])
1659 replyto = int(inpart.params['in-reply-to'])
1659 replyto = int(inpart.params['in-reply-to'])
1660 op.records.add('changegroup', {'return': ret}, replyto)
1660 op.records.add('changegroup', {'return': ret}, replyto)
1661
1661
1662 @parthandler('check:heads')
1662 @parthandler('check:heads')
1663 def handlecheckheads(op, inpart):
1663 def handlecheckheads(op, inpart):
1664 """check that head of the repo did not change
1664 """check that head of the repo did not change
1665
1665
1666 This is used to detect a push race when using unbundle.
1666 This is used to detect a push race when using unbundle.
1667 This replaces the "heads" argument of unbundle."""
1667 This replaces the "heads" argument of unbundle."""
1668 h = inpart.read(20)
1668 h = inpart.read(20)
1669 heads = []
1669 heads = []
1670 while len(h) == 20:
1670 while len(h) == 20:
1671 heads.append(h)
1671 heads.append(h)
1672 h = inpart.read(20)
1672 h = inpart.read(20)
1673 assert not h
1673 assert not h
1674 # Trigger a transaction so that we are guaranteed to have the lock now.
1674 # Trigger a transaction so that we are guaranteed to have the lock now.
1675 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1675 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1676 op.gettransaction()
1676 op.gettransaction()
1677 if sorted(heads) != sorted(op.repo.heads()):
1677 if sorted(heads) != sorted(op.repo.heads()):
1678 raise error.PushRaced('repository changed while pushing - '
1678 raise error.PushRaced('repository changed while pushing - '
1679 'please try again')
1679 'please try again')
1680
1680
1681 @parthandler('check:updated-heads')
1681 @parthandler('check:updated-heads')
1682 def handlecheckupdatedheads(op, inpart):
1682 def handlecheckupdatedheads(op, inpart):
1683 """check for race on the heads touched by a push
1683 """check for race on the heads touched by a push
1684
1684
1685 This is similar to 'check:heads' but focus on the heads actually updated
1685 This is similar to 'check:heads' but focus on the heads actually updated
1686 during the push. If other activities happen on unrelated heads, it is
1686 during the push. If other activities happen on unrelated heads, it is
1687 ignored.
1687 ignored.
1688
1688
1689 This allow server with high traffic to avoid push contention as long as
1689 This allow server with high traffic to avoid push contention as long as
1690 unrelated parts of the graph are involved."""
1690 unrelated parts of the graph are involved."""
1691 h = inpart.read(20)
1691 h = inpart.read(20)
1692 heads = []
1692 heads = []
1693 while len(h) == 20:
1693 while len(h) == 20:
1694 heads.append(h)
1694 heads.append(h)
1695 h = inpart.read(20)
1695 h = inpart.read(20)
1696 assert not h
1696 assert not h
1697 # trigger a transaction so that we are guaranteed to have the lock now.
1697 # trigger a transaction so that we are guaranteed to have the lock now.
1698 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1698 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1699 op.gettransaction()
1699 op.gettransaction()
1700
1700
1701 currentheads = set()
1701 currentheads = set()
1702 for ls in op.repo.branchmap().itervalues():
1702 for ls in op.repo.branchmap().itervalues():
1703 currentheads.update(ls)
1703 currentheads.update(ls)
1704
1704
1705 for h in heads:
1705 for h in heads:
1706 if h not in currentheads:
1706 if h not in currentheads:
1707 raise error.PushRaced('repository changed while pushing - '
1707 raise error.PushRaced('repository changed while pushing - '
1708 'please try again')
1708 'please try again')
1709
1709
1710 @parthandler('output')
1710 @parthandler('output')
1711 def handleoutput(op, inpart):
1711 def handleoutput(op, inpart):
1712 """forward output captured on the server to the client"""
1712 """forward output captured on the server to the client"""
1713 for line in inpart.read().splitlines():
1713 for line in inpart.read().splitlines():
1714 op.ui.status(_('remote: %s\n') % line)
1714 op.ui.status(_('remote: %s\n') % line)
1715
1715
1716 @parthandler('replycaps')
1716 @parthandler('replycaps')
1717 def handlereplycaps(op, inpart):
1717 def handlereplycaps(op, inpart):
1718 """Notify that a reply bundle should be created
1718 """Notify that a reply bundle should be created
1719
1719
1720 The payload contains the capabilities information for the reply"""
1720 The payload contains the capabilities information for the reply"""
1721 caps = decodecaps(inpart.read())
1721 caps = decodecaps(inpart.read())
1722 if op.reply is None:
1722 if op.reply is None:
1723 op.reply = bundle20(op.ui, caps)
1723 op.reply = bundle20(op.ui, caps)
1724
1724
1725 class AbortFromPart(error.Abort):
1725 class AbortFromPart(error.Abort):
1726 """Sub-class of Abort that denotes an error from a bundle2 part."""
1726 """Sub-class of Abort that denotes an error from a bundle2 part."""
1727
1727
1728 @parthandler('error:abort', ('message', 'hint'))
1728 @parthandler('error:abort', ('message', 'hint'))
1729 def handleerrorabort(op, inpart):
1729 def handleerrorabort(op, inpart):
1730 """Used to transmit abort error over the wire"""
1730 """Used to transmit abort error over the wire"""
1731 raise AbortFromPart(inpart.params['message'],
1731 raise AbortFromPart(inpart.params['message'],
1732 hint=inpart.params.get('hint'))
1732 hint=inpart.params.get('hint'))
1733
1733
1734 @parthandler('error:pushkey', ('namespace', 'key', 'new', 'old', 'ret',
1734 @parthandler('error:pushkey', ('namespace', 'key', 'new', 'old', 'ret',
1735 'in-reply-to'))
1735 'in-reply-to'))
1736 def handleerrorpushkey(op, inpart):
1736 def handleerrorpushkey(op, inpart):
1737 """Used to transmit failure of a mandatory pushkey over the wire"""
1737 """Used to transmit failure of a mandatory pushkey over the wire"""
1738 kwargs = {}
1738 kwargs = {}
1739 for name in ('namespace', 'key', 'new', 'old', 'ret'):
1739 for name in ('namespace', 'key', 'new', 'old', 'ret'):
1740 value = inpart.params.get(name)
1740 value = inpart.params.get(name)
1741 if value is not None:
1741 if value is not None:
1742 kwargs[name] = value
1742 kwargs[name] = value
1743 raise error.PushkeyFailed(inpart.params['in-reply-to'], **kwargs)
1743 raise error.PushkeyFailed(inpart.params['in-reply-to'], **kwargs)
1744
1744
1745 @parthandler('error:unsupportedcontent', ('parttype', 'params'))
1745 @parthandler('error:unsupportedcontent', ('parttype', 'params'))
1746 def handleerrorunsupportedcontent(op, inpart):
1746 def handleerrorunsupportedcontent(op, inpart):
1747 """Used to transmit unknown content error over the wire"""
1747 """Used to transmit unknown content error over the wire"""
1748 kwargs = {}
1748 kwargs = {}
1749 parttype = inpart.params.get('parttype')
1749 parttype = inpart.params.get('parttype')
1750 if parttype is not None:
1750 if parttype is not None:
1751 kwargs['parttype'] = parttype
1751 kwargs['parttype'] = parttype
1752 params = inpart.params.get('params')
1752 params = inpart.params.get('params')
1753 if params is not None:
1753 if params is not None:
1754 kwargs['params'] = params.split('\0')
1754 kwargs['params'] = params.split('\0')
1755
1755
1756 raise error.BundleUnknownFeatureError(**kwargs)
1756 raise error.BundleUnknownFeatureError(**kwargs)
1757
1757
1758 @parthandler('error:pushraced', ('message',))
1758 @parthandler('error:pushraced', ('message',))
1759 def handleerrorpushraced(op, inpart):
1759 def handleerrorpushraced(op, inpart):
1760 """Used to transmit push race error over the wire"""
1760 """Used to transmit push race error over the wire"""
1761 raise error.ResponseError(_('push failed:'), inpart.params['message'])
1761 raise error.ResponseError(_('push failed:'), inpart.params['message'])
1762
1762
1763 @parthandler('listkeys', ('namespace',))
1763 @parthandler('listkeys', ('namespace',))
1764 def handlelistkeys(op, inpart):
1764 def handlelistkeys(op, inpart):
1765 """retrieve pushkey namespace content stored in a bundle2"""
1765 """retrieve pushkey namespace content stored in a bundle2"""
1766 namespace = inpart.params['namespace']
1766 namespace = inpart.params['namespace']
1767 r = pushkey.decodekeys(inpart.read())
1767 r = pushkey.decodekeys(inpart.read())
1768 op.records.add('listkeys', (namespace, r))
1768 op.records.add('listkeys', (namespace, r))
1769
1769
1770 @parthandler('pushkey', ('namespace', 'key', 'old', 'new'))
1770 @parthandler('pushkey', ('namespace', 'key', 'old', 'new'))
1771 def handlepushkey(op, inpart):
1771 def handlepushkey(op, inpart):
1772 """process a pushkey request"""
1772 """process a pushkey request"""
1773 dec = pushkey.decode
1773 dec = pushkey.decode
1774 namespace = dec(inpart.params['namespace'])
1774 namespace = dec(inpart.params['namespace'])
1775 key = dec(inpart.params['key'])
1775 key = dec(inpart.params['key'])
1776 old = dec(inpart.params['old'])
1776 old = dec(inpart.params['old'])
1777 new = dec(inpart.params['new'])
1777 new = dec(inpart.params['new'])
1778 # Grab the transaction to ensure that we have the lock before performing the
1778 # Grab the transaction to ensure that we have the lock before performing the
1779 # pushkey.
1779 # pushkey.
1780 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1780 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1781 op.gettransaction()
1781 op.gettransaction()
1782 ret = op.repo.pushkey(namespace, key, old, new)
1782 ret = op.repo.pushkey(namespace, key, old, new)
1783 record = {'namespace': namespace,
1783 record = {'namespace': namespace,
1784 'key': key,
1784 'key': key,
1785 'old': old,
1785 'old': old,
1786 'new': new}
1786 'new': new}
1787 op.records.add('pushkey', record)
1787 op.records.add('pushkey', record)
1788 if op.reply is not None:
1788 if op.reply is not None:
1789 rpart = op.reply.newpart('reply:pushkey')
1789 rpart = op.reply.newpart('reply:pushkey')
1790 rpart.addparam(
1790 rpart.addparam(
1791 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1791 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1792 rpart.addparam('return', '%i' % ret, mandatory=False)
1792 rpart.addparam('return', '%i' % ret, mandatory=False)
1793 if inpart.mandatory and not ret:
1793 if inpart.mandatory and not ret:
1794 kwargs = {}
1794 kwargs = {}
1795 for key in ('namespace', 'key', 'new', 'old', 'ret'):
1795 for key in ('namespace', 'key', 'new', 'old', 'ret'):
1796 if key in inpart.params:
1796 if key in inpart.params:
1797 kwargs[key] = inpart.params[key]
1797 kwargs[key] = inpart.params[key]
1798 raise error.PushkeyFailed(partid=str(inpart.id), **kwargs)
1798 raise error.PushkeyFailed(partid=str(inpart.id), **kwargs)
1799
1799
1800 def _readphaseheads(inpart):
1800 def _readphaseheads(inpart):
1801 headsbyphase = [[] for i in phases.allphases]
1801 headsbyphase = [[] for i in phases.allphases]
1802 entrysize = struct.calcsize(_fphasesentry)
1802 entrysize = struct.calcsize(_fphasesentry)
1803 while True:
1803 while True:
1804 entry = inpart.read(entrysize)
1804 entry = inpart.read(entrysize)
1805 if len(entry) < entrysize:
1805 if len(entry) < entrysize:
1806 if entry:
1806 if entry:
1807 raise error.Abort(_('bad phase-heads bundle part'))
1807 raise error.Abort(_('bad phase-heads bundle part'))
1808 break
1808 break
1809 phase, node = struct.unpack(_fphasesentry, entry)
1809 phase, node = struct.unpack(_fphasesentry, entry)
1810 headsbyphase[phase].append(node)
1810 headsbyphase[phase].append(node)
1811 return headsbyphase
1811 return headsbyphase
1812
1812
1813 @parthandler('phase-heads')
1813 @parthandler('phase-heads')
1814 def handlephases(op, inpart):
1814 def handlephases(op, inpart):
1815 """apply phases from bundle part to repo"""
1815 """apply phases from bundle part to repo"""
1816 headsbyphase = _readphaseheads(inpart)
1816 headsbyphase = _readphaseheads(inpart)
1817 phases.updatephases(op.repo.unfiltered(), op.gettransaction(), headsbyphase)
1817 phases.updatephases(op.repo.unfiltered(), op.gettransaction(), headsbyphase)
1818
1818
1819 @parthandler('reply:pushkey', ('return', 'in-reply-to'))
1819 @parthandler('reply:pushkey', ('return', 'in-reply-to'))
1820 def handlepushkeyreply(op, inpart):
1820 def handlepushkeyreply(op, inpart):
1821 """retrieve the result of a pushkey request"""
1821 """retrieve the result of a pushkey request"""
1822 ret = int(inpart.params['return'])
1822 ret = int(inpart.params['return'])
1823 partid = int(inpart.params['in-reply-to'])
1823 partid = int(inpart.params['in-reply-to'])
1824 op.records.add('pushkey', {'return': ret}, partid)
1824 op.records.add('pushkey', {'return': ret}, partid)
1825
1825
1826 @parthandler('obsmarkers')
1826 @parthandler('obsmarkers')
1827 def handleobsmarker(op, inpart):
1827 def handleobsmarker(op, inpart):
1828 """add a stream of obsmarkers to the repo"""
1828 """add a stream of obsmarkers to the repo"""
1829 tr = op.gettransaction()
1829 tr = op.gettransaction()
1830 markerdata = inpart.read()
1830 markerdata = inpart.read()
1831 if op.ui.config('experimental', 'obsmarkers-exchange-debug'):
1831 if op.ui.config('experimental', 'obsmarkers-exchange-debug'):
1832 op.ui.write(('obsmarker-exchange: %i bytes received\n')
1832 op.ui.write(('obsmarker-exchange: %i bytes received\n')
1833 % len(markerdata))
1833 % len(markerdata))
1834 # The mergemarkers call will crash if marker creation is not enabled.
1834 # The mergemarkers call will crash if marker creation is not enabled.
1835 # we want to avoid this if the part is advisory.
1835 # we want to avoid this if the part is advisory.
1836 if not inpart.mandatory and op.repo.obsstore.readonly:
1836 if not inpart.mandatory and op.repo.obsstore.readonly:
1837 op.repo.ui.debug('ignoring obsolescence markers, feature not enabled')
1837 op.repo.ui.debug('ignoring obsolescence markers, feature not enabled')
1838 return
1838 return
1839 new = op.repo.obsstore.mergemarkers(tr, markerdata)
1839 new = op.repo.obsstore.mergemarkers(tr, markerdata)
1840 op.repo.invalidatevolatilesets()
1840 op.repo.invalidatevolatilesets()
1841 if new:
1841 if new:
1842 op.repo.ui.status(_('%i new obsolescence markers\n') % new)
1842 op.repo.ui.status(_('%i new obsolescence markers\n') % new)
1843 op.records.add('obsmarkers', {'new': new})
1843 op.records.add('obsmarkers', {'new': new})
1844 if op.reply is not None:
1844 if op.reply is not None:
1845 rpart = op.reply.newpart('reply:obsmarkers')
1845 rpart = op.reply.newpart('reply:obsmarkers')
1846 rpart.addparam(
1846 rpart.addparam(
1847 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1847 'in-reply-to', pycompat.bytestr(inpart.id), mandatory=False)
1848 rpart.addparam('new', '%i' % new, mandatory=False)
1848 rpart.addparam('new', '%i' % new, mandatory=False)
1849
1849
1850
1850
1851 @parthandler('reply:obsmarkers', ('new', 'in-reply-to'))
1851 @parthandler('reply:obsmarkers', ('new', 'in-reply-to'))
1852 def handleobsmarkerreply(op, inpart):
1852 def handleobsmarkerreply(op, inpart):
1853 """retrieve the result of a pushkey request"""
1853 """retrieve the result of a pushkey request"""
1854 ret = int(inpart.params['new'])
1854 ret = int(inpart.params['new'])
1855 partid = int(inpart.params['in-reply-to'])
1855 partid = int(inpart.params['in-reply-to'])
1856 op.records.add('obsmarkers', {'new': ret}, partid)
1856 op.records.add('obsmarkers', {'new': ret}, partid)
1857
1857
1858 @parthandler('hgtagsfnodes')
1858 @parthandler('hgtagsfnodes')
1859 def handlehgtagsfnodes(op, inpart):
1859 def handlehgtagsfnodes(op, inpart):
1860 """Applies .hgtags fnodes cache entries to the local repo.
1860 """Applies .hgtags fnodes cache entries to the local repo.
1861
1861
1862 Payload is pairs of 20 byte changeset nodes and filenodes.
1862 Payload is pairs of 20 byte changeset nodes and filenodes.
1863 """
1863 """
1864 # Grab the transaction so we ensure that we have the lock at this point.
1864 # Grab the transaction so we ensure that we have the lock at this point.
1865 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1865 if op.ui.configbool('experimental', 'bundle2lazylocking'):
1866 op.gettransaction()
1866 op.gettransaction()
1867 cache = tags.hgtagsfnodescache(op.repo.unfiltered())
1867 cache = tags.hgtagsfnodescache(op.repo.unfiltered())
1868
1868
1869 count = 0
1869 count = 0
1870 while True:
1870 while True:
1871 node = inpart.read(20)
1871 node = inpart.read(20)
1872 fnode = inpart.read(20)
1872 fnode = inpart.read(20)
1873 if len(node) < 20 or len(fnode) < 20:
1873 if len(node) < 20 or len(fnode) < 20:
1874 op.ui.debug('ignoring incomplete received .hgtags fnodes data\n')
1874 op.ui.debug('ignoring incomplete received .hgtags fnodes data\n')
1875 break
1875 break
1876 cache.setfnode(node, fnode)
1876 cache.setfnode(node, fnode)
1877 count += 1
1877 count += 1
1878
1878
1879 cache.write()
1879 cache.write()
1880 op.ui.debug('applied %i hgtags fnodes cache entries\n' % count)
1880 op.ui.debug('applied %i hgtags fnodes cache entries\n' % count)
1881
1881
1882 @parthandler('pushvars')
1882 @parthandler('pushvars')
1883 def bundle2getvars(op, part):
1883 def bundle2getvars(op, part):
1884 '''unbundle a bundle2 containing shellvars on the server'''
1884 '''unbundle a bundle2 containing shellvars on the server'''
1885 # An option to disable unbundling on server-side for security reasons
1885 # An option to disable unbundling on server-side for security reasons
1886 if op.ui.configbool('push', 'pushvars.server', False):
1886 if op.ui.configbool('push', 'pushvars.server'):
1887 hookargs = {}
1887 hookargs = {}
1888 for key, value in part.advisoryparams:
1888 for key, value in part.advisoryparams:
1889 key = key.upper()
1889 key = key.upper()
1890 # We want pushed variables to have USERVAR_ prepended so we know
1890 # We want pushed variables to have USERVAR_ prepended so we know
1891 # they came from the --pushvar flag.
1891 # they came from the --pushvar flag.
1892 key = "USERVAR_" + key
1892 key = "USERVAR_" + key
1893 hookargs[key] = value
1893 hookargs[key] = value
1894 op.addhookargs(hookargs)
1894 op.addhookargs(hookargs)
@@ -1,595 +1,598
1 # configitems.py - centralized declaration of configuration option
1 # configitems.py - centralized declaration of configuration option
2 #
2 #
3 # Copyright 2017 Pierre-Yves David <pierre-yves.david@octobus.net>
3 # Copyright 2017 Pierre-Yves David <pierre-yves.david@octobus.net>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from __future__ import absolute_import
8 from __future__ import absolute_import
9
9
10 import functools
10 import functools
11
11
12 from . import (
12 from . import (
13 error,
13 error,
14 )
14 )
15
15
16 def loadconfigtable(ui, extname, configtable):
16 def loadconfigtable(ui, extname, configtable):
17 """update config item known to the ui with the extension ones"""
17 """update config item known to the ui with the extension ones"""
18 for section, items in configtable.items():
18 for section, items in configtable.items():
19 knownitems = ui._knownconfig.setdefault(section, {})
19 knownitems = ui._knownconfig.setdefault(section, {})
20 knownkeys = set(knownitems)
20 knownkeys = set(knownitems)
21 newkeys = set(items)
21 newkeys = set(items)
22 for key in sorted(knownkeys & newkeys):
22 for key in sorted(knownkeys & newkeys):
23 msg = "extension '%s' overwrite config item '%s.%s'"
23 msg = "extension '%s' overwrite config item '%s.%s'"
24 msg %= (extname, section, key)
24 msg %= (extname, section, key)
25 ui.develwarn(msg, config='warn-config')
25 ui.develwarn(msg, config='warn-config')
26
26
27 knownitems.update(items)
27 knownitems.update(items)
28
28
29 class configitem(object):
29 class configitem(object):
30 """represent a known config item
30 """represent a known config item
31
31
32 :section: the official config section where to find this item,
32 :section: the official config section where to find this item,
33 :name: the official name within the section,
33 :name: the official name within the section,
34 :default: default value for this item,
34 :default: default value for this item,
35 :alias: optional list of tuples as alternatives.
35 :alias: optional list of tuples as alternatives.
36 """
36 """
37
37
38 def __init__(self, section, name, default=None, alias=()):
38 def __init__(self, section, name, default=None, alias=()):
39 self.section = section
39 self.section = section
40 self.name = name
40 self.name = name
41 self.default = default
41 self.default = default
42 self.alias = list(alias)
42 self.alias = list(alias)
43
43
44 coreitems = {}
44 coreitems = {}
45
45
46 def _register(configtable, *args, **kwargs):
46 def _register(configtable, *args, **kwargs):
47 item = configitem(*args, **kwargs)
47 item = configitem(*args, **kwargs)
48 section = configtable.setdefault(item.section, {})
48 section = configtable.setdefault(item.section, {})
49 if item.name in section:
49 if item.name in section:
50 msg = "duplicated config item registration for '%s.%s'"
50 msg = "duplicated config item registration for '%s.%s'"
51 raise error.ProgrammingError(msg % (item.section, item.name))
51 raise error.ProgrammingError(msg % (item.section, item.name))
52 section[item.name] = item
52 section[item.name] = item
53
53
54 # special value for case where the default is derived from other values
54 # special value for case where the default is derived from other values
55 dynamicdefault = object()
55 dynamicdefault = object()
56
56
57 # Registering actual config items
57 # Registering actual config items
58
58
59 def getitemregister(configtable):
59 def getitemregister(configtable):
60 return functools.partial(_register, configtable)
60 return functools.partial(_register, configtable)
61
61
62 coreconfigitem = getitemregister(coreitems)
62 coreconfigitem = getitemregister(coreitems)
63
63
64 coreconfigitem('auth', 'cookiefile',
64 coreconfigitem('auth', 'cookiefile',
65 default=None,
65 default=None,
66 )
66 )
67 # bookmarks.pushing: internal hack for discovery
67 # bookmarks.pushing: internal hack for discovery
68 coreconfigitem('bookmarks', 'pushing',
68 coreconfigitem('bookmarks', 'pushing',
69 default=list,
69 default=list,
70 )
70 )
71 # bundle.mainreporoot: internal hack for bundlerepo
71 # bundle.mainreporoot: internal hack for bundlerepo
72 coreconfigitem('bundle', 'mainreporoot',
72 coreconfigitem('bundle', 'mainreporoot',
73 default='',
73 default='',
74 )
74 )
75 # bundle.reorder: experimental config
75 # bundle.reorder: experimental config
76 coreconfigitem('bundle', 'reorder',
76 coreconfigitem('bundle', 'reorder',
77 default='auto',
77 default='auto',
78 )
78 )
79 coreconfigitem('censor', 'policy',
79 coreconfigitem('censor', 'policy',
80 default='abort',
80 default='abort',
81 )
81 )
82 coreconfigitem('chgserver', 'idletimeout',
82 coreconfigitem('chgserver', 'idletimeout',
83 default=3600,
83 default=3600,
84 )
84 )
85 coreconfigitem('chgserver', 'skiphash',
85 coreconfigitem('chgserver', 'skiphash',
86 default=False,
86 default=False,
87 )
87 )
88 coreconfigitem('cmdserver', 'log',
88 coreconfigitem('cmdserver', 'log',
89 default=None,
89 default=None,
90 )
90 )
91 coreconfigitem('color', 'mode',
91 coreconfigitem('color', 'mode',
92 default='auto',
92 default='auto',
93 )
93 )
94 coreconfigitem('color', 'pagermode',
94 coreconfigitem('color', 'pagermode',
95 default=dynamicdefault,
95 default=dynamicdefault,
96 )
96 )
97 coreconfigitem('commands', 'status.relative',
97 coreconfigitem('commands', 'status.relative',
98 default=False,
98 default=False,
99 )
99 )
100 coreconfigitem('commands', 'status.skipstates',
100 coreconfigitem('commands', 'status.skipstates',
101 default=[],
101 default=[],
102 )
102 )
103 coreconfigitem('commands', 'status.verbose',
103 coreconfigitem('commands', 'status.verbose',
104 default=False,
104 default=False,
105 )
105 )
106 coreconfigitem('commands', 'update.requiredest',
106 coreconfigitem('commands', 'update.requiredest',
107 default=False,
107 default=False,
108 )
108 )
109 coreconfigitem('devel', 'all-warnings',
109 coreconfigitem('devel', 'all-warnings',
110 default=False,
110 default=False,
111 )
111 )
112 coreconfigitem('devel', 'bundle2.debug',
112 coreconfigitem('devel', 'bundle2.debug',
113 default=False,
113 default=False,
114 )
114 )
115 coreconfigitem('devel', 'check-locks',
115 coreconfigitem('devel', 'check-locks',
116 default=False,
116 default=False,
117 )
117 )
118 coreconfigitem('devel', 'check-relroot',
118 coreconfigitem('devel', 'check-relroot',
119 default=False,
119 default=False,
120 )
120 )
121 coreconfigitem('devel', 'default-date',
121 coreconfigitem('devel', 'default-date',
122 default=None,
122 default=None,
123 )
123 )
124 coreconfigitem('devel', 'deprec-warn',
124 coreconfigitem('devel', 'deprec-warn',
125 default=False,
125 default=False,
126 )
126 )
127 coreconfigitem('devel', 'disableloaddefaultcerts',
127 coreconfigitem('devel', 'disableloaddefaultcerts',
128 default=False,
128 default=False,
129 )
129 )
130 coreconfigitem('devel', 'legacy.exchange',
130 coreconfigitem('devel', 'legacy.exchange',
131 default=list,
131 default=list,
132 )
132 )
133 coreconfigitem('devel', 'servercafile',
133 coreconfigitem('devel', 'servercafile',
134 default='',
134 default='',
135 )
135 )
136 coreconfigitem('devel', 'serverexactprotocol',
136 coreconfigitem('devel', 'serverexactprotocol',
137 default='',
137 default='',
138 )
138 )
139 coreconfigitem('devel', 'serverrequirecert',
139 coreconfigitem('devel', 'serverrequirecert',
140 default=False,
140 default=False,
141 )
141 )
142 coreconfigitem('devel', 'strip-obsmarkers',
142 coreconfigitem('devel', 'strip-obsmarkers',
143 default=True,
143 default=True,
144 )
144 )
145 coreconfigitem('email', 'charsets',
145 coreconfigitem('email', 'charsets',
146 default=list,
146 default=list,
147 )
147 )
148 coreconfigitem('email', 'method',
148 coreconfigitem('email', 'method',
149 default='smtp',
149 default='smtp',
150 )
150 )
151 coreconfigitem('experimental', 'bundle-phases',
151 coreconfigitem('experimental', 'bundle-phases',
152 default=False,
152 default=False,
153 )
153 )
154 coreconfigitem('experimental', 'bundle2-advertise',
154 coreconfigitem('experimental', 'bundle2-advertise',
155 default=True,
155 default=True,
156 )
156 )
157 coreconfigitem('experimental', 'bundle2-output-capture',
157 coreconfigitem('experimental', 'bundle2-output-capture',
158 default=False,
158 default=False,
159 )
159 )
160 coreconfigitem('experimental', 'bundle2.pushback',
160 coreconfigitem('experimental', 'bundle2.pushback',
161 default=False,
161 default=False,
162 )
162 )
163 coreconfigitem('experimental', 'bundle2lazylocking',
163 coreconfigitem('experimental', 'bundle2lazylocking',
164 default=False,
164 default=False,
165 )
165 )
166 coreconfigitem('experimental', 'bundlecomplevel',
166 coreconfigitem('experimental', 'bundlecomplevel',
167 default=None,
167 default=None,
168 )
168 )
169 coreconfigitem('experimental', 'changegroup3',
169 coreconfigitem('experimental', 'changegroup3',
170 default=False,
170 default=False,
171 )
171 )
172 coreconfigitem('experimental', 'clientcompressionengines',
172 coreconfigitem('experimental', 'clientcompressionengines',
173 default=list,
173 default=list,
174 )
174 )
175 coreconfigitem('experimental', 'crecordtest',
175 coreconfigitem('experimental', 'crecordtest',
176 default=None,
176 default=None,
177 )
177 )
178 coreconfigitem('experimental', 'disablecopytrace',
178 coreconfigitem('experimental', 'disablecopytrace',
179 default=False,
179 default=False,
180 )
180 )
181 coreconfigitem('experimental', 'editortmpinhg',
181 coreconfigitem('experimental', 'editortmpinhg',
182 default=False,
182 default=False,
183 )
183 )
184 coreconfigitem('experimental', 'stabilization',
184 coreconfigitem('experimental', 'stabilization',
185 default=list,
185 default=list,
186 alias=[('experimental', 'evolution')],
186 alias=[('experimental', 'evolution')],
187 )
187 )
188 coreconfigitem('experimental', 'stabilization.bundle-obsmarker',
188 coreconfigitem('experimental', 'stabilization.bundle-obsmarker',
189 default=False,
189 default=False,
190 alias=[('experimental', 'evolution.bundle-obsmarker')],
190 alias=[('experimental', 'evolution.bundle-obsmarker')],
191 )
191 )
192 coreconfigitem('experimental', 'stabilization.track-operation',
192 coreconfigitem('experimental', 'stabilization.track-operation',
193 default=False,
193 default=False,
194 alias=[('experimental', 'evolution.track-operation')]
194 alias=[('experimental', 'evolution.track-operation')]
195 )
195 )
196 coreconfigitem('experimental', 'exportableenviron',
196 coreconfigitem('experimental', 'exportableenviron',
197 default=list,
197 default=list,
198 )
198 )
199 coreconfigitem('experimental', 'extendedheader.index',
199 coreconfigitem('experimental', 'extendedheader.index',
200 default=None,
200 default=None,
201 )
201 )
202 coreconfigitem('experimental', 'extendedheader.similarity',
202 coreconfigitem('experimental', 'extendedheader.similarity',
203 default=False,
203 default=False,
204 )
204 )
205 coreconfigitem('experimental', 'format.compression',
205 coreconfigitem('experimental', 'format.compression',
206 default='zlib',
206 default='zlib',
207 )
207 )
208 coreconfigitem('experimental', 'graphshorten',
208 coreconfigitem('experimental', 'graphshorten',
209 default=False,
209 default=False,
210 )
210 )
211 coreconfigitem('experimental', 'hook-track-tags',
211 coreconfigitem('experimental', 'hook-track-tags',
212 default=False,
212 default=False,
213 )
213 )
214 coreconfigitem('experimental', 'httppostargs',
214 coreconfigitem('experimental', 'httppostargs',
215 default=False,
215 default=False,
216 )
216 )
217 coreconfigitem('experimental', 'manifestv2',
217 coreconfigitem('experimental', 'manifestv2',
218 default=False,
218 default=False,
219 )
219 )
220 coreconfigitem('experimental', 'mergedriver',
220 coreconfigitem('experimental', 'mergedriver',
221 default=None,
221 default=None,
222 )
222 )
223 coreconfigitem('experimental', 'obsmarkers-exchange-debug',
223 coreconfigitem('experimental', 'obsmarkers-exchange-debug',
224 default=False,
224 default=False,
225 )
225 )
226 coreconfigitem('experimental', 'revertalternateinteractivemode',
226 coreconfigitem('experimental', 'revertalternateinteractivemode',
227 default=True,
227 default=True,
228 )
228 )
229 coreconfigitem('experimental', 'revlogv2',
229 coreconfigitem('experimental', 'revlogv2',
230 default=None,
230 default=None,
231 )
231 )
232 coreconfigitem('experimental', 'spacemovesdown',
232 coreconfigitem('experimental', 'spacemovesdown',
233 default=False,
233 default=False,
234 )
234 )
235 coreconfigitem('experimental', 'treemanifest',
235 coreconfigitem('experimental', 'treemanifest',
236 default=False,
236 default=False,
237 )
237 )
238 coreconfigitem('experimental', 'updatecheck',
238 coreconfigitem('experimental', 'updatecheck',
239 default=None,
239 default=None,
240 )
240 )
241 coreconfigitem('format', 'aggressivemergedeltas',
241 coreconfigitem('format', 'aggressivemergedeltas',
242 default=False,
242 default=False,
243 )
243 )
244 coreconfigitem('format', 'chunkcachesize',
244 coreconfigitem('format', 'chunkcachesize',
245 default=None,
245 default=None,
246 )
246 )
247 coreconfigitem('format', 'dotencode',
247 coreconfigitem('format', 'dotencode',
248 default=True,
248 default=True,
249 )
249 )
250 coreconfigitem('format', 'generaldelta',
250 coreconfigitem('format', 'generaldelta',
251 default=False,
251 default=False,
252 )
252 )
253 coreconfigitem('format', 'manifestcachesize',
253 coreconfigitem('format', 'manifestcachesize',
254 default=None,
254 default=None,
255 )
255 )
256 coreconfigitem('format', 'maxchainlen',
256 coreconfigitem('format', 'maxchainlen',
257 default=None,
257 default=None,
258 )
258 )
259 coreconfigitem('format', 'obsstore-version',
259 coreconfigitem('format', 'obsstore-version',
260 default=None,
260 default=None,
261 )
261 )
262 coreconfigitem('format', 'usefncache',
262 coreconfigitem('format', 'usefncache',
263 default=True,
263 default=True,
264 )
264 )
265 coreconfigitem('format', 'usegeneraldelta',
265 coreconfigitem('format', 'usegeneraldelta',
266 default=True,
266 default=True,
267 )
267 )
268 coreconfigitem('format', 'usestore',
268 coreconfigitem('format', 'usestore',
269 default=True,
269 default=True,
270 )
270 )
271 coreconfigitem('hostsecurity', 'ciphers',
271 coreconfigitem('hostsecurity', 'ciphers',
272 default=None,
272 default=None,
273 )
273 )
274 coreconfigitem('hostsecurity', 'disabletls10warning',
274 coreconfigitem('hostsecurity', 'disabletls10warning',
275 default=False,
275 default=False,
276 )
276 )
277 coreconfigitem('http_proxy', 'always',
277 coreconfigitem('http_proxy', 'always',
278 default=False,
278 default=False,
279 )
279 )
280 coreconfigitem('http_proxy', 'host',
280 coreconfigitem('http_proxy', 'host',
281 default=None,
281 default=None,
282 )
282 )
283 coreconfigitem('http_proxy', 'no',
283 coreconfigitem('http_proxy', 'no',
284 default=list,
284 default=list,
285 )
285 )
286 coreconfigitem('http_proxy', 'passwd',
286 coreconfigitem('http_proxy', 'passwd',
287 default=None,
287 default=None,
288 )
288 )
289 coreconfigitem('http_proxy', 'user',
289 coreconfigitem('http_proxy', 'user',
290 default=None,
290 default=None,
291 )
291 )
292 coreconfigitem('merge', 'followcopies',
292 coreconfigitem('merge', 'followcopies',
293 default=True,
293 default=True,
294 )
294 )
295 coreconfigitem('pager', 'ignore',
295 coreconfigitem('pager', 'ignore',
296 default=list,
296 default=list,
297 )
297 )
298 coreconfigitem('patch', 'eol',
298 coreconfigitem('patch', 'eol',
299 default='strict',
299 default='strict',
300 )
300 )
301 coreconfigitem('patch', 'fuzz',
301 coreconfigitem('patch', 'fuzz',
302 default=2,
302 default=2,
303 )
303 )
304 coreconfigitem('paths', 'default',
304 coreconfigitem('paths', 'default',
305 default=None,
305 default=None,
306 )
306 )
307 coreconfigitem('paths', 'default-push',
307 coreconfigitem('paths', 'default-push',
308 default=None,
308 default=None,
309 )
309 )
310 coreconfigitem('phases', 'checksubrepos',
310 coreconfigitem('phases', 'checksubrepos',
311 default='follow',
311 default='follow',
312 )
312 )
313 coreconfigitem('phases', 'publish',
313 coreconfigitem('phases', 'publish',
314 default=True,
314 default=True,
315 )
315 )
316 coreconfigitem('profiling', 'enabled',
316 coreconfigitem('profiling', 'enabled',
317 default=False,
317 default=False,
318 )
318 )
319 coreconfigitem('profiling', 'format',
319 coreconfigitem('profiling', 'format',
320 default='text',
320 default='text',
321 )
321 )
322 coreconfigitem('profiling', 'freq',
322 coreconfigitem('profiling', 'freq',
323 default=1000,
323 default=1000,
324 )
324 )
325 coreconfigitem('profiling', 'limit',
325 coreconfigitem('profiling', 'limit',
326 default=30,
326 default=30,
327 )
327 )
328 coreconfigitem('profiling', 'nested',
328 coreconfigitem('profiling', 'nested',
329 default=0,
329 default=0,
330 )
330 )
331 coreconfigitem('profiling', 'sort',
331 coreconfigitem('profiling', 'sort',
332 default='inlinetime',
332 default='inlinetime',
333 )
333 )
334 coreconfigitem('profiling', 'statformat',
334 coreconfigitem('profiling', 'statformat',
335 default='hotpath',
335 default='hotpath',
336 )
336 )
337 coreconfigitem('progress', 'assume-tty',
337 coreconfigitem('progress', 'assume-tty',
338 default=False,
338 default=False,
339 )
339 )
340 coreconfigitem('progress', 'changedelay',
340 coreconfigitem('progress', 'changedelay',
341 default=1,
341 default=1,
342 )
342 )
343 coreconfigitem('progress', 'clear-complete',
343 coreconfigitem('progress', 'clear-complete',
344 default=True,
344 default=True,
345 )
345 )
346 coreconfigitem('progress', 'debug',
346 coreconfigitem('progress', 'debug',
347 default=False,
347 default=False,
348 )
348 )
349 coreconfigitem('progress', 'delay',
349 coreconfigitem('progress', 'delay',
350 default=3,
350 default=3,
351 )
351 )
352 coreconfigitem('progress', 'disable',
352 coreconfigitem('progress', 'disable',
353 default=False,
353 default=False,
354 )
354 )
355 coreconfigitem('progress', 'estimate',
355 coreconfigitem('progress', 'estimate',
356 default=2,
356 default=2,
357 )
357 )
358 coreconfigitem('progress', 'refresh',
358 coreconfigitem('progress', 'refresh',
359 default=0.1,
359 default=0.1,
360 )
360 )
361 coreconfigitem('progress', 'width',
361 coreconfigitem('progress', 'width',
362 default=dynamicdefault,
362 default=dynamicdefault,
363 )
363 )
364 coreconfigitem('push', 'pushvars.server',
365 default=False,
366 )
364 coreconfigitem('server', 'bundle1',
367 coreconfigitem('server', 'bundle1',
365 default=True,
368 default=True,
366 )
369 )
367 coreconfigitem('server', 'bundle1gd',
370 coreconfigitem('server', 'bundle1gd',
368 default=None,
371 default=None,
369 )
372 )
370 coreconfigitem('server', 'compressionengines',
373 coreconfigitem('server', 'compressionengines',
371 default=list,
374 default=list,
372 )
375 )
373 coreconfigitem('server', 'concurrent-push-mode',
376 coreconfigitem('server', 'concurrent-push-mode',
374 default='strict',
377 default='strict',
375 )
378 )
376 coreconfigitem('server', 'disablefullbundle',
379 coreconfigitem('server', 'disablefullbundle',
377 default=False,
380 default=False,
378 )
381 )
379 coreconfigitem('server', 'maxhttpheaderlen',
382 coreconfigitem('server', 'maxhttpheaderlen',
380 default=1024,
383 default=1024,
381 )
384 )
382 coreconfigitem('server', 'preferuncompressed',
385 coreconfigitem('server', 'preferuncompressed',
383 default=False,
386 default=False,
384 )
387 )
385 coreconfigitem('server', 'uncompressed',
388 coreconfigitem('server', 'uncompressed',
386 default=True,
389 default=True,
387 )
390 )
388 coreconfigitem('server', 'uncompressedallowsecret',
391 coreconfigitem('server', 'uncompressedallowsecret',
389 default=False,
392 default=False,
390 )
393 )
391 coreconfigitem('server', 'validate',
394 coreconfigitem('server', 'validate',
392 default=False,
395 default=False,
393 )
396 )
394 coreconfigitem('server', 'zliblevel',
397 coreconfigitem('server', 'zliblevel',
395 default=-1,
398 default=-1,
396 )
399 )
397 coreconfigitem('smtp', 'host',
400 coreconfigitem('smtp', 'host',
398 default=None,
401 default=None,
399 )
402 )
400 coreconfigitem('smtp', 'local_hostname',
403 coreconfigitem('smtp', 'local_hostname',
401 default=None,
404 default=None,
402 )
405 )
403 coreconfigitem('smtp', 'password',
406 coreconfigitem('smtp', 'password',
404 default=None,
407 default=None,
405 )
408 )
406 coreconfigitem('smtp', 'tls',
409 coreconfigitem('smtp', 'tls',
407 default='none',
410 default='none',
408 )
411 )
409 coreconfigitem('smtp', 'username',
412 coreconfigitem('smtp', 'username',
410 default=None,
413 default=None,
411 )
414 )
412 coreconfigitem('sparse', 'missingwarning',
415 coreconfigitem('sparse', 'missingwarning',
413 default=True,
416 default=True,
414 )
417 )
415 coreconfigitem('trusted', 'groups',
418 coreconfigitem('trusted', 'groups',
416 default=list,
419 default=list,
417 )
420 )
418 coreconfigitem('trusted', 'users',
421 coreconfigitem('trusted', 'users',
419 default=list,
422 default=list,
420 )
423 )
421 coreconfigitem('ui', '_usedassubrepo',
424 coreconfigitem('ui', '_usedassubrepo',
422 default=False,
425 default=False,
423 )
426 )
424 coreconfigitem('ui', 'allowemptycommit',
427 coreconfigitem('ui', 'allowemptycommit',
425 default=False,
428 default=False,
426 )
429 )
427 coreconfigitem('ui', 'archivemeta',
430 coreconfigitem('ui', 'archivemeta',
428 default=True,
431 default=True,
429 )
432 )
430 coreconfigitem('ui', 'askusername',
433 coreconfigitem('ui', 'askusername',
431 default=False,
434 default=False,
432 )
435 )
433 coreconfigitem('ui', 'clonebundlefallback',
436 coreconfigitem('ui', 'clonebundlefallback',
434 default=False,
437 default=False,
435 )
438 )
436 coreconfigitem('ui', 'clonebundleprefers',
439 coreconfigitem('ui', 'clonebundleprefers',
437 default=list,
440 default=list,
438 )
441 )
439 coreconfigitem('ui', 'clonebundles',
442 coreconfigitem('ui', 'clonebundles',
440 default=True,
443 default=True,
441 )
444 )
442 coreconfigitem('ui', 'color',
445 coreconfigitem('ui', 'color',
443 default='auto',
446 default='auto',
444 )
447 )
445 coreconfigitem('ui', 'commitsubrepos',
448 coreconfigitem('ui', 'commitsubrepos',
446 default=False,
449 default=False,
447 )
450 )
448 coreconfigitem('ui', 'debug',
451 coreconfigitem('ui', 'debug',
449 default=False,
452 default=False,
450 )
453 )
451 coreconfigitem('ui', 'debugger',
454 coreconfigitem('ui', 'debugger',
452 default=None,
455 default=None,
453 )
456 )
454 coreconfigitem('ui', 'fallbackencoding',
457 coreconfigitem('ui', 'fallbackencoding',
455 default=None,
458 default=None,
456 )
459 )
457 coreconfigitem('ui', 'forcecwd',
460 coreconfigitem('ui', 'forcecwd',
458 default=None,
461 default=None,
459 )
462 )
460 coreconfigitem('ui', 'forcemerge',
463 coreconfigitem('ui', 'forcemerge',
461 default=None,
464 default=None,
462 )
465 )
463 coreconfigitem('ui', 'formatdebug',
466 coreconfigitem('ui', 'formatdebug',
464 default=False,
467 default=False,
465 )
468 )
466 coreconfigitem('ui', 'formatjson',
469 coreconfigitem('ui', 'formatjson',
467 default=False,
470 default=False,
468 )
471 )
469 coreconfigitem('ui', 'formatted',
472 coreconfigitem('ui', 'formatted',
470 default=None,
473 default=None,
471 )
474 )
472 coreconfigitem('ui', 'graphnodetemplate',
475 coreconfigitem('ui', 'graphnodetemplate',
473 default=None,
476 default=None,
474 )
477 )
475 coreconfigitem('ui', 'http2debuglevel',
478 coreconfigitem('ui', 'http2debuglevel',
476 default=None,
479 default=None,
477 )
480 )
478 coreconfigitem('ui', 'interactive',
481 coreconfigitem('ui', 'interactive',
479 default=None,
482 default=None,
480 )
483 )
481 coreconfigitem('ui', 'interface',
484 coreconfigitem('ui', 'interface',
482 default=None,
485 default=None,
483 )
486 )
484 coreconfigitem('ui', 'logblockedtimes',
487 coreconfigitem('ui', 'logblockedtimes',
485 default=False,
488 default=False,
486 )
489 )
487 coreconfigitem('ui', 'logtemplate',
490 coreconfigitem('ui', 'logtemplate',
488 default=None,
491 default=None,
489 )
492 )
490 coreconfigitem('ui', 'merge',
493 coreconfigitem('ui', 'merge',
491 default=None,
494 default=None,
492 )
495 )
493 coreconfigitem('ui', 'mergemarkers',
496 coreconfigitem('ui', 'mergemarkers',
494 default='basic',
497 default='basic',
495 )
498 )
496 coreconfigitem('ui', 'mergemarkertemplate',
499 coreconfigitem('ui', 'mergemarkertemplate',
497 default=('{node|short} '
500 default=('{node|short} '
498 '{ifeq(tags, "tip", "", '
501 '{ifeq(tags, "tip", "", '
499 'ifeq(tags, "", "", "{tags} "))}'
502 'ifeq(tags, "", "", "{tags} "))}'
500 '{if(bookmarks, "{bookmarks} ")}'
503 '{if(bookmarks, "{bookmarks} ")}'
501 '{ifeq(branch, "default", "", "{branch} ")}'
504 '{ifeq(branch, "default", "", "{branch} ")}'
502 '- {author|user}: {desc|firstline}')
505 '- {author|user}: {desc|firstline}')
503 )
506 )
504 coreconfigitem('ui', 'nontty',
507 coreconfigitem('ui', 'nontty',
505 default=False,
508 default=False,
506 )
509 )
507 coreconfigitem('ui', 'origbackuppath',
510 coreconfigitem('ui', 'origbackuppath',
508 default=None,
511 default=None,
509 )
512 )
510 coreconfigitem('ui', 'paginate',
513 coreconfigitem('ui', 'paginate',
511 default=True,
514 default=True,
512 )
515 )
513 coreconfigitem('ui', 'patch',
516 coreconfigitem('ui', 'patch',
514 default=None,
517 default=None,
515 )
518 )
516 coreconfigitem('ui', 'portablefilenames',
519 coreconfigitem('ui', 'portablefilenames',
517 default='warn',
520 default='warn',
518 )
521 )
519 coreconfigitem('ui', 'promptecho',
522 coreconfigitem('ui', 'promptecho',
520 default=False,
523 default=False,
521 )
524 )
522 coreconfigitem('ui', 'quiet',
525 coreconfigitem('ui', 'quiet',
523 default=False,
526 default=False,
524 )
527 )
525 coreconfigitem('ui', 'quietbookmarkmove',
528 coreconfigitem('ui', 'quietbookmarkmove',
526 default=False,
529 default=False,
527 )
530 )
528 coreconfigitem('ui', 'remotecmd',
531 coreconfigitem('ui', 'remotecmd',
529 default='hg',
532 default='hg',
530 )
533 )
531 coreconfigitem('ui', 'report_untrusted',
534 coreconfigitem('ui', 'report_untrusted',
532 default=True,
535 default=True,
533 )
536 )
534 coreconfigitem('ui', 'rollback',
537 coreconfigitem('ui', 'rollback',
535 default=True,
538 default=True,
536 )
539 )
537 coreconfigitem('ui', 'slash',
540 coreconfigitem('ui', 'slash',
538 default=False,
541 default=False,
539 )
542 )
540 coreconfigitem('ui', 'ssh',
543 coreconfigitem('ui', 'ssh',
541 default='ssh',
544 default='ssh',
542 )
545 )
543 coreconfigitem('ui', 'statuscopies',
546 coreconfigitem('ui', 'statuscopies',
544 default=False,
547 default=False,
545 )
548 )
546 coreconfigitem('ui', 'strict',
549 coreconfigitem('ui', 'strict',
547 default=False,
550 default=False,
548 )
551 )
549 coreconfigitem('ui', 'style',
552 coreconfigitem('ui', 'style',
550 default='',
553 default='',
551 )
554 )
552 coreconfigitem('ui', 'supportcontact',
555 coreconfigitem('ui', 'supportcontact',
553 default=None,
556 default=None,
554 )
557 )
555 coreconfigitem('ui', 'textwidth',
558 coreconfigitem('ui', 'textwidth',
556 default=78,
559 default=78,
557 )
560 )
558 coreconfigitem('ui', 'timeout',
561 coreconfigitem('ui', 'timeout',
559 default='600',
562 default='600',
560 )
563 )
561 coreconfigitem('ui', 'traceback',
564 coreconfigitem('ui', 'traceback',
562 default=False,
565 default=False,
563 )
566 )
564 coreconfigitem('ui', 'tweakdefaults',
567 coreconfigitem('ui', 'tweakdefaults',
565 default=False,
568 default=False,
566 )
569 )
567 coreconfigitem('ui', 'usehttp2',
570 coreconfigitem('ui', 'usehttp2',
568 default=False,
571 default=False,
569 )
572 )
570 coreconfigitem('ui', 'username',
573 coreconfigitem('ui', 'username',
571 alias=[('ui', 'user')]
574 alias=[('ui', 'user')]
572 )
575 )
573 coreconfigitem('ui', 'verbose',
576 coreconfigitem('ui', 'verbose',
574 default=False,
577 default=False,
575 )
578 )
576 coreconfigitem('verify', 'skipflags',
579 coreconfigitem('verify', 'skipflags',
577 default=None,
580 default=None,
578 )
581 )
579 coreconfigitem('worker', 'backgroundclose',
582 coreconfigitem('worker', 'backgroundclose',
580 default=dynamicdefault,
583 default=dynamicdefault,
581 )
584 )
582 # Windows defaults to a limit of 512 open files. A buffer of 128
585 # Windows defaults to a limit of 512 open files. A buffer of 128
583 # should give us enough headway.
586 # should give us enough headway.
584 coreconfigitem('worker', 'backgroundclosemaxqueue',
587 coreconfigitem('worker', 'backgroundclosemaxqueue',
585 default=384,
588 default=384,
586 )
589 )
587 coreconfigitem('worker', 'backgroundcloseminfilecount',
590 coreconfigitem('worker', 'backgroundcloseminfilecount',
588 default=2048,
591 default=2048,
589 )
592 )
590 coreconfigitem('worker', 'backgroundclosethreadcount',
593 coreconfigitem('worker', 'backgroundclosethreadcount',
591 default=4,
594 default=4,
592 )
595 )
593 coreconfigitem('worker', 'numcpus',
596 coreconfigitem('worker', 'numcpus',
594 default=None,
597 default=None,
595 )
598 )
General Comments 0
You need to be logged in to leave comments. Login now