| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917 |
- # pack.py -- For dealing with packed git objects.
- # Copyright (C) 2007 James Westby <jw+debian@jameswestby.net>
- # Copyright (C) 2008-2013 Jelmer Vernooij <jelmer@jelmer.uk>
- #
- # SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
- # Dulwich is dual-licensed under the Apache License, Version 2.0 and the GNU
- # General Public License as published by the Free Software Foundation; version 2.0
- # or (at your option) any later version. You can redistribute it and/or
- # modify it under the terms of either of these two licenses.
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- #
- # You should have received a copy of the licenses; if not, see
- # <http://www.gnu.org/licenses/> for a copy of the GNU General Public License
- # and <http://www.apache.org/licenses/LICENSE-2.0> for a copy of the Apache
- # License, Version 2.0.
- #
- """Classes for dealing with packed git objects.
- A pack is a compact representation of a bunch of objects, stored
- using deltas where possible.
- They have two parts, the pack file, which stores the data, and an index
- that tells you where the data is.
- To find an object you look in all of the index files 'til you find a
- match for the object name. You then use the pointer got from this as
- a pointer in to the corresponding packfile.
- """
- import binascii
- from collections import defaultdict, deque
- from contextlib import suppress
- from io import BytesIO, UnsupportedOperation
- try:
- from cdifflib import CSequenceMatcher as SequenceMatcher
- except ModuleNotFoundError:
- from difflib import SequenceMatcher
- import os
- import struct
- import sys
- import warnings
- import zlib
- from collections.abc import Callable, Iterable, Iterator, Sequence, Set
- from hashlib import sha1
- from itertools import chain
- from os import SEEK_CUR, SEEK_END
- from struct import unpack_from
- from types import TracebackType
- from typing import (
- IO,
- TYPE_CHECKING,
- Any,
- BinaryIO,
- Generic,
- Protocol,
- TypeVar,
- )
- try:
- import mmap
- except ImportError:
- has_mmap = False
- else:
- has_mmap = True
- if TYPE_CHECKING:
- from _hashlib import HASH as HashObject
- from .bitmap import PackBitmap
- from .commit_graph import CommitGraph
- # For some reason the above try, except fails to set has_mmap = False for plan9
- if sys.platform == "Plan9":
- has_mmap = False
- from . import replace_me
- from .errors import ApplyDeltaError, ChecksumMismatch
- from .file import GitFile, _GitFile
- from .lru_cache import LRUSizeCache
- from .objects import ObjectID, ShaFile, hex_to_sha, object_header, sha_to_hex
- OFS_DELTA = 6
- REF_DELTA = 7
- DELTA_TYPES = (OFS_DELTA, REF_DELTA)
- DEFAULT_PACK_DELTA_WINDOW_SIZE = 10
- # Keep pack files under 16Mb in memory, otherwise write them out to disk
- PACK_SPOOL_FILE_MAX_SIZE = 16 * 1024 * 1024
- # Default pack index version to use when none is specified
- DEFAULT_PACK_INDEX_VERSION = 2
- OldUnpackedObject = tuple[bytes | int, list[bytes]] | list[bytes]
- ResolveExtRefFn = Callable[[bytes], tuple[int, OldUnpackedObject]]
- ProgressFn = Callable[[int, str], None]
- PackHint = tuple[int, bytes | None]
- class UnresolvedDeltas(Exception):
- """Delta objects could not be resolved."""
- def __init__(self, shas: list[bytes]) -> None:
- """Initialize UnresolvedDeltas exception.
- Args:
- shas: List of SHA hashes for unresolved delta objects
- """
- self.shas = shas
- class ObjectContainer(Protocol):
- """Protocol for objects that can contain git objects."""
- def add_object(self, obj: ShaFile) -> None:
- """Add a single object to this object store."""
- def add_objects(
- self,
- objects: Sequence[tuple[ShaFile, str | None]],
- progress: Callable[..., None] | None = None,
- ) -> "Pack" | None:
- """Add a set of objects to this object store.
- Args:
- objects: Iterable over a list of (object, path) tuples
- progress: Progress callback for object insertion
- Returns: Optional Pack object of the objects written.
- """
- def __contains__(self, sha1: bytes) -> bool:
- """Check if a hex sha is present."""
- def __getitem__(self, sha1: bytes) -> ShaFile:
- """Retrieve an object."""
- def get_commit_graph(self) -> "CommitGraph" | None:
- """Get the commit graph for this object store.
- Returns:
- CommitGraph object if available, None otherwise
- """
- return None
- class PackedObjectContainer(ObjectContainer):
- """Container for objects packed in a pack file."""
- def get_unpacked_object(
- self, sha1: bytes, *, include_comp: bool = False
- ) -> "UnpackedObject":
- """Get a raw unresolved object.
- Args:
- sha1: SHA-1 hash of the object
- include_comp: Whether to include compressed data
- Returns:
- UnpackedObject instance
- """
- raise NotImplementedError(self.get_unpacked_object)
- def iterobjects_subset(
- self, shas: Iterable[bytes], *, allow_missing: bool = False
- ) -> Iterator[ShaFile]:
- """Iterate over a subset of objects.
- Args:
- shas: Iterable of object SHAs to retrieve
- allow_missing: If True, skip missing objects
- Returns:
- Iterator of ShaFile objects
- """
- raise NotImplementedError(self.iterobjects_subset)
- def iter_unpacked_subset(
- self,
- shas: Iterable[bytes],
- *,
- include_comp: bool = False,
- allow_missing: bool = False,
- convert_ofs_delta: bool = True,
- ) -> Iterator["UnpackedObject"]:
- """Iterate over unpacked objects from a subset of SHAs.
- Args:
- shas: Set of object SHAs to retrieve
- include_comp: Include compressed data if True
- allow_missing: If True, skip missing objects
- convert_ofs_delta: If True, convert offset deltas to ref deltas
- Returns:
- Iterator of UnpackedObject instances
- """
- raise NotImplementedError(self.iter_unpacked_subset)
- class UnpackedObjectStream:
- """Abstract base class for a stream of unpacked objects."""
- def __iter__(self) -> Iterator["UnpackedObject"]:
- """Iterate over unpacked objects."""
- raise NotImplementedError(self.__iter__)
- def __len__(self) -> int:
- """Return the number of objects in the stream."""
- raise NotImplementedError(self.__len__)
- def take_msb_bytes(
- read: Callable[[int], bytes], crc32: int | None = None
- ) -> tuple[list[int], int | None]:
- """Read bytes marked with most significant bit.
- Args:
- read: Read function
- crc32: Optional CRC32 checksum to update
- Returns:
- Tuple of (list of bytes read, updated CRC32 or None)
- """
- ret: list[int] = []
- while len(ret) == 0 or ret[-1] & 0x80:
- b = read(1)
- if crc32 is not None:
- crc32 = binascii.crc32(b, crc32)
- ret.append(ord(b[:1]))
- return ret, crc32
- class PackFileDisappeared(Exception):
- """Raised when a pack file unexpectedly disappears."""
- def __init__(self, obj: object) -> None:
- """Initialize PackFileDisappeared exception.
- Args:
- obj: The object that triggered the exception
- """
- self.obj = obj
- class UnpackedObject:
- """Class encapsulating an object unpacked from a pack file.
- These objects should only be created from within unpack_object. Most
- members start out as empty and are filled in at various points by
- read_zlib_chunks, unpack_object, DeltaChainIterator, etc.
- End users of this object should take care that the function they're getting
- this object from is guaranteed to set the members they need.
- """
- __slots__ = [
- "_sha", # Cached binary SHA.
- "comp_chunks", # Compressed object chunks.
- "crc32", # CRC32.
- "decomp_chunks", # Decompressed object chunks.
- "decomp_len", # Decompressed length of this object.
- "delta_base", # Delta base offset or SHA.
- "obj_chunks", # Decompressed and delta-resolved chunks.
- "obj_type_num", # Type of this object.
- "offset", # Offset in its pack.
- "pack_type_num", # Type of this object in the pack (may be a delta).
- ]
- obj_type_num: int | None
- obj_chunks: list[bytes] | None
- delta_base: None | bytes | int
- decomp_chunks: list[bytes]
- comp_chunks: list[bytes] | None
- decomp_len: int | None
- crc32: int | None
- offset: int | None
- pack_type_num: int
- _sha: bytes | None
- # TODO(dborowitz): read_zlib_chunks and unpack_object could very well be
- # methods of this object.
- def __init__(
- self,
- pack_type_num: int,
- *,
- delta_base: None | bytes | int = None,
- decomp_len: int | None = None,
- crc32: int | None = None,
- sha: bytes | None = None,
- decomp_chunks: list[bytes] | None = None,
- offset: int | None = None,
- ) -> None:
- """Initialize an UnpackedObject.
- Args:
- pack_type_num: Type number of this object in the pack
- delta_base: Delta base (offset or SHA) if this is a delta object
- decomp_len: Decompressed length of this object
- crc32: CRC32 checksum
- sha: SHA-1 hash of the object
- decomp_chunks: Decompressed chunks
- offset: Offset in the pack file
- """
- self.offset = offset
- self._sha = sha
- self.pack_type_num = pack_type_num
- self.delta_base = delta_base
- self.comp_chunks = None
- self.decomp_chunks: list[bytes] = decomp_chunks or []
- if decomp_chunks is not None and decomp_len is None:
- self.decomp_len = sum(map(len, decomp_chunks))
- else:
- self.decomp_len = decomp_len
- self.crc32 = crc32
- if pack_type_num in DELTA_TYPES:
- self.obj_type_num = None
- self.obj_chunks = None
- else:
- self.obj_type_num = pack_type_num
- self.obj_chunks = self.decomp_chunks
- self.delta_base = delta_base
- def sha(self) -> bytes:
- """Return the binary SHA of this object."""
- if self._sha is None:
- assert self.obj_type_num is not None and self.obj_chunks is not None
- self._sha = obj_sha(self.obj_type_num, self.obj_chunks)
- return self._sha
- def sha_file(self) -> ShaFile:
- """Return a ShaFile from this object."""
- assert self.obj_type_num is not None and self.obj_chunks is not None
- return ShaFile.from_raw_chunks(self.obj_type_num, self.obj_chunks)
- # Only provided for backwards compatibility with code that expects either
- # chunks or a delta tuple.
- def _obj(self) -> OldUnpackedObject:
- """Return the decompressed chunks, or (delta base, delta chunks)."""
- if self.pack_type_num in DELTA_TYPES:
- assert isinstance(self.delta_base, (bytes, int))
- return (self.delta_base, self.decomp_chunks)
- else:
- return self.decomp_chunks
- def __eq__(self, other: object) -> bool:
- """Check equality with another UnpackedObject."""
- if not isinstance(other, UnpackedObject):
- return False
- for slot in self.__slots__:
- if getattr(self, slot) != getattr(other, slot):
- return False
- return True
- def __ne__(self, other: object) -> bool:
- """Check inequality with another UnpackedObject."""
- return not (self == other)
- def __repr__(self) -> str:
- """Return string representation of this UnpackedObject."""
- data = [f"{s}={getattr(self, s)!r}" for s in self.__slots__]
- return "{}({})".format(self.__class__.__name__, ", ".join(data))
- _ZLIB_BUFSIZE = 65536 # 64KB buffer for better I/O performance
- def read_zlib_chunks(
- read_some: Callable[[int], bytes],
- unpacked: UnpackedObject,
- include_comp: bool = False,
- buffer_size: int = _ZLIB_BUFSIZE,
- ) -> bytes:
- """Read zlib data from a buffer.
- This function requires that the buffer have additional data following the
- compressed data, which is guaranteed to be the case for git pack files.
- Args:
- read_some: Read function that returns at least one byte, but may
- return less than the requested size.
- unpacked: An UnpackedObject to write result data to. If its crc32
- attr is not None, the CRC32 of the compressed bytes will be computed
- using this starting CRC32.
- After this function, will have the following attrs set:
- * comp_chunks (if include_comp is True)
- * decomp_chunks
- * decomp_len
- * crc32
- include_comp: If True, include compressed data in the result.
- buffer_size: Size of the read buffer.
- Returns: Leftover unused data from the decompression.
- Raises:
- zlib.error: if a decompression error occurred.
- """
- if unpacked.decomp_len is None or unpacked.decomp_len <= -1:
- raise ValueError("non-negative zlib data stream size expected")
- decomp_obj = zlib.decompressobj()
- comp_chunks = []
- decomp_chunks = unpacked.decomp_chunks
- decomp_len = 0
- crc32 = unpacked.crc32
- while True:
- add = read_some(buffer_size)
- if not add:
- raise zlib.error("EOF before end of zlib stream")
- comp_chunks.append(add)
- decomp = decomp_obj.decompress(add)
- decomp_len += len(decomp)
- decomp_chunks.append(decomp)
- unused = decomp_obj.unused_data
- if unused:
- left = len(unused)
- if crc32 is not None:
- crc32 = binascii.crc32(add[:-left], crc32)
- if include_comp:
- comp_chunks[-1] = add[:-left]
- break
- elif crc32 is not None:
- crc32 = binascii.crc32(add, crc32)
- if crc32 is not None:
- crc32 &= 0xFFFFFFFF
- if decomp_len != unpacked.decomp_len:
- raise zlib.error("decompressed data does not match expected size")
- unpacked.crc32 = crc32
- if include_comp:
- unpacked.comp_chunks = comp_chunks
- return unused
- def iter_sha1(iter: Iterable[bytes]) -> bytes:
- """Return the hexdigest of the SHA1 over a set of names.
- Args:
- iter: Iterator over string objects
- Returns: 40-byte hex sha1 digest
- """
- sha = sha1()
- for name in iter:
- sha.update(name)
- return sha.hexdigest().encode("ascii")
- def load_pack_index(path: str | os.PathLike[str]) -> "PackIndex":
- """Load an index file by path.
- Args:
- path: Path to the index file
- Returns: A PackIndex loaded from the given path
- """
- with GitFile(path, "rb") as f:
- return load_pack_index_file(path, f)
- def _load_file_contents(
- f: IO[bytes] | _GitFile, size: int | None = None
- ) -> tuple[bytes | Any, int]:
- """Load contents from a file, preferring mmap when possible.
- Args:
- f: File-like object to load
- size: Expected size, or None to determine from file
- Returns: Tuple of (contents, size)
- """
- try:
- fd = f.fileno()
- except (UnsupportedOperation, AttributeError):
- fd = None
- # Attempt to use mmap if possible
- if fd is not None:
- if size is None:
- size = os.fstat(fd).st_size
- if has_mmap:
- try:
- contents = mmap.mmap(fd, size, access=mmap.ACCESS_READ)
- except (OSError, ValueError):
- # Can't mmap - perhaps a socket or invalid file descriptor
- pass
- else:
- return contents, size
- contents_bytes = f.read()
- size = len(contents_bytes)
- return contents_bytes, size
- def load_pack_index_file(
- path: str | os.PathLike[str], f: IO[bytes] | _GitFile
- ) -> "PackIndex":
- """Load an index file from a file-like object.
- Args:
- path: Path for the index file
- f: File-like object
- Returns: A PackIndex loaded from the given file
- """
- contents, size = _load_file_contents(f)
- if contents[:4] == b"\377tOc":
- version = struct.unpack(b">L", contents[4:8])[0]
- if version == 2:
- return PackIndex2(path, file=f, contents=contents, size=size)
- elif version == 3:
- return PackIndex3(path, file=f, contents=contents, size=size)
- else:
- raise KeyError(f"Unknown pack index format {version}")
- else:
- return PackIndex1(path, file=f, contents=contents, size=size)
- def bisect_find_sha(
- start: int, end: int, sha: bytes, unpack_name: Callable[[int], bytes]
- ) -> int | None:
- """Find a SHA in a data blob with sorted SHAs.
- Args:
- start: Start index of range to search
- end: End index of range to search
- sha: Sha to find
- unpack_name: Callback to retrieve SHA by index
- Returns: Index of the SHA, or None if it wasn't found
- """
- assert start <= end
- while start <= end:
- i = (start + end) // 2
- file_sha = unpack_name(i)
- if file_sha < sha:
- start = i + 1
- elif file_sha > sha:
- end = i - 1
- else:
- return i
- return None
- PackIndexEntry = tuple[bytes, int, int | None]
- class PackIndex:
- """An index in to a packfile.
- Given a sha id of an object a pack index can tell you the location in the
- packfile of that object if it has it.
- """
- # Default to SHA-1 for backward compatibility
- hash_algorithm = 1
- hash_size = 20
- def __eq__(self, other: object) -> bool:
- """Check equality with another PackIndex."""
- if not isinstance(other, PackIndex):
- return False
- for (name1, _, _), (name2, _, _) in zip(
- self.iterentries(), other.iterentries()
- ):
- if name1 != name2:
- return False
- return True
- def __ne__(self, other: object) -> bool:
- """Check if this pack index is not equal to another."""
- return not self.__eq__(other)
- def __len__(self) -> int:
- """Return the number of entries in this pack index."""
- raise NotImplementedError(self.__len__)
- def __iter__(self) -> Iterator[bytes]:
- """Iterate over the SHAs in this pack."""
- return map(sha_to_hex, self._itersha())
- def iterentries(self) -> Iterator[PackIndexEntry]:
- """Iterate over the entries in this pack index.
- Returns: iterator over tuples with object name, offset in packfile and
- crc32 checksum.
- """
- raise NotImplementedError(self.iterentries)
- def get_pack_checksum(self) -> bytes | None:
- """Return the SHA1 checksum stored for the corresponding packfile.
- Returns: 20-byte binary digest, or None if not available
- """
- raise NotImplementedError(self.get_pack_checksum)
- @replace_me(since="0.21.0", remove_in="0.23.0")
- def object_index(self, sha: bytes) -> int:
- """Return the index for the given SHA.
- Args:
- sha: SHA-1 hash
- Returns:
- Index position
- """
- return self.object_offset(sha)
- def object_offset(self, sha: bytes) -> int:
- """Return the offset in to the corresponding packfile for the object.
- Given the name of an object it will return the offset that object
- lives at within the corresponding pack file. If the pack file doesn't
- have the object then None will be returned.
- """
- raise NotImplementedError(self.object_offset)
- def object_sha1(self, index: int) -> bytes:
- """Return the SHA1 corresponding to the index in the pack file."""
- for name, offset, _crc32 in self.iterentries():
- if offset == index:
- return name
- else:
- raise KeyError(index)
- def _object_offset(self, sha: bytes) -> int:
- """See object_offset.
- Args:
- sha: A *binary* SHA string. (20 characters long)_
- """
- raise NotImplementedError(self._object_offset)
- def objects_sha1(self) -> bytes:
- """Return the hex SHA1 over all the shas of all objects in this pack.
- Note: This is used for the filename of the pack.
- """
- return iter_sha1(self._itersha())
- def _itersha(self) -> Iterator[bytes]:
- """Yield all the SHA1's of the objects in the index, sorted."""
- raise NotImplementedError(self._itersha)
- def iter_prefix(self, prefix: bytes) -> Iterator[bytes]:
- """Iterate over all SHA1s with the given prefix.
- Args:
- prefix: Binary prefix to match
- Returns: Iterator of matching SHA1s
- """
- # Default implementation for PackIndex classes that don't override
- for sha, _, _ in self.iterentries():
- if sha.startswith(prefix):
- yield sha
- def close(self) -> None:
- """Close any open files."""
- def check(self) -> None:
- """Check the consistency of this pack index."""
- class MemoryPackIndex(PackIndex):
- """Pack index that is stored entirely in memory."""
- def __init__(
- self,
- entries: list[tuple[bytes, int, int | None]],
- pack_checksum: bytes | None = None,
- ) -> None:
- """Create a new MemoryPackIndex.
- Args:
- entries: Sequence of name, idx, crc32 (sorted)
- pack_checksum: Optional pack checksum
- """
- self._by_sha = {}
- self._by_offset = {}
- for name, offset, _crc32 in entries:
- self._by_sha[name] = offset
- self._by_offset[offset] = name
- self._entries = entries
- self._pack_checksum = pack_checksum
- def get_pack_checksum(self) -> bytes | None:
- """Return the SHA checksum stored for the corresponding packfile."""
- return self._pack_checksum
- def __len__(self) -> int:
- """Return the number of entries in this pack index."""
- return len(self._entries)
- def object_offset(self, sha: bytes) -> int:
- """Return the offset for the given SHA.
- Args:
- sha: SHA to look up (binary or hex)
- Returns: Offset in the pack file
- """
- if len(sha) == 40:
- sha = hex_to_sha(sha)
- return self._by_sha[sha]
- def object_sha1(self, offset: int) -> bytes:
- """Return the SHA1 for the object at the given offset."""
- return self._by_offset[offset]
- def _itersha(self) -> Iterator[bytes]:
- """Iterate over all SHA1s in the index."""
- return iter(self._by_sha)
- def iterentries(self) -> Iterator[PackIndexEntry]:
- """Iterate over all index entries."""
- return iter(self._entries)
- @classmethod
- def for_pack(cls, pack_data: "PackData") -> "MemoryPackIndex":
- """Create a MemoryPackIndex from a PackData object."""
- return MemoryPackIndex(
- list(pack_data.sorted_entries()), pack_data.get_stored_checksum()
- )
- @classmethod
- def clone(cls, other_index: "PackIndex") -> "MemoryPackIndex":
- """Create a copy of another PackIndex in memory."""
- return cls(list(other_index.iterentries()), other_index.get_pack_checksum())
- class FilePackIndex(PackIndex):
- """Pack index that is based on a file.
- To do the loop it opens the file, and indexes first 256 4 byte groups
- with the first byte of the sha id. The value in the four byte group indexed
- is the end of the group that shares the same starting byte. Subtract one
- from the starting byte and index again to find the start of the group.
- The values are sorted by sha id within the group, so do the math to find
- the start and end offset and then bisect in to find if the value is
- present.
- """
- _fan_out_table: list[int]
- _file: IO[bytes] | _GitFile
- def __init__(
- self,
- filename: str | os.PathLike[str],
- file: IO[bytes] | _GitFile | None = None,
- contents: bytes | "mmap.mmap" | None = None,
- size: int | None = None,
- ) -> None:
- """Create a pack index object.
- Provide it with the name of the index file to consider, and it will map
- it whenever required.
- """
- self._filename = filename
- # Take the size now, so it can be checked each time we map the file to
- # ensure that it hasn't changed.
- if file is None:
- self._file = GitFile(filename, "rb")
- else:
- self._file = file
- if contents is None:
- self._contents, self._size = _load_file_contents(self._file, size)
- else:
- self._contents = contents
- self._size = size if size is not None else len(contents)
- @property
- def path(self) -> str:
- """Return the path to this index file."""
- return os.fspath(self._filename)
- def __eq__(self, other: object) -> bool:
- """Check equality with another FilePackIndex."""
- # Quick optimization:
- if (
- isinstance(other, FilePackIndex)
- and self._fan_out_table != other._fan_out_table
- ):
- return False
- return super().__eq__(other)
- def close(self) -> None:
- """Close the underlying file and any mmap."""
- self._file.close()
- close_fn = getattr(self._contents, "close", None)
- if close_fn is not None:
- close_fn()
- def __len__(self) -> int:
- """Return the number of entries in this pack index."""
- return self._fan_out_table[-1]
- def _unpack_entry(self, i: int) -> PackIndexEntry:
- """Unpack the i-th entry in the index file.
- Returns: Tuple with object name (SHA), offset in pack file and CRC32
- checksum (if known).
- """
- raise NotImplementedError(self._unpack_entry)
- def _unpack_name(self, i: int) -> bytes:
- """Unpack the i-th name from the index file."""
- raise NotImplementedError(self._unpack_name)
- def _unpack_offset(self, i: int) -> int:
- """Unpack the i-th object offset from the index file."""
- raise NotImplementedError(self._unpack_offset)
- def _unpack_crc32_checksum(self, i: int) -> int | None:
- """Unpack the crc32 checksum for the ith object from the index file."""
- raise NotImplementedError(self._unpack_crc32_checksum)
- def _itersha(self) -> Iterator[bytes]:
- """Iterate over all SHA1s in the index."""
- for i in range(len(self)):
- yield self._unpack_name(i)
- def iterentries(self) -> Iterator[PackIndexEntry]:
- """Iterate over the entries in this pack index.
- Returns: iterator over tuples with object name, offset in packfile and
- crc32 checksum.
- """
- for i in range(len(self)):
- yield self._unpack_entry(i)
- def _read_fan_out_table(self, start_offset: int) -> list[int]:
- """Read the fan-out table from the index.
- The fan-out table contains 256 entries mapping first byte values
- to the number of objects with SHA1s less than or equal to that byte.
- Args:
- start_offset: Offset in the file where the fan-out table starts
- Returns: List of 256 integers
- """
- ret = []
- for i in range(0x100):
- fanout_entry = self._contents[
- start_offset + i * 4 : start_offset + (i + 1) * 4
- ]
- ret.append(struct.unpack(">L", fanout_entry)[0])
- return ret
- def check(self) -> None:
- """Check that the stored checksum matches the actual checksum."""
- actual = self.calculate_checksum()
- stored = self.get_stored_checksum()
- if actual != stored:
- raise ChecksumMismatch(stored, actual)
- def calculate_checksum(self) -> bytes:
- """Calculate the SHA1 checksum over this pack index.
- Returns: This is a 20-byte binary digest
- """
- return sha1(self._contents[:-20]).digest()
- def get_pack_checksum(self) -> bytes:
- """Return the SHA1 checksum stored for the corresponding packfile.
- Returns: 20-byte binary digest
- """
- return bytes(self._contents[-40:-20])
- def get_stored_checksum(self) -> bytes:
- """Return the SHA1 checksum stored for this index.
- Returns: 20-byte binary digest
- """
- return bytes(self._contents[-20:])
- def object_offset(self, sha: bytes) -> int:
- """Return the offset in to the corresponding packfile for the object.
- Given the name of an object it will return the offset that object
- lives at within the corresponding pack file. If the pack file doesn't
- have the object then None will be returned.
- """
- if len(sha) == 40:
- sha = hex_to_sha(sha)
- try:
- return self._object_offset(sha)
- except ValueError as exc:
- closed = getattr(self._contents, "closed", None)
- if closed in (None, True):
- raise PackFileDisappeared(self) from exc
- raise
- def _object_offset(self, sha: bytes) -> int:
- """See object_offset.
- Args:
- sha: A *binary* SHA string. (20 characters long)_
- """
- assert len(sha) == 20
- idx = ord(sha[:1])
- if idx == 0:
- start = 0
- else:
- start = self._fan_out_table[idx - 1]
- end = self._fan_out_table[idx]
- i = bisect_find_sha(start, end, sha, self._unpack_name)
- if i is None:
- raise KeyError(sha)
- return self._unpack_offset(i)
- def iter_prefix(self, prefix: bytes) -> Iterator[bytes]:
- """Iterate over all SHA1s with the given prefix."""
- start = ord(prefix[:1])
- if start == 0:
- start = 0
- else:
- start = self._fan_out_table[start - 1]
- end = ord(prefix[:1]) + 1
- if end == 0x100:
- end = len(self)
- else:
- end = self._fan_out_table[end]
- assert start <= end
- started = False
- for i in range(start, end):
- name: bytes = self._unpack_name(i)
- if name.startswith(prefix):
- yield name
- started = True
- elif started:
- break
- class PackIndex1(FilePackIndex):
- """Version 1 Pack Index file."""
- def __init__(
- self,
- filename: str | os.PathLike[str],
- file: IO[bytes] | _GitFile | None = None,
- contents: bytes | None = None,
- size: int | None = None,
- ) -> None:
- """Initialize a version 1 pack index.
- Args:
- filename: Path to the index file
- file: Optional file object
- contents: Optional mmap'd contents
- size: Optional size of the index
- """
- super().__init__(filename, file, contents, size)
- self.version = 1
- self._fan_out_table = self._read_fan_out_table(0)
- def _unpack_entry(self, i: int) -> tuple[bytes, int, None]:
- (offset, name) = unpack_from(">L20s", self._contents, (0x100 * 4) + (i * 24))
- return (name, offset, None)
- def _unpack_name(self, i: int) -> bytes:
- offset = (0x100 * 4) + (i * 24) + 4
- return self._contents[offset : offset + 20]
- def _unpack_offset(self, i: int) -> int:
- offset = (0x100 * 4) + (i * 24)
- result = unpack_from(">L", self._contents, offset)[0]
- assert isinstance(result, int)
- return result
- def _unpack_crc32_checksum(self, i: int) -> None:
- # Not stored in v1 index files
- return None
- class PackIndex2(FilePackIndex):
- """Version 2 Pack Index file."""
- def __init__(
- self,
- filename: str | os.PathLike[str],
- file: IO[bytes] | _GitFile | None = None,
- contents: bytes | None = None,
- size: int | None = None,
- ) -> None:
- """Initialize a version 2 pack index.
- Args:
- filename: Path to the index file
- file: Optional file object
- contents: Optional mmap'd contents
- size: Optional size of the index
- """
- super().__init__(filename, file, contents, size)
- if self._contents[:4] != b"\377tOc":
- raise AssertionError("Not a v2 pack index file")
- (self.version,) = unpack_from(b">L", self._contents, 4)
- if self.version != 2:
- raise AssertionError(f"Version was {self.version}")
- self._fan_out_table = self._read_fan_out_table(8)
- self._name_table_offset = 8 + 0x100 * 4
- self._crc32_table_offset = self._name_table_offset + 20 * len(self)
- self._pack_offset_table_offset = self._crc32_table_offset + 4 * len(self)
- self._pack_offset_largetable_offset = self._pack_offset_table_offset + 4 * len(
- self
- )
- def _unpack_entry(self, i: int) -> tuple[bytes, int, int]:
- return (
- self._unpack_name(i),
- self._unpack_offset(i),
- self._unpack_crc32_checksum(i),
- )
- def _unpack_name(self, i: int) -> bytes:
- offset = self._name_table_offset + i * 20
- return self._contents[offset : offset + 20]
- def _unpack_offset(self, i: int) -> int:
- offset_pos = self._pack_offset_table_offset + i * 4
- offset = unpack_from(">L", self._contents, offset_pos)[0]
- assert isinstance(offset, int)
- if offset & (2**31):
- large_offset_pos = (
- self._pack_offset_largetable_offset + (offset & (2**31 - 1)) * 8
- )
- offset = unpack_from(">Q", self._contents, large_offset_pos)[0]
- assert isinstance(offset, int)
- return offset
- def _unpack_crc32_checksum(self, i: int) -> int:
- result = unpack_from(">L", self._contents, self._crc32_table_offset + i * 4)[0]
- assert isinstance(result, int)
- return result
- class PackIndex3(FilePackIndex):
- """Version 3 Pack Index file.
- Supports variable hash sizes for SHA-1 (20 bytes) and SHA-256 (32 bytes).
- """
- def __init__(
- self,
- filename: str | os.PathLike[str],
- file: IO[bytes] | _GitFile | None = None,
- contents: bytes | None = None,
- size: int | None = None,
- ) -> None:
- """Initialize a version 3 pack index.
- Args:
- filename: Path to the index file
- file: Optional file object
- contents: Optional mmap'd contents
- size: Optional size of the index
- """
- super().__init__(filename, file, contents, size)
- if self._contents[:4] != b"\377tOc":
- raise AssertionError("Not a v3 pack index file")
- (self.version,) = unpack_from(b">L", self._contents, 4)
- if self.version != 3:
- raise AssertionError(f"Version was {self.version}")
- # Read hash algorithm identifier (1 = SHA-1, 2 = SHA-256)
- (self.hash_algorithm,) = unpack_from(b">L", self._contents, 8)
- if self.hash_algorithm == 1:
- self.hash_size = 20 # SHA-1
- elif self.hash_algorithm == 2:
- self.hash_size = 32 # SHA-256
- else:
- raise AssertionError(f"Unknown hash algorithm {self.hash_algorithm}")
- # Read length of shortened object names
- (self.shortened_oid_len,) = unpack_from(b">L", self._contents, 12)
- # Calculate offsets based on variable hash size
- self._fan_out_table = self._read_fan_out_table(
- 16
- ) # After header (4 + 4 + 4 + 4)
- self._name_table_offset = 16 + 0x100 * 4
- self._crc32_table_offset = self._name_table_offset + self.hash_size * len(self)
- self._pack_offset_table_offset = self._crc32_table_offset + 4 * len(self)
- self._pack_offset_largetable_offset = self._pack_offset_table_offset + 4 * len(
- self
- )
- def _unpack_entry(self, i: int) -> tuple[bytes, int, int]:
- return (
- self._unpack_name(i),
- self._unpack_offset(i),
- self._unpack_crc32_checksum(i),
- )
- def _unpack_name(self, i: int) -> bytes:
- offset = self._name_table_offset + i * self.hash_size
- return self._contents[offset : offset + self.hash_size]
- def _unpack_offset(self, i: int) -> int:
- offset_pos = self._pack_offset_table_offset + i * 4
- offset = unpack_from(">L", self._contents, offset_pos)[0]
- assert isinstance(offset, int)
- if offset & (2**31):
- large_offset_pos = (
- self._pack_offset_largetable_offset + (offset & (2**31 - 1)) * 8
- )
- offset = unpack_from(">Q", self._contents, large_offset_pos)[0]
- assert isinstance(offset, int)
- return offset
- def _unpack_crc32_checksum(self, i: int) -> int:
- result = unpack_from(">L", self._contents, self._crc32_table_offset + i * 4)[0]
- assert isinstance(result, int)
- return result
- def read_pack_header(read: Callable[[int], bytes]) -> tuple[int, int]:
- """Read the header of a pack file.
- Args:
- read: Read function
- Returns: Tuple of (pack version, number of objects). If no data is
- available to read, returns (None, None).
- """
- header = read(12)
- if not header:
- raise AssertionError("file too short to contain pack")
- if header[:4] != b"PACK":
- raise AssertionError(f"Invalid pack header {header!r}")
- (version,) = unpack_from(b">L", header, 4)
- if version not in (2, 3):
- raise AssertionError(f"Version was {version}")
- (num_objects,) = unpack_from(b">L", header, 8)
- return (version, num_objects)
- def chunks_length(chunks: bytes | Iterable[bytes]) -> int:
- """Get the total length of a sequence of chunks.
- Args:
- chunks: Either a single bytes object or an iterable of bytes
- Returns: Total length in bytes
- """
- if isinstance(chunks, bytes):
- return len(chunks)
- else:
- return sum(map(len, chunks))
- def unpack_object(
- read_all: Callable[[int], bytes],
- read_some: Callable[[int], bytes] | None = None,
- compute_crc32: bool = False,
- include_comp: bool = False,
- zlib_bufsize: int = _ZLIB_BUFSIZE,
- ) -> tuple[UnpackedObject, bytes]:
- """Unpack a Git object.
- Args:
- read_all: Read function that blocks until the number of requested
- bytes are read.
- read_some: Read function that returns at least one byte, but may not
- return the number of bytes requested.
- compute_crc32: If True, compute the CRC32 of the compressed data. If
- False, the returned CRC32 will be None.
- include_comp: If True, include compressed data in the result.
- zlib_bufsize: An optional buffer size for zlib operations.
- Returns: A tuple of (unpacked, unused), where unused is the unused data
- leftover from decompression, and unpacked in an UnpackedObject with
- the following attrs set:
- * obj_chunks (for non-delta types)
- * pack_type_num
- * delta_base (for delta types)
- * comp_chunks (if include_comp is True)
- * decomp_chunks
- * decomp_len
- * crc32 (if compute_crc32 is True)
- """
- if read_some is None:
- read_some = read_all
- if compute_crc32:
- crc32 = 0
- else:
- crc32 = None
- raw, crc32 = take_msb_bytes(read_all, crc32=crc32)
- type_num = (raw[0] >> 4) & 0x07
- size = raw[0] & 0x0F
- for i, byte in enumerate(raw[1:]):
- size += (byte & 0x7F) << ((i * 7) + 4)
- delta_base: int | bytes | None
- raw_base = len(raw)
- if type_num == OFS_DELTA:
- raw, crc32 = take_msb_bytes(read_all, crc32=crc32)
- raw_base += len(raw)
- if raw[-1] & 0x80:
- raise AssertionError
- delta_base_offset = raw[0] & 0x7F
- for byte in raw[1:]:
- delta_base_offset += 1
- delta_base_offset <<= 7
- delta_base_offset += byte & 0x7F
- delta_base = delta_base_offset
- elif type_num == REF_DELTA:
- delta_base_obj = read_all(20)
- if crc32 is not None:
- crc32 = binascii.crc32(delta_base_obj, crc32)
- delta_base = delta_base_obj
- raw_base += 20
- else:
- delta_base = None
- unpacked = UnpackedObject(
- type_num, delta_base=delta_base, decomp_len=size, crc32=crc32
- )
- unused = read_zlib_chunks(
- read_some,
- unpacked,
- buffer_size=zlib_bufsize,
- include_comp=include_comp,
- )
- return unpacked, unused
- def _compute_object_size(value: tuple[int, Any]) -> int:
- """Compute the size of a unresolved object for use with LRUSizeCache."""
- (num, obj) = value
- if num in DELTA_TYPES:
- return chunks_length(obj[1])
- return chunks_length(obj)
- class PackStreamReader:
- """Class to read a pack stream.
- The pack is read from a ReceivableProtocol using read() or recv() as
- appropriate.
- """
- def __init__(
- self,
- read_all: Callable[[int], bytes],
- read_some: Callable[[int], bytes] | None = None,
- zlib_bufsize: int = _ZLIB_BUFSIZE,
- ) -> None:
- """Initialize pack stream reader.
- Args:
- read_all: Function to read all requested bytes
- read_some: Function to read some bytes (optional)
- zlib_bufsize: Buffer size for zlib decompression
- """
- self.read_all = read_all
- if read_some is None:
- self.read_some = read_all
- else:
- self.read_some = read_some
- self.sha = sha1()
- self._offset = 0
- self._rbuf = BytesIO()
- # trailer is a deque to avoid memory allocation on small reads
- self._trailer: deque[int] = deque()
- self._zlib_bufsize = zlib_bufsize
- def _read(self, read: Callable[[int], bytes], size: int) -> bytes:
- """Read up to size bytes using the given callback.
- As a side effect, update the verifier's hash (excluding the last 20
- bytes read).
- Args:
- read: The read callback to read from.
- size: The maximum number of bytes to read; the particular
- behavior is callback-specific.
- Returns: Bytes read
- """
- data = read(size)
- # maintain a trailer of the last 20 bytes we've read
- n = len(data)
- self._offset += n
- tn = len(self._trailer)
- if n >= 20:
- to_pop = tn
- to_add = 20
- else:
- to_pop = max(n + tn - 20, 0)
- to_add = n
- self.sha.update(
- bytes(bytearray([self._trailer.popleft() for _ in range(to_pop)]))
- )
- self._trailer.extend(data[-to_add:])
- # hash everything but the trailer
- self.sha.update(data[:-to_add])
- return data
- def _buf_len(self) -> int:
- buf = self._rbuf
- start = buf.tell()
- buf.seek(0, SEEK_END)
- end = buf.tell()
- buf.seek(start)
- return end - start
- @property
- def offset(self) -> int:
- """Return current offset in the stream."""
- return self._offset - self._buf_len()
- def read(self, size: int) -> bytes:
- """Read, blocking until size bytes are read."""
- buf_len = self._buf_len()
- if buf_len >= size:
- return self._rbuf.read(size)
- buf_data = self._rbuf.read()
- self._rbuf = BytesIO()
- return buf_data + self._read(self.read_all, size - buf_len)
- def recv(self, size: int) -> bytes:
- """Read up to size bytes, blocking until one byte is read."""
- buf_len = self._buf_len()
- if buf_len:
- data = self._rbuf.read(size)
- if size >= buf_len:
- self._rbuf = BytesIO()
- return data
- return self._read(self.read_some, size)
- def __len__(self) -> int:
- """Return the number of objects in this pack."""
- return self._num_objects
- def read_objects(self, compute_crc32: bool = False) -> Iterator[UnpackedObject]:
- """Read the objects in this pack file.
- Args:
- compute_crc32: If True, compute the CRC32 of the compressed
- data. If False, the returned CRC32 will be None.
- Returns: Iterator over UnpackedObjects with the following members set:
- offset
- obj_type_num
- obj_chunks (for non-delta types)
- delta_base (for delta types)
- decomp_chunks
- decomp_len
- crc32 (if compute_crc32 is True)
- Raises:
- ChecksumMismatch: if the checksum of the pack contents does not
- match the checksum in the pack trailer.
- zlib.error: if an error occurred during zlib decompression.
- IOError: if an error occurred writing to the output file.
- """
- _pack_version, self._num_objects = read_pack_header(self.read)
- for _ in range(self._num_objects):
- offset = self.offset
- unpacked, unused = unpack_object(
- self.read,
- read_some=self.recv,
- compute_crc32=compute_crc32,
- zlib_bufsize=self._zlib_bufsize,
- )
- unpacked.offset = offset
- # prepend any unused data to current read buffer
- buf = BytesIO()
- buf.write(unused)
- buf.write(self._rbuf.read())
- buf.seek(0)
- self._rbuf = buf
- yield unpacked
- if self._buf_len() < 20:
- # If the read buffer is full, then the last read() got the whole
- # trailer off the wire. If not, it means there is still some of the
- # trailer to read. We need to read() all 20 bytes; N come from the
- # read buffer and (20 - N) come from the wire.
- self.read(20)
- pack_sha = bytearray(self._trailer)
- if pack_sha != self.sha.digest():
- raise ChecksumMismatch(sha_to_hex(bytes(pack_sha)), self.sha.hexdigest())
- class PackStreamCopier(PackStreamReader):
- """Class to verify a pack stream as it is being read.
- The pack is read from a ReceivableProtocol using read() or recv() as
- appropriate and written out to the given file-like object.
- """
- def __init__(
- self,
- read_all: Callable[[int], bytes],
- read_some: Callable[[int], bytes] | None,
- outfile: IO[bytes],
- delta_iter: "DeltaChainIterator[UnpackedObject]" | None = None,
- ) -> None:
- """Initialize the copier.
- Args:
- read_all: Read function that blocks until the number of
- requested bytes are read.
- read_some: Read function that returns at least one byte, but may
- not return the number of bytes requested.
- outfile: File-like object to write output through.
- delta_iter: Optional DeltaChainIterator to record deltas as we
- read them.
- """
- super().__init__(read_all, read_some=read_some)
- self.outfile = outfile
- self._delta_iter = delta_iter
- def _read(self, read: Callable[[int], bytes], size: int) -> bytes:
- """Read data from the read callback and write it to the file."""
- data = super()._read(read, size)
- self.outfile.write(data)
- return data
- def verify(self, progress: Callable[..., None] | None = None) -> None:
- """Verify a pack stream and write it to the output file.
- See PackStreamReader.iterobjects for a list of exceptions this may
- throw.
- """
- i = 0 # default count of entries if read_objects() is empty
- for i, unpacked in enumerate(self.read_objects()):
- if self._delta_iter:
- self._delta_iter.record(unpacked)
- if progress is not None:
- progress(f"copying pack entries: {i}/{len(self)}\r".encode("ascii"))
- if progress is not None:
- progress(f"copied {i} pack entries\n".encode("ascii"))
- def obj_sha(type: int, chunks: bytes | Iterable[bytes]) -> bytes:
- """Compute the SHA for a numeric type and object chunks."""
- sha = sha1()
- sha.update(object_header(type, chunks_length(chunks)))
- if isinstance(chunks, bytes):
- sha.update(chunks)
- else:
- for chunk in chunks:
- sha.update(chunk)
- return sha.digest()
- def compute_file_sha(
- f: IO[bytes], start_ofs: int = 0, end_ofs: int = 0, buffer_size: int = 1 << 16
- ) -> "HashObject":
- """Hash a portion of a file into a new SHA.
- Args:
- f: A file-like object to read from that supports seek().
- start_ofs: The offset in the file to start reading at.
- end_ofs: The offset in the file to end reading at, relative to the
- end of the file.
- buffer_size: A buffer size for reading.
- Returns: A new SHA object updated with data read from the file.
- """
- sha = sha1()
- f.seek(0, SEEK_END)
- length = f.tell()
- if (end_ofs < 0 and length + end_ofs < start_ofs) or end_ofs > length:
- raise AssertionError(
- f"Attempt to read beyond file length. start_ofs: {start_ofs}, end_ofs: {end_ofs}, file length: {length}"
- )
- todo = length + end_ofs - start_ofs
- f.seek(start_ofs)
- while todo:
- data = f.read(min(todo, buffer_size))
- sha.update(data)
- todo -= len(data)
- return sha
- class PackData:
- """The data contained in a packfile.
- Pack files can be accessed both sequentially for exploding a pack, and
- directly with the help of an index to retrieve a specific object.
- The objects within are either complete or a delta against another.
- The header is variable length. If the MSB of each byte is set then it
- indicates that the subsequent byte is still part of the header.
- For the first byte the next MS bits are the type, which tells you the type
- of object, and whether it is a delta. The LS byte is the lowest bits of the
- size. For each subsequent byte the LS 7 bits are the next MS bits of the
- size, i.e. the last byte of the header contains the MS bits of the size.
- For the complete objects the data is stored as zlib deflated data.
- The size in the header is the uncompressed object size, so to uncompress
- you need to just keep feeding data to zlib until you get an object back,
- or it errors on bad data. This is done here by just giving the complete
- buffer from the start of the deflated object on. This is bad, but until I
- get mmap sorted out it will have to do.
- Currently there are no integrity checks done. Also no attempt is made to
- try and detect the delta case, or a request for an object at the wrong
- position. It will all just throw a zlib or KeyError.
- """
- def __init__(
- self,
- filename: str | os.PathLike[str],
- file: IO[bytes] | None = None,
- size: int | None = None,
- *,
- delta_window_size: int | None = None,
- window_memory: int | None = None,
- delta_cache_size: int | None = None,
- depth: int | None = None,
- threads: int | None = None,
- big_file_threshold: int | None = None,
- ) -> None:
- """Create a PackData object representing the pack in the given filename.
- The file must exist and stay readable until the object is disposed of.
- It must also stay the same size. It will be mapped whenever needed.
- Currently there is a restriction on the size of the pack as the python
- mmap implementation is flawed.
- """
- self._filename = filename
- self._size = size
- self._header_size = 12
- self.delta_window_size = delta_window_size
- self.window_memory = window_memory
- self.delta_cache_size = delta_cache_size
- self.depth = depth
- self.threads = threads
- self.big_file_threshold = big_file_threshold
- self._file: IO[bytes]
- if file is None:
- self._file = GitFile(self._filename, "rb")
- else:
- self._file = file
- (_version, self._num_objects) = read_pack_header(self._file.read)
- # Use delta_cache_size config if available, otherwise default
- cache_size = delta_cache_size or (1024 * 1024 * 20)
- self._offset_cache = LRUSizeCache[int, tuple[int, OldUnpackedObject]](
- cache_size, compute_size=_compute_object_size
- )
- @property
- def filename(self) -> str:
- """Get the filename of the pack file.
- Returns:
- Base filename without directory path
- """
- return os.path.basename(self._filename)
- @property
- def path(self) -> str | os.PathLike[str]:
- """Get the full path of the pack file.
- Returns:
- Full path to the pack file
- """
- return self._filename
- @classmethod
- def from_file(cls, file: IO[bytes], size: int | None = None) -> "PackData":
- """Create a PackData object from an open file.
- Args:
- file: Open file object
- size: Optional file size
- Returns:
- PackData instance
- """
- return cls(str(file), file=file, size=size)
- @classmethod
- def from_path(cls, path: str | os.PathLike[str]) -> "PackData":
- """Create a PackData object from a file path.
- Args:
- path: Path to the pack file
- Returns:
- PackData instance
- """
- return cls(filename=path)
- def close(self) -> None:
- """Close the underlying pack file."""
- self._file.close()
- def __enter__(self) -> "PackData":
- """Enter context manager."""
- return self
- def __exit__(
- self,
- exc_type: type | None,
- exc_val: BaseException | None,
- exc_tb: TracebackType | None,
- ) -> None:
- """Exit context manager."""
- self.close()
- def __eq__(self, other: object) -> bool:
- """Check equality with another object."""
- if isinstance(other, PackData):
- return self.get_stored_checksum() == other.get_stored_checksum()
- return False
- def _get_size(self) -> int:
- if self._size is not None:
- return self._size
- self._size = os.path.getsize(self._filename)
- if self._size < self._header_size:
- errmsg = f"{self._filename} is too small for a packfile ({self._size} < {self._header_size})"
- raise AssertionError(errmsg)
- return self._size
- def __len__(self) -> int:
- """Returns the number of objects in this pack."""
- return self._num_objects
- def calculate_checksum(self) -> bytes:
- """Calculate the checksum for this pack.
- Returns: 20-byte binary SHA1 digest
- """
- return compute_file_sha(self._file, end_ofs=-20).digest()
- def iter_unpacked(self, *, include_comp: bool = False) -> Iterator[UnpackedObject]:
- """Iterate over unpacked objects in the pack."""
- self._file.seek(self._header_size)
- if self._num_objects is None:
- return
- for _ in range(self._num_objects):
- offset = self._file.tell()
- unpacked, unused = unpack_object(
- self._file.read, compute_crc32=False, include_comp=include_comp
- )
- unpacked.offset = offset
- yield unpacked
- # Back up over unused data.
- self._file.seek(-len(unused), SEEK_CUR)
- def iterentries(
- self,
- progress: Callable[[int, int], None] | None = None,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- ) -> Iterator[tuple[bytes, int, int | None]]:
- """Yield entries summarizing the contents of this pack.
- Args:
- progress: Progress function, called with current and total
- object count.
- resolve_ext_ref: Optional function to resolve external references
- Returns: iterator of tuples with (sha, offset, crc32)
- """
- num_objects = self._num_objects
- indexer = PackIndexer.for_pack_data(self, resolve_ext_ref=resolve_ext_ref)
- for i, result in enumerate(indexer):
- if progress is not None:
- progress(i, num_objects)
- yield result
- def sorted_entries(
- self,
- progress: ProgressFn | None = None,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- ) -> list[tuple[bytes, int, int]]:
- """Return entries in this pack, sorted by SHA.
- Args:
- progress: Progress function, called with current and total
- object count
- resolve_ext_ref: Optional function to resolve external references
- Returns: Iterator of tuples with (sha, offset, crc32)
- """
- return sorted(
- self.iterentries(progress=progress, resolve_ext_ref=resolve_ext_ref) # type: ignore
- )
- def create_index_v1(
- self,
- filename: str,
- progress: Callable[..., None] | None = None,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- ) -> bytes:
- """Create a version 1 file for this data file.
- Args:
- filename: Index filename.
- progress: Progress report function
- resolve_ext_ref: Optional function to resolve external references
- Returns: Checksum of index file
- """
- entries = self.sorted_entries(
- progress=progress, resolve_ext_ref=resolve_ext_ref
- )
- checksum = self.calculate_checksum()
- with GitFile(filename, "wb") as f:
- write_pack_index_v1(
- f,
- entries,
- checksum,
- )
- return checksum
- def create_index_v2(
- self,
- filename: str,
- progress: Callable[..., None] | None = None,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- ) -> bytes:
- """Create a version 2 index file for this data file.
- Args:
- filename: Index filename.
- progress: Progress report function
- resolve_ext_ref: Optional function to resolve external references
- Returns: Checksum of index file
- """
- entries = self.sorted_entries(
- progress=progress, resolve_ext_ref=resolve_ext_ref
- )
- with GitFile(filename, "wb") as f:
- return write_pack_index_v2(f, entries, self.calculate_checksum())
- def create_index_v3(
- self,
- filename: str,
- progress: Callable[..., None] | None = None,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- hash_algorithm: int = 1,
- ) -> bytes:
- """Create a version 3 index file for this data file.
- Args:
- filename: Index filename.
- progress: Progress report function
- resolve_ext_ref: Function to resolve external references
- hash_algorithm: Hash algorithm identifier (1 = SHA-1, 2 = SHA-256)
- Returns: Checksum of index file
- """
- entries = self.sorted_entries(
- progress=progress, resolve_ext_ref=resolve_ext_ref
- )
- with GitFile(filename, "wb") as f:
- return write_pack_index_v3(
- f, entries, self.calculate_checksum(), hash_algorithm
- )
- def create_index(
- self,
- filename: str,
- progress: Callable[..., None] | None = None,
- version: int = 2,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- hash_algorithm: int = 1,
- ) -> bytes:
- """Create an index file for this data file.
- Args:
- filename: Index filename.
- progress: Progress report function
- version: Index version (1, 2, or 3)
- resolve_ext_ref: Function to resolve external references
- hash_algorithm: Hash algorithm identifier for v3 (1 = SHA-1, 2 = SHA-256)
- Returns: Checksum of index file
- """
- if version == 1:
- return self.create_index_v1(
- filename, progress, resolve_ext_ref=resolve_ext_ref
- )
- elif version == 2:
- return self.create_index_v2(
- filename, progress, resolve_ext_ref=resolve_ext_ref
- )
- elif version == 3:
- return self.create_index_v3(
- filename,
- progress,
- resolve_ext_ref=resolve_ext_ref,
- hash_algorithm=hash_algorithm,
- )
- else:
- raise ValueError(f"unknown index format {version}")
- def get_stored_checksum(self) -> bytes:
- """Return the expected checksum stored in this pack."""
- self._file.seek(-20, SEEK_END)
- return self._file.read(20)
- def check(self) -> None:
- """Check the consistency of this pack."""
- actual = self.calculate_checksum()
- stored = self.get_stored_checksum()
- if actual != stored:
- raise ChecksumMismatch(stored, actual)
- def get_unpacked_object_at(
- self, offset: int, *, include_comp: bool = False
- ) -> UnpackedObject:
- """Given offset in the packfile return a UnpackedObject."""
- assert offset >= self._header_size
- self._file.seek(offset)
- unpacked, _ = unpack_object(self._file.read, include_comp=include_comp)
- unpacked.offset = offset
- return unpacked
- def get_object_at(self, offset: int) -> tuple[int, OldUnpackedObject]:
- """Given an offset in to the packfile return the object that is there.
- Using the associated index the location of an object can be looked up,
- and then the packfile can be asked directly for that object using this
- function.
- """
- try:
- return self._offset_cache[offset]
- except KeyError:
- pass
- unpacked = self.get_unpacked_object_at(offset, include_comp=False)
- return (unpacked.pack_type_num, unpacked._obj())
- T = TypeVar("T")
- class DeltaChainIterator(Generic[T]):
- """Abstract iterator over pack data based on delta chains.
- Each object in the pack is guaranteed to be inflated exactly once,
- regardless of how many objects reference it as a delta base. As a result,
- memory usage is proportional to the length of the longest delta chain.
- Subclasses can override _result to define the result type of the iterator.
- By default, results are UnpackedObjects with the following members set:
- * offset
- * obj_type_num
- * obj_chunks
- * pack_type_num
- * delta_base (for delta types)
- * comp_chunks (if _include_comp is True)
- * decomp_chunks
- * decomp_len
- * crc32 (if _compute_crc32 is True)
- """
- _compute_crc32 = False
- _include_comp = False
- def __init__(
- self,
- file_obj: IO[bytes] | None,
- *,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- ) -> None:
- """Initialize DeltaChainIterator.
- Args:
- file_obj: File object to read pack data from
- resolve_ext_ref: Optional function to resolve external references
- """
- self._file = file_obj
- self._resolve_ext_ref = resolve_ext_ref
- self._pending_ofs: dict[int, list[int]] = defaultdict(list)
- self._pending_ref: dict[bytes, list[int]] = defaultdict(list)
- self._full_ofs: list[tuple[int, int]] = []
- self._ext_refs: list[bytes] = []
- @classmethod
- def for_pack_data(
- cls, pack_data: PackData, resolve_ext_ref: ResolveExtRefFn | None = None
- ) -> "DeltaChainIterator[T]":
- """Create a DeltaChainIterator from pack data.
- Args:
- pack_data: PackData object to iterate
- resolve_ext_ref: Optional function to resolve external refs
- Returns:
- DeltaChainIterator instance
- """
- walker = cls(None, resolve_ext_ref=resolve_ext_ref)
- walker.set_pack_data(pack_data)
- for unpacked in pack_data.iter_unpacked(include_comp=False):
- walker.record(unpacked)
- return walker
- @classmethod
- def for_pack_subset(
- cls,
- pack: "Pack",
- shas: Iterable[bytes],
- *,
- allow_missing: bool = False,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- ) -> "DeltaChainIterator[T]":
- """Create a DeltaChainIterator for a subset of objects.
- Args:
- pack: Pack object containing the data
- shas: Iterable of object SHAs to include
- allow_missing: If True, skip missing objects
- resolve_ext_ref: Optional function to resolve external refs
- Returns:
- DeltaChainIterator instance
- """
- walker = cls(None, resolve_ext_ref=resolve_ext_ref)
- walker.set_pack_data(pack.data)
- todo = set()
- for sha in shas:
- assert isinstance(sha, bytes)
- try:
- off = pack.index.object_offset(sha)
- except KeyError:
- if not allow_missing:
- raise
- else:
- todo.add(off)
- done = set()
- while todo:
- off = todo.pop()
- unpacked = pack.data.get_unpacked_object_at(off)
- walker.record(unpacked)
- done.add(off)
- base_ofs = None
- if unpacked.pack_type_num == OFS_DELTA:
- assert unpacked.offset is not None
- assert unpacked.delta_base is not None
- assert isinstance(unpacked.delta_base, int)
- base_ofs = unpacked.offset - unpacked.delta_base
- elif unpacked.pack_type_num == REF_DELTA:
- with suppress(KeyError):
- assert isinstance(unpacked.delta_base, bytes)
- base_ofs = pack.index.object_index(unpacked.delta_base)
- if base_ofs is not None and base_ofs not in done:
- todo.add(base_ofs)
- return walker
- def record(self, unpacked: UnpackedObject) -> None:
- """Record an unpacked object for later processing.
- Args:
- unpacked: UnpackedObject to record
- """
- type_num = unpacked.pack_type_num
- offset = unpacked.offset
- assert offset is not None
- if type_num == OFS_DELTA:
- assert unpacked.delta_base is not None
- assert isinstance(unpacked.delta_base, int)
- base_offset = offset - unpacked.delta_base
- self._pending_ofs[base_offset].append(offset)
- elif type_num == REF_DELTA:
- assert isinstance(unpacked.delta_base, bytes)
- self._pending_ref[unpacked.delta_base].append(offset)
- else:
- self._full_ofs.append((offset, type_num))
- def set_pack_data(self, pack_data: PackData) -> None:
- """Set the pack data for iteration.
- Args:
- pack_data: PackData object to use
- """
- self._file = pack_data._file
- def _walk_all_chains(self) -> Iterator[T]:
- for offset, type_num in self._full_ofs:
- yield from self._follow_chain(offset, type_num, None)
- yield from self._walk_ref_chains()
- assert not self._pending_ofs, repr(self._pending_ofs)
- def _ensure_no_pending(self) -> None:
- if self._pending_ref:
- raise UnresolvedDeltas([sha_to_hex(s) for s in self._pending_ref])
- def _walk_ref_chains(self) -> Iterator[T]:
- if not self._resolve_ext_ref:
- self._ensure_no_pending()
- return
- for base_sha, pending in sorted(self._pending_ref.items()):
- if base_sha not in self._pending_ref:
- continue
- try:
- type_num, chunks = self._resolve_ext_ref(base_sha)
- except KeyError:
- # Not an external ref, but may depend on one. Either it will
- # get popped via a _follow_chain call, or we will raise an
- # error below.
- continue
- self._ext_refs.append(base_sha)
- self._pending_ref.pop(base_sha)
- for new_offset in pending:
- yield from self._follow_chain(new_offset, type_num, chunks) # type: ignore[arg-type]
- self._ensure_no_pending()
- def _result(self, unpacked: UnpackedObject) -> T:
- raise NotImplementedError
- def _resolve_object(
- self, offset: int, obj_type_num: int, base_chunks: list[bytes] | None
- ) -> UnpackedObject:
- assert self._file is not None
- self._file.seek(offset)
- unpacked, _ = unpack_object(
- self._file.read,
- include_comp=self._include_comp,
- compute_crc32=self._compute_crc32,
- )
- unpacked.offset = offset
- if base_chunks is None:
- assert unpacked.pack_type_num == obj_type_num
- else:
- assert unpacked.pack_type_num in DELTA_TYPES
- unpacked.obj_type_num = obj_type_num
- unpacked.obj_chunks = apply_delta(base_chunks, unpacked.decomp_chunks)
- return unpacked
- def _follow_chain(
- self, offset: int, obj_type_num: int, base_chunks: list[bytes] | None
- ) -> Iterator[T]:
- # Unlike PackData.get_object_at, there is no need to cache offsets as
- # this approach by design inflates each object exactly once.
- todo = [(offset, obj_type_num, base_chunks)]
- while todo:
- (offset, obj_type_num, base_chunks) = todo.pop()
- unpacked = self._resolve_object(offset, obj_type_num, base_chunks)
- yield self._result(unpacked)
- assert unpacked.offset is not None
- unblocked = chain(
- self._pending_ofs.pop(unpacked.offset, []),
- self._pending_ref.pop(unpacked.sha(), []),
- )
- todo.extend(
- (new_offset, unpacked.obj_type_num, unpacked.obj_chunks) # type: ignore
- for new_offset in unblocked
- )
- def __iter__(self) -> Iterator[T]:
- """Iterate over objects in the pack."""
- return self._walk_all_chains()
- def ext_refs(self) -> list[bytes]:
- """Return external references."""
- return self._ext_refs
- class UnpackedObjectIterator(DeltaChainIterator[UnpackedObject]):
- """Delta chain iterator that yield unpacked objects."""
- def _result(self, unpacked: UnpackedObject) -> UnpackedObject:
- """Return the unpacked object.
- Args:
- unpacked: The unpacked object
- Returns:
- The unpacked object unchanged
- """
- return unpacked
- class PackIndexer(DeltaChainIterator[PackIndexEntry]):
- """Delta chain iterator that yields index entries."""
- _compute_crc32 = True
- def _result(self, unpacked: UnpackedObject) -> tuple[bytes, int, int | None]:
- """Convert unpacked object to pack index entry.
- Args:
- unpacked: The unpacked object
- Returns:
- Tuple of (sha, offset, crc32) for index entry
- """
- assert unpacked.offset is not None
- return unpacked.sha(), unpacked.offset, unpacked.crc32
- class PackInflater(DeltaChainIterator[ShaFile]):
- """Delta chain iterator that yields ShaFile objects."""
- def _result(self, unpacked: UnpackedObject) -> ShaFile:
- """Convert unpacked object to ShaFile.
- Args:
- unpacked: The unpacked object
- Returns:
- ShaFile object from the unpacked data
- """
- return unpacked.sha_file()
- class SHA1Reader(BinaryIO):
- """Wrapper for file-like object that remembers the SHA1 of its data."""
- def __init__(self, f: IO[bytes]) -> None:
- """Initialize SHA1Reader.
- Args:
- f: File-like object to wrap
- """
- self.f = f
- self.sha1 = sha1(b"")
- def read(self, size: int = -1) -> bytes:
- """Read bytes and update SHA1.
- Args:
- size: Number of bytes to read, -1 for all
- Returns:
- Bytes read from file
- """
- data = self.f.read(size)
- self.sha1.update(data)
- return data
- def check_sha(self, allow_empty: bool = False) -> None:
- """Check if the SHA1 matches the expected value.
- Args:
- allow_empty: Allow empty SHA1 hash
- Raises:
- ChecksumMismatch: If SHA1 doesn't match
- """
- stored = self.f.read(20)
- # If git option index.skipHash is set the index will be empty
- if stored != self.sha1.digest() and (
- not allow_empty
- or sha_to_hex(stored) != b"0000000000000000000000000000000000000000"
- ):
- raise ChecksumMismatch(self.sha1.hexdigest(), sha_to_hex(stored))
- def close(self) -> None:
- """Close the underlying file."""
- return self.f.close()
- def tell(self) -> int:
- """Return current file position."""
- return self.f.tell()
- # BinaryIO abstract methods
- def readable(self) -> bool:
- """Check if file is readable."""
- return True
- def writable(self) -> bool:
- """Check if file is writable."""
- return False
- def seekable(self) -> bool:
- """Check if file is seekable."""
- return getattr(self.f, "seekable", lambda: False)()
- def seek(self, offset: int, whence: int = 0) -> int:
- """Seek to position in file.
- Args:
- offset: Position offset
- whence: Reference point (0=start, 1=current, 2=end)
- Returns:
- New file position
- """
- return self.f.seek(offset, whence)
- def flush(self) -> None:
- """Flush the file buffer."""
- if hasattr(self.f, "flush"):
- self.f.flush()
- def readline(self, size: int = -1) -> bytes:
- """Read a line from the file.
- Args:
- size: Maximum bytes to read
- Returns:
- Line read from file
- """
- return self.f.readline(size)
- def readlines(self, hint: int = -1) -> list[bytes]:
- """Read all lines from the file.
- Args:
- hint: Approximate number of bytes to read
- Returns:
- List of lines
- """
- return self.f.readlines(hint)
- def writelines(self, lines: Iterable[bytes], /) -> None: # type: ignore[override]
- """Write multiple lines to the file (not supported)."""
- raise UnsupportedOperation("writelines")
- def write(self, data: bytes, /) -> int: # type: ignore[override]
- """Write data to the file (not supported)."""
- raise UnsupportedOperation("write")
- def __enter__(self) -> "SHA1Reader":
- """Enter context manager."""
- return self
- def __exit__(
- self,
- type: type | None,
- value: BaseException | None,
- traceback: TracebackType | None,
- ) -> None:
- """Exit context manager and close file."""
- self.close()
- def __iter__(self) -> "SHA1Reader":
- """Return iterator for reading file lines."""
- return self
- def __next__(self) -> bytes:
- """Get next line from file.
- Returns:
- Next line
- Raises:
- StopIteration: When no more lines
- """
- line = self.readline()
- if not line:
- raise StopIteration
- return line
- def fileno(self) -> int:
- """Return file descriptor number."""
- return self.f.fileno()
- def isatty(self) -> bool:
- """Check if file is a terminal."""
- return getattr(self.f, "isatty", lambda: False)()
- def truncate(self, size: int | None = None) -> int:
- """Not supported for read-only file.
- Raises:
- UnsupportedOperation: Always raised
- """
- raise UnsupportedOperation("truncate")
- class SHA1Writer(BinaryIO):
- """Wrapper for file-like object that remembers the SHA1 of its data."""
- def __init__(self, f: BinaryIO | IO[bytes]) -> None:
- """Initialize SHA1Writer.
- Args:
- f: File-like object to wrap
- """
- self.f = f
- self.length = 0
- self.sha1 = sha1(b"")
- self.digest: bytes | None = None
- def write(self, data: bytes | bytearray | memoryview, /) -> int: # type: ignore[override]
- """Write data and update SHA1.
- Args:
- data: Data to write
- Returns:
- Number of bytes written
- """
- self.sha1.update(data)
- written = self.f.write(data)
- self.length += written
- return written
- def write_sha(self) -> bytes:
- """Write the SHA1 digest to the file.
- Returns:
- The SHA1 digest bytes
- """
- sha = self.sha1.digest()
- assert len(sha) == 20
- self.f.write(sha)
- self.length += len(sha)
- return sha
- def close(self) -> None:
- """Close the pack file and finalize the SHA."""
- self.digest = self.write_sha()
- self.f.close()
- def offset(self) -> int:
- """Get the total number of bytes written.
- Returns:
- Total bytes written
- """
- return self.length
- def tell(self) -> int:
- """Return current file position."""
- return self.f.tell()
- # BinaryIO abstract methods
- def readable(self) -> bool:
- """Check if file is readable."""
- return False
- def writable(self) -> bool:
- """Check if file is writable."""
- return True
- def seekable(self) -> bool:
- """Check if file is seekable."""
- return getattr(self.f, "seekable", lambda: False)()
- def seek(self, offset: int, whence: int = 0) -> int:
- """Seek to position in file.
- Args:
- offset: Position offset
- whence: Reference point (0=start, 1=current, 2=end)
- Returns:
- New file position
- """
- return self.f.seek(offset, whence)
- def flush(self) -> None:
- """Flush the file buffer."""
- if hasattr(self.f, "flush"):
- self.f.flush()
- def readline(self, size: int = -1) -> bytes:
- """Not supported for write-only file.
- Raises:
- UnsupportedOperation: Always raised
- """
- raise UnsupportedOperation("readline")
- def readlines(self, hint: int = -1) -> list[bytes]:
- """Not supported for write-only file.
- Raises:
- UnsupportedOperation: Always raised
- """
- raise UnsupportedOperation("readlines")
- def writelines(self, lines: Iterable[bytes], /) -> None: # type: ignore[override]
- """Write multiple lines to the file.
- Args:
- lines: Iterable of lines to write
- """
- for line in lines:
- self.write(line)
- def read(self, size: int = -1) -> bytes:
- """Not supported for write-only file.
- Raises:
- UnsupportedOperation: Always raised
- """
- raise UnsupportedOperation("read")
- def __enter__(self) -> "SHA1Writer":
- """Enter context manager."""
- return self
- def __exit__(
- self,
- type: type | None,
- value: BaseException | None,
- traceback: TracebackType | None,
- ) -> None:
- """Exit context manager and close file."""
- self.close()
- def __iter__(self) -> "SHA1Writer":
- """Return iterator."""
- return self
- def __next__(self) -> bytes:
- """Not supported for write-only file.
- Raises:
- UnsupportedOperation: Always raised
- """
- raise UnsupportedOperation("__next__")
- def fileno(self) -> int:
- """Return file descriptor number."""
- return self.f.fileno()
- def isatty(self) -> bool:
- """Check if file is a terminal."""
- return getattr(self.f, "isatty", lambda: False)()
- def truncate(self, size: int | None = None) -> int:
- """Not supported for write-only file.
- Raises:
- UnsupportedOperation: Always raised
- """
- raise UnsupportedOperation("truncate")
- def pack_object_header(
- type_num: int, delta_base: bytes | int | None, size: int
- ) -> bytearray:
- """Create a pack object header for the given object info.
- Args:
- type_num: Numeric type of the object.
- delta_base: Delta base offset or ref, or None for whole objects.
- size: Uncompressed object size.
- Returns: A header for a packed object.
- """
- header = []
- c = (type_num << 4) | (size & 15)
- size >>= 4
- while size:
- header.append(c | 0x80)
- c = size & 0x7F
- size >>= 7
- header.append(c)
- if type_num == OFS_DELTA:
- assert isinstance(delta_base, int)
- ret = [delta_base & 0x7F]
- delta_base >>= 7
- while delta_base:
- delta_base -= 1
- ret.insert(0, 0x80 | (delta_base & 0x7F))
- delta_base >>= 7
- header.extend(ret)
- elif type_num == REF_DELTA:
- assert isinstance(delta_base, bytes)
- assert len(delta_base) == 20
- header += delta_base
- return bytearray(header)
- def pack_object_chunks(
- type: int,
- object: list[bytes] | tuple[bytes | int, list[bytes]],
- compression_level: int = -1,
- ) -> Iterator[bytes]:
- """Generate chunks for a pack object.
- Args:
- type: Numeric type of the object
- object: Object to write
- compression_level: the zlib compression level
- Returns: Chunks
- """
- if type in DELTA_TYPES:
- if isinstance(object, tuple):
- delta_base, object = object
- else:
- raise TypeError("Delta types require a tuple of (delta_base, object)")
- else:
- delta_base = None
- # Convert object to list of bytes chunks
- if isinstance(object, bytes):
- chunks = [object]
- elif isinstance(object, list):
- chunks = object
- elif isinstance(object, ShaFile):
- chunks = object.as_raw_chunks()
- else:
- # Shouldn't reach here with proper typing
- raise TypeError(f"Unexpected object type: {object.__class__.__name__}")
- yield bytes(pack_object_header(type, delta_base, sum(map(len, chunks))))
- compressor = zlib.compressobj(level=compression_level)
- for data in chunks:
- yield compressor.compress(data)
- yield compressor.flush()
- def write_pack_object(
- write: Callable[[bytes], int],
- type: int,
- object: list[bytes] | tuple[bytes | int, list[bytes]],
- sha: "HashObject" | None = None,
- compression_level: int = -1,
- ) -> int:
- """Write pack object to a file.
- Args:
- write: Write function to use
- type: Numeric type of the object
- object: Object to write
- sha: Optional SHA-1 hasher to update
- compression_level: the zlib compression level
- Returns: CRC32 checksum of the written object
- """
- crc32 = 0
- for chunk in pack_object_chunks(type, object, compression_level=compression_level):
- write(chunk)
- if sha is not None:
- sha.update(chunk)
- crc32 = binascii.crc32(chunk, crc32)
- return crc32 & 0xFFFFFFFF
- def write_pack(
- filename: str,
- objects: Sequence[ShaFile] | Sequence[tuple[ShaFile, bytes | None]],
- *,
- deltify: bool | None = None,
- delta_window_size: int | None = None,
- compression_level: int = -1,
- ) -> tuple[bytes, bytes]:
- """Write a new pack data file.
- Args:
- filename: Path to the new pack file (without .pack extension)
- objects: Objects to write to the pack
- delta_window_size: Delta window size
- deltify: Whether to deltify pack objects
- compression_level: the zlib compression level
- Returns: Tuple with checksum of pack file and index file
- """
- with GitFile(filename + ".pack", "wb") as f:
- entries, data_sum = write_pack_objects(
- f,
- objects,
- delta_window_size=delta_window_size,
- deltify=deltify,
- compression_level=compression_level,
- )
- entries_list = sorted([(k, v[0], v[1]) for (k, v) in entries.items()])
- with GitFile(filename + ".idx", "wb") as f:
- idx_sha = write_pack_index(f, entries_list, data_sum)
- return data_sum, idx_sha
- def pack_header_chunks(num_objects: int) -> Iterator[bytes]:
- """Yield chunks for a pack header."""
- yield b"PACK" # Pack header
- yield struct.pack(b">L", 2) # Pack version
- yield struct.pack(b">L", num_objects) # Number of objects in pack
- def write_pack_header(
- write: Callable[[bytes], int] | IO[bytes], num_objects: int
- ) -> None:
- """Write a pack header for the given number of objects."""
- write_fn: Callable[[bytes], int]
- if hasattr(write, "write"):
- write_fn = write.write
- warnings.warn(
- "write_pack_header() now takes a write rather than file argument",
- DeprecationWarning,
- stacklevel=2,
- )
- else:
- write_fn = write
- for chunk in pack_header_chunks(num_objects):
- write_fn(chunk)
- def find_reusable_deltas(
- container: PackedObjectContainer,
- object_ids: Set[bytes],
- *,
- other_haves: Set[bytes] | None = None,
- progress: Callable[..., None] | None = None,
- ) -> Iterator[UnpackedObject]:
- """Find deltas in a pack that can be reused.
- Args:
- container: Pack container to search for deltas
- object_ids: Set of object IDs to find deltas for
- other_haves: Set of other object IDs we have
- progress: Optional progress reporting callback
- Returns:
- Iterator of UnpackedObject entries that can be reused
- """
- if other_haves is None:
- other_haves = set()
- reused = 0
- for i, unpacked in enumerate(
- container.iter_unpacked_subset(
- object_ids, allow_missing=True, convert_ofs_delta=True
- )
- ):
- if progress is not None and i % 1000 == 0:
- progress(f"checking for reusable deltas: {i}/{len(object_ids)}\r".encode())
- if unpacked.pack_type_num == REF_DELTA:
- hexsha = sha_to_hex(unpacked.delta_base) # type: ignore
- if hexsha in object_ids or hexsha in other_haves:
- yield unpacked
- reused += 1
- if progress is not None:
- progress((f"found {reused} deltas to reuse\n").encode())
- def deltify_pack_objects(
- objects: Iterator[ShaFile] | Iterator[tuple[ShaFile, bytes | None]],
- *,
- window_size: int | None = None,
- progress: Callable[..., None] | None = None,
- ) -> Iterator[UnpackedObject]:
- """Generate deltas for pack objects.
- Args:
- objects: An iterable of (object, path) tuples to deltify.
- window_size: Window size; None for default
- progress: Optional progress reporting callback
- Returns: Iterator over type_num, object id, delta_base, content
- delta_base is None for full text entries
- """
- def objects_with_hints() -> Iterator[tuple[ShaFile, tuple[int, bytes | None]]]:
- for e in objects:
- if isinstance(e, ShaFile):
- yield (e, (e.type_num, None))
- else:
- yield (e[0], (e[0].type_num, e[1]))
- sorted_objs = sort_objects_for_delta(objects_with_hints())
- yield from deltas_from_sorted_objects(
- sorted_objs,
- window_size=window_size,
- progress=progress,
- )
- def sort_objects_for_delta(
- objects: Iterator[ShaFile] | Iterator[tuple[ShaFile, PackHint | None]],
- ) -> Iterator[tuple[ShaFile, bytes | None]]:
- """Sort objects for optimal delta compression.
- Args:
- objects: Iterator of objects or (object, hint) tuples
- Returns:
- Iterator of sorted (ShaFile, path) tuples
- """
- magic = []
- for entry in objects:
- if isinstance(entry, tuple):
- obj, hint = entry
- if hint is None:
- type_num = None
- path = None
- else:
- (type_num, path) = hint
- else:
- obj = entry
- type_num = None
- path = None
- magic.append((type_num, path, -obj.raw_length(), obj))
- # Build a list of objects ordered by the magic Linus heuristic
- # This helps us find good objects to diff against us
- magic.sort()
- return ((x[3], x[1]) for x in magic)
- def deltas_from_sorted_objects(
- objects: Iterator[tuple[ShaFile, bytes | None]],
- window_size: int | None = None,
- progress: Callable[..., None] | None = None,
- ) -> Iterator[UnpackedObject]:
- """Create deltas from sorted objects.
- Args:
- objects: Iterator of sorted objects to deltify
- window_size: Delta window size; None for default
- progress: Optional progress reporting callback
- Returns:
- Iterator of UnpackedObject entries
- """
- # TODO(jelmer): Use threads
- if window_size is None:
- window_size = DEFAULT_PACK_DELTA_WINDOW_SIZE
- possible_bases: deque[tuple[bytes, int, list[bytes]]] = deque()
- for i, (o, path) in enumerate(objects):
- if progress is not None and i % 1000 == 0:
- progress((f"generating deltas: {i}\r").encode())
- raw = o.as_raw_chunks()
- winner = raw
- winner_len = sum(map(len, winner))
- winner_base = None
- for base_id, base_type_num, base in possible_bases:
- if base_type_num != o.type_num:
- continue
- delta_len = 0
- delta = []
- for chunk in create_delta(b"".join(base), b"".join(raw)):
- delta_len += len(chunk)
- if delta_len >= winner_len:
- break
- delta.append(chunk)
- else:
- winner_base = base_id
- winner = delta
- winner_len = sum(map(len, winner))
- yield UnpackedObject(
- o.type_num,
- sha=o.sha().digest(),
- delta_base=winner_base,
- decomp_len=winner_len,
- decomp_chunks=winner,
- )
- possible_bases.appendleft((o.sha().digest(), o.type_num, raw))
- while len(possible_bases) > window_size:
- possible_bases.pop()
- def pack_objects_to_data(
- objects: Sequence[ShaFile]
- | Sequence[tuple[ShaFile, bytes | None]]
- | Sequence[tuple[ShaFile, PackHint | None]],
- *,
- deltify: bool | None = None,
- delta_window_size: int | None = None,
- ofs_delta: bool = True,
- progress: Callable[..., None] | None = None,
- ) -> tuple[int, Iterator[UnpackedObject]]:
- """Create pack data from objects.
- Args:
- objects: Pack objects
- deltify: Whether to deltify pack objects
- delta_window_size: Delta window size
- ofs_delta: Whether to use offset deltas
- progress: Optional progress reporting callback
- Returns: Tuples with (type_num, hexdigest, delta base, object chunks)
- """
- count = len(objects)
- if deltify is None:
- # PERFORMANCE/TODO(jelmer): This should be enabled but the python
- # implementation is *much* too slow at the moment.
- # Maybe consider enabling it just if the rust extension is available?
- deltify = False
- if deltify:
- return (
- count,
- deltify_pack_objects(
- iter(objects), # type: ignore
- window_size=delta_window_size,
- progress=progress,
- ),
- )
- else:
- def iter_without_path() -> Iterator[UnpackedObject]:
- for o in objects:
- if isinstance(o, tuple):
- yield full_unpacked_object(o[0])
- else:
- yield full_unpacked_object(o)
- return (count, iter_without_path())
- def generate_unpacked_objects(
- container: PackedObjectContainer,
- object_ids: Sequence[tuple[ObjectID, PackHint | None]],
- delta_window_size: int | None = None,
- deltify: bool | None = None,
- reuse_deltas: bool = True,
- ofs_delta: bool = True,
- other_haves: set[bytes] | None = None,
- progress: Callable[..., None] | None = None,
- ) -> Iterator[UnpackedObject]:
- """Create pack data from objects.
- Returns: Tuples with (type_num, hexdigest, delta base, object chunks)
- """
- todo = dict(object_ids)
- if reuse_deltas:
- for unpack in find_reusable_deltas(
- container, set(todo), other_haves=other_haves, progress=progress
- ):
- del todo[sha_to_hex(unpack.sha())]
- yield unpack
- if deltify is None:
- # PERFORMANCE/TODO(jelmer): This should be enabled but is *much* too
- # slow at the moment.
- deltify = False
- if deltify:
- objects_to_delta = container.iterobjects_subset(
- todo.keys(), allow_missing=False
- )
- sorted_objs = sort_objects_for_delta((o, todo[o.id]) for o in objects_to_delta)
- yield from deltas_from_sorted_objects(
- sorted_objs,
- window_size=delta_window_size,
- progress=progress,
- )
- else:
- for oid in todo:
- yield full_unpacked_object(container[oid])
- def full_unpacked_object(o: ShaFile) -> UnpackedObject:
- """Create an UnpackedObject from a ShaFile.
- Args:
- o: ShaFile object to convert
- Returns:
- UnpackedObject with full object data
- """
- return UnpackedObject(
- o.type_num,
- delta_base=None,
- crc32=None,
- decomp_chunks=o.as_raw_chunks(),
- sha=o.sha().digest(),
- )
- def write_pack_from_container(
- write: Callable[[bytes], None]
- | Callable[[bytes | bytearray | memoryview], int]
- | IO[bytes],
- container: PackedObjectContainer,
- object_ids: Sequence[tuple[ObjectID, PackHint | None]],
- delta_window_size: int | None = None,
- deltify: bool | None = None,
- reuse_deltas: bool = True,
- compression_level: int = -1,
- other_haves: set[bytes] | None = None,
- ) -> tuple[dict[bytes, tuple[int, int]], bytes]:
- """Write a new pack data file.
- Args:
- write: write function to use
- container: PackedObjectContainer
- object_ids: Sequence of (object_id, hint) tuples to write
- delta_window_size: Sliding window size for searching for deltas;
- Set to None for default window size.
- deltify: Whether to deltify objects
- reuse_deltas: Whether to reuse existing deltas
- compression_level: the zlib compression level to use
- other_haves: Set of additional object IDs the receiver has
- Returns: Dict mapping id -> (offset, crc32 checksum), pack checksum
- """
- pack_contents_count = len(object_ids)
- pack_contents = generate_unpacked_objects(
- container,
- object_ids,
- delta_window_size=delta_window_size,
- deltify=deltify,
- reuse_deltas=reuse_deltas,
- other_haves=other_haves,
- )
- return write_pack_data(
- write,
- pack_contents,
- num_records=pack_contents_count,
- compression_level=compression_level,
- )
- def write_pack_objects(
- write: Callable[[bytes], None] | IO[bytes],
- objects: Sequence[ShaFile] | Sequence[tuple[ShaFile, bytes | None]],
- *,
- delta_window_size: int | None = None,
- deltify: bool | None = None,
- compression_level: int = -1,
- ) -> tuple[dict[bytes, tuple[int, int]], bytes]:
- """Write a new pack data file.
- Args:
- write: write function to use
- objects: Sequence of (object, path) tuples to write
- delta_window_size: Sliding window size for searching for deltas;
- Set to None for default window size.
- deltify: Whether to deltify objects
- compression_level: the zlib compression level to use
- Returns: Dict mapping id -> (offset, crc32 checksum), pack checksum
- """
- pack_contents_count, pack_contents = pack_objects_to_data(objects, deltify=deltify)
- return write_pack_data(
- write,
- pack_contents,
- num_records=pack_contents_count,
- compression_level=compression_level,
- )
- class PackChunkGenerator:
- """Generator for pack data chunks."""
- def __init__(
- self,
- num_records: int | None = None,
- records: Iterator[UnpackedObject] | None = None,
- progress: Callable[..., None] | None = None,
- compression_level: int = -1,
- reuse_compressed: bool = True,
- ) -> None:
- """Initialize PackChunkGenerator.
- Args:
- num_records: Expected number of records
- records: Iterator of pack records
- progress: Optional progress callback
- compression_level: Compression level (-1 for default)
- reuse_compressed: Whether to reuse compressed chunks
- """
- self.cs = sha1(b"")
- self.entries: dict[bytes, tuple[int, int]] = {}
- if records is None:
- records = iter([]) # Empty iterator if None
- self._it = self._pack_data_chunks(
- records=records,
- num_records=num_records,
- progress=progress,
- compression_level=compression_level,
- reuse_compressed=reuse_compressed,
- )
- def sha1digest(self) -> bytes:
- """Return the SHA1 digest of the pack data."""
- return self.cs.digest()
- def __iter__(self) -> Iterator[bytes]:
- """Iterate over pack data chunks."""
- return self._it
- def _pack_data_chunks(
- self,
- records: Iterator[UnpackedObject],
- *,
- num_records: int | None = None,
- progress: Callable[..., None] | None = None,
- compression_level: int = -1,
- reuse_compressed: bool = True,
- ) -> Iterator[bytes]:
- """Iterate pack data file chunks.
- Args:
- records: Iterator over UnpackedObject
- num_records: Number of records (defaults to len(records) if not specified)
- progress: Function to report progress to
- compression_level: the zlib compression level
- reuse_compressed: Whether to reuse compressed chunks
- Returns: Dict mapping id -> (offset, crc32 checksum), pack checksum
- """
- # Write the pack
- if num_records is None:
- num_records = len(records) # type: ignore
- offset = 0
- for chunk in pack_header_chunks(num_records):
- yield chunk
- self.cs.update(chunk)
- offset += len(chunk)
- actual_num_records = 0
- for i, unpacked in enumerate(records):
- type_num = unpacked.pack_type_num
- if progress is not None and i % 1000 == 0:
- progress((f"writing pack data: {i}/{num_records}\r").encode("ascii"))
- raw: list[bytes] | tuple[int, list[bytes]] | tuple[bytes, list[bytes]]
- if unpacked.delta_base is not None:
- assert isinstance(unpacked.delta_base, bytes), (
- f"Expected bytes, got {type(unpacked.delta_base)}"
- )
- try:
- base_offset, _base_crc32 = self.entries[unpacked.delta_base]
- except KeyError:
- type_num = REF_DELTA
- assert isinstance(unpacked.delta_base, bytes)
- raw = (unpacked.delta_base, unpacked.decomp_chunks)
- else:
- type_num = OFS_DELTA
- raw = (offset - base_offset, unpacked.decomp_chunks)
- else:
- raw = unpacked.decomp_chunks
- chunks: list[bytes] | Iterator[bytes]
- if unpacked.comp_chunks is not None and reuse_compressed:
- chunks = unpacked.comp_chunks
- else:
- chunks = pack_object_chunks(
- type_num, raw, compression_level=compression_level
- )
- crc32 = 0
- object_size = 0
- for chunk in chunks:
- yield chunk
- crc32 = binascii.crc32(chunk, crc32)
- self.cs.update(chunk)
- object_size += len(chunk)
- actual_num_records += 1
- self.entries[unpacked.sha()] = (offset, crc32)
- offset += object_size
- if actual_num_records != num_records:
- raise AssertionError(
- f"actual records written differs: {actual_num_records} != {num_records}"
- )
- yield self.cs.digest()
- def write_pack_data(
- write: Callable[[bytes], None]
- | Callable[[bytes | bytearray | memoryview], int]
- | IO[bytes],
- records: Iterator[UnpackedObject],
- *,
- num_records: int | None = None,
- progress: Callable[..., None] | None = None,
- compression_level: int = -1,
- ) -> tuple[dict[bytes, tuple[int, int]], bytes]:
- """Write a new pack data file.
- Args:
- write: Write function to use
- num_records: Number of records (defaults to len(records) if None)
- records: Iterator over type_num, object_id, delta_base, raw
- progress: Function to report progress to
- compression_level: the zlib compression level
- Returns: Dict mapping id -> (offset, crc32 checksum), pack checksum
- """
- chunk_generator = PackChunkGenerator(
- num_records=num_records,
- records=records,
- progress=progress,
- compression_level=compression_level,
- )
- for chunk in chunk_generator:
- if callable(write):
- write(chunk)
- else:
- write.write(chunk)
- return chunk_generator.entries, chunk_generator.sha1digest()
- def write_pack_index_v1(
- f: IO[bytes],
- entries: Iterable[tuple[bytes, int, int | None]],
- pack_checksum: bytes,
- ) -> bytes:
- """Write a new pack index file.
- Args:
- f: A file-like object to write to
- entries: List of tuples with object name (sha), offset_in_pack,
- and crc32_checksum.
- pack_checksum: Checksum of the pack file.
- Returns: The SHA of the written index file
- """
- f = SHA1Writer(f)
- fan_out_table: dict[int, int] = defaultdict(lambda: 0)
- for name, _offset, _entry_checksum in entries:
- fan_out_table[ord(name[:1])] += 1
- # Fan-out table
- for i in range(0x100):
- f.write(struct.pack(">L", fan_out_table[i]))
- fan_out_table[i + 1] += fan_out_table[i]
- for name, offset, _entry_checksum in entries:
- if not (offset <= 0xFFFFFFFF):
- raise TypeError("pack format 1 only supports offsets < 2Gb")
- f.write(struct.pack(">L20s", offset, name))
- assert len(pack_checksum) == 20
- f.write(pack_checksum)
- return f.write_sha()
- def _delta_encode_size(size: int) -> bytes:
- ret = bytearray()
- c = size & 0x7F
- size >>= 7
- while size:
- ret.append(c | 0x80)
- c = size & 0x7F
- size >>= 7
- ret.append(c)
- return bytes(ret)
- # The length of delta compression copy operations in version 2 packs is limited
- # to 64K. To copy more, we use several copy operations. Version 3 packs allow
- # 24-bit lengths in copy operations, but we always make version 2 packs.
- _MAX_COPY_LEN = 0xFFFF
- def _encode_copy_operation(start: int, length: int) -> bytes:
- scratch = bytearray([0x80])
- for i in range(4):
- if start & 0xFF << i * 8:
- scratch.append((start >> i * 8) & 0xFF)
- scratch[0] |= 1 << i
- for i in range(2):
- if length & 0xFF << i * 8:
- scratch.append((length >> i * 8) & 0xFF)
- scratch[0] |= 1 << (4 + i)
- return bytes(scratch)
- def _create_delta_py(base_buf: bytes, target_buf: bytes) -> Iterator[bytes]:
- """Use python difflib to work out how to transform base_buf to target_buf.
- Args:
- base_buf: Base buffer
- target_buf: Target buffer
- """
- if isinstance(base_buf, list):
- base_buf = b"".join(base_buf)
- if isinstance(target_buf, list):
- target_buf = b"".join(target_buf)
- assert isinstance(base_buf, bytes)
- assert isinstance(target_buf, bytes)
- # write delta header
- yield _delta_encode_size(len(base_buf))
- yield _delta_encode_size(len(target_buf))
- # write out delta opcodes
- seq = SequenceMatcher(isjunk=None, a=base_buf, b=target_buf)
- for opcode, i1, i2, j1, j2 in seq.get_opcodes():
- # Git patch opcodes don't care about deletes!
- # if opcode == 'replace' or opcode == 'delete':
- # pass
- if opcode == "equal":
- # If they are equal, unpacker will use data from base_buf
- # Write out an opcode that says what range to use
- copy_start = i1
- copy_len = i2 - i1
- while copy_len > 0:
- to_copy = min(copy_len, _MAX_COPY_LEN)
- yield _encode_copy_operation(copy_start, to_copy)
- copy_start += to_copy
- copy_len -= to_copy
- if opcode == "replace" or opcode == "insert":
- # If we are replacing a range or adding one, then we just
- # output it to the stream (prefixed by its size)
- s = j2 - j1
- o = j1
- while s > 127:
- yield bytes([127])
- yield bytes(memoryview(target_buf)[o : o + 127])
- s -= 127
- o += 127
- yield bytes([s])
- yield bytes(memoryview(target_buf)[o : o + s])
- # Default to pure Python implementation
- create_delta = _create_delta_py
- def apply_delta(
- src_buf: bytes | list[bytes], delta: bytes | list[bytes]
- ) -> list[bytes]:
- """Based on the similar function in git's patch-delta.c.
- Args:
- src_buf: Source buffer
- delta: Delta instructions
- """
- if not isinstance(src_buf, bytes):
- src_buf = b"".join(src_buf)
- if not isinstance(delta, bytes):
- delta = b"".join(delta)
- out = []
- index = 0
- delta_length = len(delta)
- def get_delta_header_size(delta: bytes, index: int) -> tuple[int, int]:
- size = 0
- i = 0
- while delta:
- cmd = ord(delta[index : index + 1])
- index += 1
- size |= (cmd & ~0x80) << i
- i += 7
- if not cmd & 0x80:
- break
- return size, index
- src_size, index = get_delta_header_size(delta, index)
- dest_size, index = get_delta_header_size(delta, index)
- if src_size != len(src_buf):
- raise ApplyDeltaError(
- f"Unexpected source buffer size: {src_size} vs {len(src_buf)}"
- )
- while index < delta_length:
- cmd = ord(delta[index : index + 1])
- index += 1
- if cmd & 0x80:
- cp_off = 0
- for i in range(4):
- if cmd & (1 << i):
- x = ord(delta[index : index + 1])
- index += 1
- cp_off |= x << (i * 8)
- cp_size = 0
- # Version 3 packs can contain copy sizes larger than 64K.
- for i in range(3):
- if cmd & (1 << (4 + i)):
- x = ord(delta[index : index + 1])
- index += 1
- cp_size |= x << (i * 8)
- if cp_size == 0:
- cp_size = 0x10000
- if (
- cp_off + cp_size < cp_size
- or cp_off + cp_size > src_size
- or cp_size > dest_size
- ):
- break
- out.append(src_buf[cp_off : cp_off + cp_size])
- elif cmd != 0:
- out.append(delta[index : index + cmd])
- index += cmd
- else:
- raise ApplyDeltaError("Invalid opcode 0")
- if index != delta_length:
- raise ApplyDeltaError(f"delta not empty: {delta[index:]!r}")
- if dest_size != chunks_length(out):
- raise ApplyDeltaError("dest size incorrect")
- return out
- def write_pack_index_v2(
- f: IO[bytes],
- entries: Iterable[tuple[bytes, int, int | None]],
- pack_checksum: bytes,
- ) -> bytes:
- """Write a new pack index file.
- Args:
- f: File-like object to write to
- entries: List of tuples with object name (sha), offset_in_pack, and
- crc32_checksum.
- pack_checksum: Checksum of the pack file.
- Returns: The SHA of the index file written
- """
- f = SHA1Writer(f)
- f.write(b"\377tOc") # Magic!
- f.write(struct.pack(">L", 2))
- fan_out_table: dict[int, int] = defaultdict(lambda: 0)
- for name, offset, entry_checksum in entries:
- fan_out_table[ord(name[:1])] += 1
- # Fan-out table
- largetable: list[int] = []
- for i in range(0x100):
- f.write(struct.pack(b">L", fan_out_table[i]))
- fan_out_table[i + 1] += fan_out_table[i]
- for name, offset, entry_checksum in entries:
- f.write(name)
- for name, offset, entry_checksum in entries:
- f.write(struct.pack(b">L", entry_checksum))
- for name, offset, entry_checksum in entries:
- if offset < 2**31:
- f.write(struct.pack(b">L", offset))
- else:
- f.write(struct.pack(b">L", 2**31 + len(largetable)))
- largetable.append(offset)
- for offset in largetable:
- f.write(struct.pack(b">Q", offset))
- assert len(pack_checksum) == 20
- f.write(pack_checksum)
- return f.write_sha()
- def write_pack_index_v3(
- f: IO[bytes],
- entries: Iterable[tuple[bytes, int, int | None]],
- pack_checksum: bytes,
- hash_algorithm: int = 1,
- ) -> bytes:
- """Write a new pack index file in v3 format.
- Args:
- f: File-like object to write to
- entries: List of tuples with object name (sha), offset_in_pack, and
- crc32_checksum.
- pack_checksum: Checksum of the pack file.
- hash_algorithm: Hash algorithm identifier (1 = SHA-1, 2 = SHA-256)
- Returns: The SHA of the index file written
- """
- if hash_algorithm == 1:
- hash_size = 20 # SHA-1
- writer_cls = SHA1Writer
- elif hash_algorithm == 2:
- hash_size = 32 # SHA-256
- # TODO: Add SHA256Writer when SHA-256 support is implemented
- raise NotImplementedError("SHA-256 support not yet implemented")
- else:
- raise ValueError(f"Unknown hash algorithm {hash_algorithm}")
- # Convert entries to list to allow multiple iterations
- entries_list = list(entries)
- # Calculate shortest unambiguous prefix length for object names
- # For now, use full hash size (this could be optimized)
- shortened_oid_len = hash_size
- f = writer_cls(f)
- f.write(b"\377tOc") # Magic!
- f.write(struct.pack(">L", 3)) # Version 3
- f.write(struct.pack(">L", hash_algorithm)) # Hash algorithm
- f.write(struct.pack(">L", shortened_oid_len)) # Shortened OID length
- fan_out_table: dict[int, int] = defaultdict(lambda: 0)
- for name, offset, entry_checksum in entries_list:
- if len(name) != hash_size:
- raise ValueError(
- f"Object name has wrong length: expected {hash_size}, got {len(name)}"
- )
- fan_out_table[ord(name[:1])] += 1
- # Fan-out table
- largetable: list[int] = []
- for i in range(0x100):
- f.write(struct.pack(b">L", fan_out_table[i]))
- fan_out_table[i + 1] += fan_out_table[i]
- # Object names table
- for name, offset, entry_checksum in entries_list:
- f.write(name)
- # CRC32 checksums table
- for name, offset, entry_checksum in entries_list:
- f.write(struct.pack(b">L", entry_checksum))
- # Offset table
- for name, offset, entry_checksum in entries_list:
- if offset < 2**31:
- f.write(struct.pack(b">L", offset))
- else:
- f.write(struct.pack(b">L", 2**31 + len(largetable)))
- largetable.append(offset)
- # Large offset table
- for offset in largetable:
- f.write(struct.pack(b">Q", offset))
- assert len(pack_checksum) == hash_size, (
- f"Pack checksum has wrong length: expected {hash_size}, got {len(pack_checksum)}"
- )
- f.write(pack_checksum)
- return f.write_sha()
- def write_pack_index(
- f: IO[bytes],
- entries: Iterable[tuple[bytes, int, int | None]],
- pack_checksum: bytes,
- progress: Callable[..., None] | None = None,
- version: int | None = None,
- ) -> bytes:
- """Write a pack index file.
- Args:
- f: File-like object to write to.
- entries: List of (checksum, offset, crc32) tuples
- pack_checksum: Checksum of the pack file.
- progress: Progress function (not currently used)
- version: Pack index version to use (1, 2, or 3). If None, defaults to DEFAULT_PACK_INDEX_VERSION.
- Returns:
- SHA of the written index file
- """
- if version is None:
- version = DEFAULT_PACK_INDEX_VERSION
- if version == 1:
- return write_pack_index_v1(f, entries, pack_checksum)
- elif version == 2:
- return write_pack_index_v2(f, entries, pack_checksum)
- elif version == 3:
- return write_pack_index_v3(f, entries, pack_checksum)
- else:
- raise ValueError(f"Unsupported pack index version: {version}")
- class Pack:
- """A Git pack object."""
- _data_load: Callable[[], PackData] | None
- _idx_load: Callable[[], PackIndex] | None
- _data: PackData | None
- _idx: PackIndex | None
- _bitmap: "PackBitmap | None"
- def __init__(
- self,
- basename: str,
- resolve_ext_ref: ResolveExtRefFn | None = None,
- *,
- delta_window_size: int | None = None,
- window_memory: int | None = None,
- delta_cache_size: int | None = None,
- depth: int | None = None,
- threads: int | None = None,
- big_file_threshold: int | None = None,
- ) -> None:
- """Initialize a Pack object.
- Args:
- basename: Base path for pack files (without .pack/.idx extension)
- resolve_ext_ref: Optional function to resolve external references
- delta_window_size: Size of the delta compression window
- window_memory: Memory limit for delta compression window
- delta_cache_size: Size of the delta cache
- depth: Maximum depth for delta chains
- threads: Number of threads to use for operations
- big_file_threshold: Size threshold for big file handling
- """
- self._basename = basename
- self._data = None
- self._idx = None
- self._bitmap = None
- self._idx_path = self._basename + ".idx"
- self._data_path = self._basename + ".pack"
- self._bitmap_path = self._basename + ".bitmap"
- self.delta_window_size = delta_window_size
- self.window_memory = window_memory
- self.delta_cache_size = delta_cache_size
- self.depth = depth
- self.threads = threads
- self.big_file_threshold = big_file_threshold
- self._data_load = lambda: PackData(
- self._data_path,
- delta_window_size=delta_window_size,
- window_memory=window_memory,
- delta_cache_size=delta_cache_size,
- depth=depth,
- threads=threads,
- big_file_threshold=big_file_threshold,
- )
- self._idx_load = lambda: load_pack_index(self._idx_path)
- self.resolve_ext_ref = resolve_ext_ref
- @classmethod
- def from_lazy_objects(
- cls, data_fn: Callable[[], PackData], idx_fn: Callable[[], PackIndex]
- ) -> "Pack":
- """Create a new pack object from callables to load pack data and index objects."""
- ret = cls("")
- ret._data_load = data_fn
- ret._idx_load = idx_fn
- return ret
- @classmethod
- def from_objects(cls, data: PackData, idx: PackIndex) -> "Pack":
- """Create a new pack object from pack data and index objects."""
- ret = cls("")
- ret._data = data
- ret._data_load = None
- ret._idx = idx
- ret._idx_load = None
- ret.check_length_and_checksum()
- return ret
- def name(self) -> bytes:
- """The SHA over the SHAs of the objects in this pack."""
- return self.index.objects_sha1()
- @property
- def data(self) -> PackData:
- """The pack data object being used."""
- if self._data is None:
- assert self._data_load
- self._data = self._data_load()
- self.check_length_and_checksum()
- return self._data
- @property
- def index(self) -> PackIndex:
- """The index being used.
- Note: This may be an in-memory index
- """
- if self._idx is None:
- assert self._idx_load
- self._idx = self._idx_load()
- return self._idx
- @property
- def bitmap(self) -> "PackBitmap" | None:
- """The bitmap being used, if available.
- Returns:
- PackBitmap instance or None if no bitmap exists
- Raises:
- ValueError: If bitmap file is invalid or corrupt
- """
- if self._bitmap is None:
- from .bitmap import read_bitmap
- self._bitmap = read_bitmap(self._bitmap_path, pack_index=self.index)
- return self._bitmap
- def close(self) -> None:
- """Close the pack file and index."""
- if self._data is not None:
- self._data.close()
- if self._idx is not None:
- self._idx.close()
- def __enter__(self) -> "Pack":
- """Enter context manager."""
- return self
- def __exit__(
- self,
- exc_type: type | None,
- exc_val: BaseException | None,
- exc_tb: TracebackType | None,
- ) -> None:
- """Exit context manager."""
- self.close()
- def __eq__(self, other: object) -> bool:
- """Check equality with another pack."""
- if not isinstance(other, Pack):
- return False
- return self.index == other.index
- def __len__(self) -> int:
- """Number of entries in this pack."""
- return len(self.index)
- def __repr__(self) -> str:
- """Return string representation of this pack."""
- return f"{self.__class__.__name__}({self._basename!r})"
- def __iter__(self) -> Iterator[bytes]:
- """Iterate over all the sha1s of the objects in this pack."""
- return iter(self.index)
- def check_length_and_checksum(self) -> None:
- """Sanity check the length and checksum of the pack index and data."""
- assert len(self.index) == len(self.data), (
- f"Length mismatch: {len(self.index)} (index) != {len(self.data)} (data)"
- )
- idx_stored_checksum = self.index.get_pack_checksum()
- data_stored_checksum = self.data.get_stored_checksum()
- if (
- idx_stored_checksum is not None
- and idx_stored_checksum != data_stored_checksum
- ):
- raise ChecksumMismatch(
- sha_to_hex(idx_stored_checksum),
- sha_to_hex(data_stored_checksum),
- )
- def check(self) -> None:
- """Check the integrity of this pack.
- Raises:
- ChecksumMismatch: if a checksum for the index or data is wrong
- """
- self.index.check()
- self.data.check()
- for obj in self.iterobjects():
- obj.check()
- # TODO: object connectivity checks
- def get_stored_checksum(self) -> bytes:
- """Return the stored checksum of the pack data."""
- return self.data.get_stored_checksum()
- def pack_tuples(self) -> list[tuple[ShaFile, None]]:
- """Return pack tuples for all objects in pack."""
- return [(o, None) for o in self.iterobjects()]
- def __contains__(self, sha1: bytes) -> bool:
- """Check whether this pack contains a particular SHA1."""
- try:
- self.index.object_offset(sha1)
- return True
- except KeyError:
- return False
- def get_raw(self, sha1: bytes) -> tuple[int, bytes]:
- """Get raw object data by SHA1."""
- offset = self.index.object_offset(sha1)
- obj_type, obj = self.data.get_object_at(offset)
- type_num, chunks = self.resolve_object(offset, obj_type, obj)
- return type_num, b"".join(chunks) # type: ignore[arg-type]
- def __getitem__(self, sha1: bytes) -> ShaFile:
- """Retrieve the specified SHA1."""
- type, uncomp = self.get_raw(sha1)
- return ShaFile.from_raw_string(type, uncomp, sha=sha1)
- def iterobjects(self) -> Iterator[ShaFile]:
- """Iterate over the objects in this pack."""
- return iter(
- PackInflater.for_pack_data(self.data, resolve_ext_ref=self.resolve_ext_ref)
- )
- def iterobjects_subset(
- self, shas: Iterable[ObjectID], *, allow_missing: bool = False
- ) -> Iterator[ShaFile]:
- """Iterate over a subset of objects in this pack."""
- return (
- uo
- for uo in PackInflater.for_pack_subset(
- self,
- shas,
- allow_missing=allow_missing,
- resolve_ext_ref=self.resolve_ext_ref,
- )
- if uo.id in shas
- )
- def iter_unpacked_subset(
- self,
- shas: Iterable[ObjectID],
- *,
- include_comp: bool = False,
- allow_missing: bool = False,
- convert_ofs_delta: bool = False,
- ) -> Iterator[UnpackedObject]:
- """Iterate over unpacked objects in subset."""
- ofs_pending: dict[int, list[UnpackedObject]] = defaultdict(list)
- ofs: dict[int, bytes] = {}
- todo = set(shas)
- for unpacked in self.iter_unpacked(include_comp=include_comp):
- sha = unpacked.sha()
- if unpacked.offset is not None:
- ofs[unpacked.offset] = sha
- hexsha = sha_to_hex(sha)
- if hexsha in todo:
- if unpacked.pack_type_num == OFS_DELTA:
- assert isinstance(unpacked.delta_base, int)
- assert unpacked.offset is not None
- base_offset = unpacked.offset - unpacked.delta_base
- try:
- unpacked.delta_base = ofs[base_offset]
- except KeyError:
- ofs_pending[base_offset].append(unpacked)
- continue
- else:
- unpacked.pack_type_num = REF_DELTA
- yield unpacked
- todo.remove(hexsha)
- if unpacked.offset is not None:
- for child in ofs_pending.pop(unpacked.offset, []):
- child.pack_type_num = REF_DELTA
- child.delta_base = sha
- yield child
- assert not ofs_pending
- if not allow_missing and todo:
- raise UnresolvedDeltas(list(todo))
- def iter_unpacked(self, include_comp: bool = False) -> Iterator[UnpackedObject]:
- """Iterate over all unpacked objects in this pack."""
- ofs_to_entries = {
- ofs: (sha, crc32) for (sha, ofs, crc32) in self.index.iterentries()
- }
- for unpacked in self.data.iter_unpacked(include_comp=include_comp):
- assert unpacked.offset is not None
- (sha, crc32) = ofs_to_entries[unpacked.offset]
- unpacked._sha = sha
- unpacked.crc32 = crc32
- yield unpacked
- def keep(self, msg: bytes | None = None) -> str:
- """Add a .keep file for the pack, preventing git from garbage collecting it.
- Args:
- msg: A message written inside the .keep file; can be used later
- to determine whether or not a .keep file is obsolete.
- Returns: The path of the .keep file, as a string.
- """
- keepfile_name = f"{self._basename}.keep"
- with GitFile(keepfile_name, "wb") as keepfile:
- if msg:
- keepfile.write(msg)
- keepfile.write(b"\n")
- return keepfile_name
- def get_ref(self, sha: bytes) -> tuple[int | None, int, OldUnpackedObject]:
- """Get the object for a ref SHA, only looking in this pack."""
- # TODO: cache these results
- try:
- offset = self.index.object_offset(sha)
- except KeyError:
- offset = None
- if offset:
- type, obj = self.data.get_object_at(offset)
- elif self.resolve_ext_ref:
- type, obj = self.resolve_ext_ref(sha)
- else:
- raise KeyError(sha)
- return offset, type, obj
- def resolve_object(
- self,
- offset: int,
- type: int,
- obj: OldUnpackedObject,
- get_ref: Callable[[bytes], tuple[int | None, int, OldUnpackedObject]]
- | None = None,
- ) -> tuple[int, OldUnpackedObject]:
- """Resolve an object, possibly resolving deltas when necessary.
- Returns: Tuple with object type and contents.
- """
- # Walk down the delta chain, building a stack of deltas to reach
- # the requested object.
- base_offset = offset
- base_type = type
- base_obj = obj
- delta_stack = []
- while base_type in DELTA_TYPES:
- prev_offset = base_offset
- if get_ref is None:
- get_ref = self.get_ref
- if base_type == OFS_DELTA:
- (delta_offset, delta) = base_obj
- # TODO: clean up asserts and replace with nicer error messages
- assert isinstance(delta_offset, int), (
- f"Expected int, got {delta_offset.__class__}"
- )
- base_offset = base_offset - delta_offset
- base_type, base_obj = self.data.get_object_at(base_offset)
- assert isinstance(base_type, int)
- elif base_type == REF_DELTA:
- (basename, delta) = base_obj
- assert isinstance(basename, bytes) and len(basename) == 20
- base_offset, base_type, base_obj = get_ref(basename) # type: ignore[assignment]
- assert isinstance(base_type, int)
- if base_offset == prev_offset: # object is based on itself
- raise UnresolvedDeltas([basename])
- delta_stack.append((prev_offset, base_type, delta))
- # Now grab the base object (mustn't be a delta) and apply the
- # deltas all the way up the stack.
- chunks = base_obj
- for prev_offset, _delta_type, delta in reversed(delta_stack):
- # Convert chunks to bytes for apply_delta if needed
- if isinstance(chunks, list):
- chunks_bytes = b"".join(chunks)
- elif isinstance(chunks, tuple):
- # For tuple type, second element is the actual data
- _, chunk_data = chunks
- if isinstance(chunk_data, list):
- chunks_bytes = b"".join(chunk_data)
- else:
- chunks_bytes = chunk_data
- else:
- chunks_bytes = chunks
- # Apply delta and get result as list
- chunks = apply_delta(chunks_bytes, delta)
- if prev_offset is not None:
- self.data._offset_cache[prev_offset] = base_type, chunks
- return base_type, chunks
- def entries(
- self, progress: Callable[[int, int], None] | None = None
- ) -> Iterator[PackIndexEntry]:
- """Yield entries summarizing the contents of this pack.
- Args:
- progress: Progress function, called with current and total
- object count.
- Returns: iterator of tuples with (sha, offset, crc32)
- """
- return self.data.iterentries(
- progress=progress, resolve_ext_ref=self.resolve_ext_ref
- )
- def sorted_entries(
- self, progress: ProgressFn | None = None
- ) -> Iterator[PackIndexEntry]:
- """Return entries in this pack, sorted by SHA.
- Args:
- progress: Progress function, called with current and total
- object count
- Returns: Iterator of tuples with (sha, offset, crc32)
- """
- return iter(
- self.data.sorted_entries(
- progress=progress, resolve_ext_ref=self.resolve_ext_ref
- )
- )
- def get_unpacked_object(
- self, sha: bytes, *, include_comp: bool = False, convert_ofs_delta: bool = True
- ) -> UnpackedObject:
- """Get the unpacked object for a sha.
- Args:
- sha: SHA of object to fetch
- include_comp: Whether to include compression data in UnpackedObject
- convert_ofs_delta: Whether to convert offset deltas to ref deltas
- """
- offset = self.index.object_offset(sha)
- unpacked = self.data.get_unpacked_object_at(offset, include_comp=include_comp)
- if unpacked.pack_type_num == OFS_DELTA and convert_ofs_delta:
- assert isinstance(unpacked.delta_base, int)
- unpacked.delta_base = self.index.object_sha1(offset - unpacked.delta_base)
- unpacked.pack_type_num = REF_DELTA
- return unpacked
- def extend_pack(
- f: BinaryIO,
- object_ids: Set[ObjectID],
- get_raw: Callable[[ObjectID], tuple[int, bytes]],
- *,
- compression_level: int = -1,
- progress: Callable[[bytes], None] | None = None,
- ) -> tuple[bytes, list[tuple[bytes, int, int]]]:
- """Extend a pack file with more objects.
- The caller should make sure that object_ids does not contain any objects
- that are already in the pack
- """
- # Update the header with the new number of objects.
- f.seek(0)
- _version, num_objects = read_pack_header(f.read)
- if object_ids:
- f.seek(0)
- write_pack_header(f.write, num_objects + len(object_ids))
- # Must flush before reading (http://bugs.python.org/issue3207)
- f.flush()
- # Rescan the rest of the pack, computing the SHA with the new header.
- new_sha = compute_file_sha(f, end_ofs=-20)
- # Must reposition before writing (http://bugs.python.org/issue3207)
- f.seek(0, os.SEEK_CUR)
- extra_entries = []
- # Complete the pack.
- for i, object_id in enumerate(object_ids):
- if progress is not None:
- progress(
- (f"writing extra base objects: {i}/{len(object_ids)}\r").encode("ascii")
- )
- assert len(object_id) == 20
- type_num, data = get_raw(object_id)
- offset = f.tell()
- crc32 = write_pack_object(
- f.write,
- type_num,
- [data], # Convert bytes to list[bytes]
- sha=new_sha,
- compression_level=compression_level,
- )
- extra_entries.append((object_id, offset, crc32))
- pack_sha = new_sha.digest()
- f.write(pack_sha)
- return pack_sha, extra_entries
- try:
- from dulwich._pack import ( # type: ignore
- apply_delta,
- bisect_find_sha,
- )
- except ImportError:
- pass
- # Try to import the Rust version of create_delta
- try:
- from dulwich._pack import create_delta as _create_delta_rs
- except ImportError:
- pass
- else:
- # Wrap the Rust version to match the Python API (returns bytes instead of Iterator)
- def _create_delta_rs_wrapper(base_buf: bytes, target_buf: bytes) -> Iterator[bytes]:
- """Wrapper for Rust create_delta to match Python API."""
- yield _create_delta_rs(base_buf, target_buf)
- create_delta = _create_delta_rs_wrapper
|