File: //kunden/lib/python3/dist-packages/dulwich/__pycache__/server.cpython-39.pyc
a
�_� � @ s� d Z ddlZddlZddlZddlZddlZddlmZmZm Z m
Z
m Z ddl Z ddl
Z
ddlmZ ddlmZmZmZmZmZmZmZ ddlmZ ddlmZm Z ddl m Z dd l m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZD dd
lEmFZFmGZG dd lHmIZImJZJ e�KeL�ZMG d d
� d
eN�ZOG dd� deN�ZPG dd� deO�ZQG dd� deO�ZRG dd� deN�ZSG dd� deS�ZTG dd� deT�ZUdd� ZVd d � ZWd d � ZXd d!� ZYG d"d#� d#eN�ZZe2e1dfZ[G d$d%� d%eN�Z\G d&d'� d'eN�Z]G d(d)� d)eN�Z^G d*d+� d+eT�Z_G d,d-� d-eS�Z`eUe_e`d.�ZaG d/d0� d0e
jb�ZcG d1d2� d2e
jd�Zeejffd3d4�Zgejfdejhejifd5d6�Zjd7d8� Zkd9d:� Zld;d<� ZmeLd=k�r�eg� dS )>a� Git smart network protocol server implementation.
For more detailed implementation on the network protocol, see the
Documentation/technical directory in the cgit distribution, and in particular:
* Documentation/technical/protocol-capabilities.txt
* Documentation/technical/pack-protocol.txt
Currently supported capabilities:
* include-tag
* thin-pack
* multi_ack_detailed
* multi_ack
* side-band-64k
* ofs-delta
* no-progress
* report-status
* delete-refs
* shallow
* symref
� N)�List�Tuple�Dict�Optional�Iterable)�
tar_stream)�ApplyDeltaError�ChecksumMismatch�GitProtocolError� HookError�NotGitRepository�UnexpectedCommandError�ObjectFormatException)� log_utils)�Commit� valid_hexsha)�write_pack_objects)%�BufferedPktLineWriter�capability_agent�CAPABILITIES_REF�CAPABILITY_AGENT�CAPABILITY_DELETE_REFS�CAPABILITY_INCLUDE_TAG� CAPABILITY_MULTI_ACK_DETAILED�CAPABILITY_MULTI_ACK�CAPABILITY_NO_DONE�CAPABILITY_NO_PROGRESS�CAPABILITY_OFS_DELTA�CAPABILITY_QUIET�CAPABILITY_REPORT_STATUS�CAPABILITY_SHALLOW�CAPABILITY_SIDE_BAND_64K�CAPABILITY_THIN_PACK�COMMAND_DEEPEN� COMMAND_DONE� COMMAND_HAVE�COMMAND_SHALLOW�COMMAND_UNSHALLOW� COMMAND_WANT� MULTI_ACK�MULTI_ACK_DETAILED�Protocol� ProtocolFile�ReceivableProtocol�SIDE_BAND_CHANNEL_DATA�SIDE_BAND_CHANNEL_PROGRESS�SIDE_BAND_CHANNEL_FATAL�
SINGLE_ACK� TCP_GIT_PORT�ZERO_SHA�ack_type�extract_capabilities� extract_want_line_capabilities�symref_capabilities)�ANNOTATED_TAG_SUFFIX�write_info_refs)�BaseRepo�Repoc @ s e Zd ZdZdd� ZdS )�Backendz2A backend for the Git smart server implementation.c C s t | j��dS )z�Open the repository at a path.
Args:
path: Path to the repository
Raises:
NotGitRepository: no git repository was found at path
Returns: Instance of BackendRepo
N)�NotImplementedError�open_repository��self�path� rB �0/usr/lib/python3/dist-packages/dulwich/server.pyr> � s zBackend.open_repositoryN)�__name__�
__module__� __qualname__�__doc__r> rB rB rB rC r< } s r< c @ sL e Zd ZdZdZdZeeef d�dd�Zee e d�dd�Z
d d d
�Z dS ) � BackendRepoz�Repository abstraction used by the Git server.
The methods required here are a subset of those provided by
dulwich.repo.Repo.
N��returnc C s t �dS )zZ
Get all the refs in the repository
Returns: dict of name -> sha
N�r= �r@ rB rB rC �get_refs� s zBackendRepo.get_refs)�namerJ c C s dS )a� Return the cached peeled value of a ref, if available.
Args:
name: Name of the ref to peel
Returns: The peeled value of the ref. If the ref is known not point to
a tag, this will be the SHA the ref refers to. If no cached
information about a tag is available, this method may return None,
but it should attempt to peel the tag if possible.
NrB )r@ rN rB rB rC �
get_peeled� s
zBackendRepo.get_peeledc C s t �dS )a
Yield the objects required for a list of commits.
Args:
progress: is a callback to send progress messages to the client
get_tagged: Function that returns a dict of pointed-to sha ->
tag sha for including tags.
NrK )r@ �determine_wants� graph_walker�progress�
get_taggedrB rB rC �
fetch_objects� s
zBackendRepo.fetch_objects)N) rD rE rF rG � object_store�refsr �bytesrM r rO rT rB rB rB rC rH � s
�rH c @ s( e Zd ZdZdd� Zeed�dd�ZdS )� DictBackendz?Trivial backend that looks up Git repositories in a dictionary.c C s
|| _ d S �N)�repos)r@ rZ rB rB rC �__init__� s zDictBackend.__init__)rA rJ c C sB t �d|� z | j| W S ty< tdt|d� ��Y n0 d S )NzOpening repository at %sz'No git repository was found at %(path)s)rA )�logger�debugrZ �KeyErrorr �dictr? rB rB rC r> � s �zDictBackend.open_repositoryN)rD rE rF rG r[ �strr: r> rB rB rB rC rX � s rX c s. e Zd ZdZejf� fdd� Zdd� Z� ZS )�FileSystemBackendzDSimple backend looking up Git repositories in the local file system.c s6 t t| ��� tj�|�tj �tjd tj�| _d S )N� ) �superra r[ �osrA �abspath�sep�replace�root)r@ rh �� __class__rB rC r[ � s �zFileSystemBackend.__init__c C sh t �d|� tj�tj�| j|��tj }tj�|�}tj�| j�}|� |�s`t
d|| jf ��t |�S )Nzopening repository at %szPath %r not inside root %r) r\ r] rd rA re �joinrh rf �normcase�
startswithr r; )r@ rA re Znormcase_abspathZ
normcase_rootrB rB rC r> � s
��z!FileSystemBackend.open_repository) rD rE rF rG rd rf r[ r> �
__classcell__rB rB ri rC ra � s ra c @ s" e Zd ZdZddd�Zdd� ZdS )�Handlerz*Smart protocol command handler base class.Nc C s || _ || _|| _d S rY )�backend�proto�
stateless_rpc�r@ rp rq rr rB rB rC r[ � s zHandler.__init__c C s t | j��d S rY )r= �handlerL rB rB rC rt � s zHandler.handle)N)rD rE rF rG r[ rt rB rB rB rC ro � s
ro c s� e Zd ZdZd� fdd� Zedd� �Zeee d�dd ��Z eee d�d
d ��Z
eee d�d d
��Z ee dd�dd�Z ee
d�dd�Zdd�dd�Z� ZS )� PackHandlerzProtocol handler for packs.Nc s$ t t| ��|||� d | _d| _d S �NF)rc ru r[ �_client_capabilities�_done_receivedrs ri rB rC r[ � s zPackHandler.__init__c C s t �d|� d�dd� |D ��S )NzSending capabilities: %s� c S s g | ] }d | �qS )� rB )�.0�crB rB rC �
<listcomp>� ry z/PackHandler.capability_line.<locals>.<listcomp>)r\ �infork )�cls� capabilitiesrB rB rC �capability_line� s zPackHandler.capability_linerI c C s t | j��d S rY )r= r� �r rB rB rC r� � s zPackHandler.capabilitiesc C s t tttt� gS rY )r r" r r r r� rB rB rC �innocuous_capabilities� s �z"PackHandler.innocuous_capabilitiesc C s g S )zAReturn a list of capabilities that we require the client to have.rB r� rB rB rC �required_capabilities� s z!PackHandler.required_capabilities)�capsrJ c C s� t | �� �}|�| �� � |D ](}|�td �r2q ||vr td| ��q | �� D ]}||vrPtd| ��qPt |�| _t �
d|� d S )N� =z7Client asked for capability %r that was not advertised.z/Client does not support required capability %r.zClient capabilities: %s) �setr� �updater� rm r r
r� rw r\ r~ )r@ r� Zallowable_caps�caprB rB rC �set_client_capabilities s � �
z#PackHandler.set_client_capabilities)r� rJ c C s | j d u rtd| ��|| j v S )Nz=Server attempted to access capability %r before asking client)rw r
)r@ r� rB rB rC �has_capability s
�zPackHandler.has_capabilityc C s
d| _ d S )NT)rx rL rB rB rC � notify_done s zPackHandler.notify_done)N)rD rE rF rG r[ � classmethodr� r rW r� r� r� r� �boolr� r� rn rB rB ri rC ru � s
ru c sT e Zd ZdZd� fdd� Zedd� �Zedd � �Zd
d � Zdd d
�Z dd� Z
� Z S )�UploadPackHandlerz4Protocol handler for uploading a pack to the client.NFc s< t t| �j|||d� |�|d �| _d | _|| _d| _d S )N�rr r F)rc r� r[ r> �repoZ
_graph_walker�advertise_refs�_processing_have_lines�r@ rp �argsrq rr r� ri rB rC r[ ! s
�zUploadPackHandler.__init__c C s t ttttttttg S rY ) r r r! r" r r r r r r� rB rB rC r� - s
�z UploadPackHandler.capabilitiesc C s
t ttfS rY )r! r" r r� rB rB rC r� 4 s �z'UploadPackHandler.required_capabilitiesc C s& | � t�s| jrd S | j�t|� d S rY )r� r r� rq �write_sidebandr/ )r@ �messagerB rB rC rR 9 s
�zUploadPackHandler.progressc C sv | � t�si S |du r | j�� }|du rBt| jdd�}|du rBi S i }|�� D ]"\}}|�|�}||krN|||< qN|S )a� Get a dict of peeled values of tags to their original tag shas.
Args:
refs: dict of refname -> sha of possible tags; defaults to all
of the backend's refs.
repo: optional Repo instance for getting peeled refs; defaults
to the backend's repo, if available
Returns: dict of peeled_sha -> tag_sha, where tag_sha is the sha of a
tag whose peeled value is peeled_sha.
Nr� )r� r r� rM �getattr�itemsrO )r@ rV r� ZtaggedrN �sha�
peeled_sharB rB rC rS ? s
z UploadPackHandler.get_taggedc s� �fdd�}t ��jj�jj�jjj�� g �� �fdd�}�jj|� �j�jd�}d�_ t
��dkrhd S d�_ � � �� t
� �j�s�d S ��d t
|� �d
�� ttd |�|� �j�d � d S ) Nc s � j �t| �S rY �rq r� r. ��xrL rB rC �write` s z'UploadPackHandler.handle.<locals>.writec s �� � �| �� �S rY )�extendrP )rV )rQ �wantsrB rC �
wants_wrapperh s z/UploadPackHandler.handle.<locals>.wants_wrapper)rS Tr Fz counting objects: %d, done.
�ascii)�_ProtocolGraphWalkerr� rU rO rV � get_symrefsrT rR rS r� �len� handle_doner� r rx �encoder r, rq �write_pkt_line)r@ r� r� Z objects_iterrB )rQ r@ r� rC rt _ s6 ��
� ��zUploadPackHandler.handle)NF)NN) rD rE rF rG r[ r� r� r� rR rS rt rn rB rB ri rC r� s �
r� c C s� | s dg}n| � d��dd�}|d }|dur>||vr>t|��t|�dkr^|tdfv r^|dfS t|�dkr�|ttttfv r�t |d �s�t
d��t |�S |t kr�|t
|d �fS t
d| ��dS ) ac Split a line read from the wire.
Args:
line: The line read from the wire.
allowed: An iterable of command names that should be allowed.
Command names not listed below as possible return values will be
ignored. If None, any commands from the possible return values are
allowed.
Returns: a tuple having one of the following forms:
('want', obj_id)
('have', obj_id)
('done', None)
(None, None) (for a flush-pkt)
Raises:
UnexpectedCommandError: if the line cannot be parsed into one of the
allowed return values.
N�
rz � r rb z Invalid shaz%Received invalid line from client: %r)�rstrip�splitr
r� r$ r( r% r& r' r r
�tupler# �int)�line�allowed�fields�commandrB rB rC �_split_proto_line� s$ � r� c s� i ���fdd�}g }|D ](}�� |�}t|t�r|�|jdf� qt� }t� }|r�|�� \} }
|
|k r�|�| � |
d � |�� fdd�|| �D �� qP|�| � qP||fS )a Find shallow commits according to a given depth.
Args:
store: An ObjectStore for looking up objects.
heads: Iterable of head SHAs to start walking from.
depth: The depth of ancestors to include. A depth of one includes
only the heads themselves.
Returns: A tuple of (shallow, not_shallow), sets of SHAs that should be
considered shallow and unshallow according to the arguments. Note that
these sets may overlap if a commit is reachable along multiple paths.
c s& � � | d �}|s"�| j}|� | < |S rY )�get�parents)r� �result)r� �storerB rC � get_parents� s
z"_find_shallow.<locals>.get_parentsr� c 3 s | ]}|� fV qd S rY rB )r{ �p)� new_depthrB rC � <genexpr>� ry z _find_shallow.<locals>.<genexpr>) Zpeel_sha�
isinstancer �append�idr� �pop�addr� ) r� �heads�depthr� ZtodoZhead_sha�obj� not_shallow�shallowr� Z cur_depthrB )r� r� r� rC �
_find_shallow� s"
r� c
C s� | | }t �|g�}t|g�}|r�|�� }|j|v r8dS |jdkrDq |jD ]4}||v rXqJ|�|� | | } | j|krJ|� | � qJq dS )NTs commitF)
� collections�dequer� �popleftr� Z type_namer� r� � commit_timer� )
r� �haves�want�earliest�o�pendingZknownZcommit�parentZ
parent_objrB rB rC �_want_satisfied� s"
r� c sJ t |�}|r$t� fdd�|D ��}nd}|D ]}t� |||�s, dS q,dS )a~ Check whether all the current wants are satisfied by a set of haves.
Args:
store: Object store to retrieve objects from
haves: A set of commits we know the client has.
wants: A set of commits the client wants
Note: Wants are specified with set_wants rather than passed in since
in the current interface they are determined outside this class.
c s g | ]}� | j �qS rB )r� )r{ �h�r� rB rC r} ry z(_all_wants_satisfied.<locals>.<listcomp>r FT)r� �minr� )r� r� r� r� r� rB r� rC �_all_wants_satisfied� s
r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e Z
dd� Z dd� Z dd� Z
d"dd�Zdd� Zdd� Zdd � Zd d � Zd d � Zd!S )#r� a7 A graph walker that knows the git protocol.
As a graph walker, this class implements ack(), next(), and reset(). It
also contains some base methods for interacting with the wire and walking
the commit tree.
The work of determining which acks to send is passed on to the
implementation instance stored in _impl. The reason for this is that we do
not know at object creation time what ack level the protocol requires. A
call to set_ack_type() is required to set up the implementation, before
any calls to next() or ack() are made.
c C sj || _ || _|| _|| _|j| _|j| _|j| _g | _t� | _ t� | _
t� | _ d| _ g | _
d| _d | _d S )NFr )�handlerr� rO r� rq rr r� �_wantsr� r� �client_shallow� unshallow�_cached�_cache� _cache_index�_impl)r@ r� rU rO r� rB rB rC r[ s z _ProtocolGraphWalker.__init__c C s� | � � }t|�� �}| js | js�tt|�� ��D ]�\}\}}z| �|�}W n t y` Y q0Y n0 |d | }|s�|d| j
� | j
� � t
|�� � � 7 }| j�|d � ||kr0| j�|d | t d � q0| j�d� | jr�g S | j�� } | s�g S t| �\}}
| j
�|
� | �t|
�� tttdf} t|| �\} }g }
| tk�rv||v�rZtd| ��|
�|� | � | �\} }�q:| � |
� | ttfv �r�| � | |� | � |
� | j�r�| j� � �r�g S |
S )a- Determine the wants for a set of heads.
The given heads are advertised to the client, who then specifies which
refs they want using 'want' lines. This portion of the protocol is the
same regardless of ack type, and in fact is used to set the ack type of
the ProtocolGraphWalker.
If the client has the 'shallow' capability, this method also reads and
responds to the 'shallow' and 'deepen' lines from the client. These are
not part of the wants per se, but they set up necessary state for
walking the graph. Additionally, later code depends on this method
consuming everything up to the first 'have' line.
Args:
heads: a dict of refname->SHA1 to advertise
Returns: a list of SHA1s requested by the client
rz � r� Nz Client wants invalid object %s)!r� r� �valuesr� rr � enumerate�sortedr� rO r^ r� r� r� r7 rq r� r8 �
read_pkt_liner6 r� � set_ack_typer4 r( r& r# r� r
r� �read_proto_line� set_wants�unread_proto_line�_handle_shallow_request�eof)r@ r� �symrefsr� �i�refr� r� r� r� r� r� r� Z want_revsrB rB rC rP ) s`
����
�
z$_ProtocolGraphWalker.determine_wantsc C s0 t |t�rt|��d�}| j�|d | � d S )Nr� rz )r� r� r` r� rq Zunread_pkt_line)r@ r� �valuerB rB rC r� w s
z&_ProtocolGraphWalker.unread_proto_linec C s$ t |�dkrtd| ��| j�|�S )N�( zinvalid sha %r)r� �
ValueErrorr� �ack�r@ Zhave_refrB rB rC r� | s z_ProtocolGraphWalker.ackc C s d| _ d| _d S )NTr )r� r� rL rB rB rC �reset� s z_ProtocolGraphWalker.resetc C sN | j s | js| jrd S t| j�S | jd7 _| jt| j�krBd S | j| j S )Nr� )r� r� rr �nextr� r� r� rL rB rB rC r� � s
z_ProtocolGraphWalker.nextc C s t | j�� |�S )a Read a line from the wire.
Args:
allowed: An iterable of command names that should be allowed.
Returns: A tuple of (command, value); see _split_proto_line.
Raises:
UnexpectedCommandError: If an error occurred reading the line.
)r� rq r� )r@ r� rB rB rC r� � s z$_ProtocolGraphWalker.read_proto_linec
C s� | � ttf�\}}|tkr |}q.| j�|� q | � d� t| j||�\}}| j�|| � | j| j }|| j@ }| _ t
|�D ]} | j � td | � q~t
|�D ]} | j � t
d | � q�| j � d � d S )NrY rz )r� r# r& r� r� r� r� r� r� r� r� rq r� r' )
r@ r� r� �valr� r� r� Z new_shallowr� r� rB rB rC r� � s" �
z,_ProtocolGraphWalker._handle_shallow_requestc C s | j �� d S rY )r� r� rL rB rB rC r� � s z _ProtocolGraphWalker.notify_donery c C s( |r d| }| j �d| | d � d S )Nrz s ACK r� �rq r� )r@ r� r4 rB rB rC �send_ack� s z _ProtocolGraphWalker.send_ackc C s | j �d� d S )Ns NAK
r� rL rB rB rC �send_nak� s z _ProtocolGraphWalker.send_nakc C s | j �||�S rY )r� r� �r@ Z
done_requiredZ
done_receivedrB rB rC r� � s z _ProtocolGraphWalker.handle_donec C s
|| _ d S rY )r� )r@ r� rB rB rC r� � s z _ProtocolGraphWalker.set_wantsc C s t | j|| j�S )a0 Check whether all the current wants are satisfied by a set of haves.
Args:
haves: A set of commits we know the client has.
Note: Wants are specified with set_wants rather than passed in since
in the current interface they are determined outside this class.
)r� r� r� )r@ r� rB rB rC �all_wants_satisfied� s z(_ProtocolGraphWalker.all_wants_satisfiedc C s" t ttttti}|| | �| _d S rY )r) �MultiAckGraphWalkerImplr* � MultiAckDetailedGraphWalkerImplr1 �SingleAckGraphWalkerImplr� )r@ r4 Z impl_classesrB rB rC r� � s
�z!_ProtocolGraphWalker.set_ack_typeN)ry )rD rE rF rG r[ rP r� r� r� r� �__next__r� r� r� r� r� r� r� r� r� rB rB rB rC r� s" N
r� c @ s4 e Zd ZdZdd� Zdd� Zdd� ZeZdd � Zd
S ) r� z@Graph walker implementation that speaks the single-ack protocol.c C s || _ g | _d S rY ��walker�_common�r@ r� rB rB rC r[ � s z!SingleAckGraphWalkerImpl.__init__c C s" | j s | j�|� | j �|� d S rY )r r� r� r� r� rB rB rC r� � s z SingleAckGraphWalkerImpl.ackc C s: | j �t�\}}|d tfv r*| j �� d S |tkr6|S d S rY )r� r� �_GRAPH_WALKER_COMMANDSr$ r� r% �r@ r� r� rB rB rC r� � s
z SingleAckGraphWalkerImpl.nextc C s. | j s| j�� |r |s dS |s*| j s*dS dS )NFT)r r� r� r� rB rB rC r� � s
z$SingleAckGraphWalkerImpl.handle_doneN� rD rE rF rG r[ r� r� r� r� rB rB rB rC r� � s r� c @ s4 e Zd ZdZdd� Zdd� Zdd� ZeZdd � Zd
S ) r� z?Graph walker implementation that speaks the multi-ack protocol.c C s || _ d| _g | _d S rv )r� � _found_baser r rB rB rC r[
s z MultiAckGraphWalkerImpl.__init__c C s8 | j �|� | js4| j�|d� | j�| j �r4d| _d S )N� continueT)r r� r r� r� r� r� rB rB rC r� s
zMultiAckGraphWalkerImpl.ackc C sb | j �t�\}}|d u r&| j �� q q |tkr<| j �� d S |tkr | jrX| j �|d� |S q d S )Nr ) r� r� r r� r$ r� r% r r� r rB rB rC r� s
z MultiAckGraphWalkerImpl.nextc C sB |r |s dS |s| j sdS | j r4| j�| j d � n
| j�� dS �NF���T�r r� r� r� r� rB rB rC r� - s
z#MultiAckGraphWalkerImpl.handle_doneNr rB rB rB rC r�
s r� c @ s4 e Zd ZdZdd� Zdd� Zdd� ZeZdd � Zd
S ) r� zEGraph walker implementation speaking the multi-ack-detailed protocol.c C s || _ g | _d S rY r� r rB rB rC r[ H s z(MultiAckDetailedGraphWalkerImpl.__init__c C s | j �|� | j�|d� d S )Ns common)r r� r� r� r� rB rB rC r� L s z#MultiAckDetailedGraphWalkerImpl.ackc C sz | j �t�\}}|d u rR| j �| j�r:| j �| jd d� | j �� | j jrtd S q |tkrh| j � � qvq |t
kr |S q d S )Nr s ready) r� r� r r� r r� r� rr r$ r� r% r rB rB rC r� Q s
z$MultiAckDetailedGraphWalkerImpl.nextc C sB |r |s dS |s| j sdS | j r4| j�| j d � n
| j�� dS r r r� rB rB rC r� n s
z+MultiAckDetailedGraphWalkerImpl.handle_doneNr rB rB rB rC r� E s r� c s� e Zd ZdZd� fdd� Zeee d�dd��Ze e
eeef e e
eef d �d
d �Z e e
eef dd �d
d�Z dd� Z
dd�dd�Z� ZS )�ReceivePackHandlerz8Protocol handler for downloading a pack from the client.NFc s0 t t| �j|||d� |�|d �| _|| _d S )Nr� r )rc r
r[ r> r� r� r� ri rB rC r[ � s
�zReceivePackHandler.__init__rI c C s t tttttgS rY )r r r r r! r r� rB rB rC r� � s �z ReceivePackHandler.capabilities)rV rJ c
C s� t tttttjtjtf}g }d}|D ]}|d t kr$d}q$|r�z0t
| j dd �}| j j
�| j j|� |�d� W q� |y� } z,|�dt|��dd��d �f� W Y d }~q�d }~0 0 n
|�d� |D ]�\}} }
d
} z�| t k�r"t| �� vr�td ��z| j j�|
|� W n |�y d } Y n0 n0z| j j�|
|| � W n |�yP d
} Y n0 W n t�yl d} Y n0 |�|
| f� q�|S )NFr� T�recv)� unpack� okr �
� zutf-8r
z8Attempted to delete refs without delete-refs capability.s failed to deletes failed to writes bad ref)�IOError�OSErrorr r �AssertionError�socket�error�zlibr r3 r� rq r� rU Z
add_thin_pack�readr� r` rg r� r r� r
rV Zremove_if_equalsZ
set_if_equalsr^ ) r@ rV Zall_exceptions�statusZwill_send_packr� r �eZoldshar� r� Z
ref_statusrB rB rC � _apply_pack� sN
� �
�
z ReceivePackHandler._apply_pack)r rJ c s� � � t�r0t� fdd����j}� �fdd�}n� jj}dd� }|D ]T\}}|dkrf|d| d � qD|d kr�|d
| d � qD|d | d | d � qD|d � |� d S )
Nc s � j �t| �S rY r� )�drL rB rC �<lambda>� ry z3ReceivePackHandler._report_status.<locals>.<lambda>c s �� � � j�d � d S rY )�flushrq r� rB �r@ �writerrB rC r � s z0ReceivePackHandler._report_status.<locals>.flushc S s d S rY rB rB rB rB rC r � s r s unpack r� r
s ok s ng rz )r� r! r r� rq r� )r@ r r� r rN �msgrB r rC �_report_status� s
� z!ReceivePackHandler._report_statusc
C st | j j�dd �}|sd S z |�|�}|r6| j�t|� W n6 tyn } z | j�tt |�� W Y d }~n
d }~0 0 d S )Nz post-receive)
r� Zhooksr� Zexecuterq r� r/ r r0 �repr)r@ � client_refs�hook�output�errrB rB rC �_on_post_receive� s
z#ReceivePackHandler._on_post_receivec C sT | j s | js�t| j�� �� �}t| jj�� �� �}|s@tt fg}| j
� |d d d |d d d | � | �
� t|� � d � tdt|��D ],}|| }| j
� |d d |d d � q�| j
� d � | j r�d S g }| j
�� }|d u r�d S t|�\}}| �|� |�r&|�|�� � | j
�� }�q| �|�}| �|� | �t��rP| �|� d S )Nr r� rz r� r� )r� rr r� r� rM r� rV r� r r3 rq r� r� r� r7 �ranger� r� r5 r� r� r� r r& r� r r )r@ rV r� r� r� r" r� r rB rB rC rt � sB
����"
zReceivePackHandler.handle)NF)rD rE rF rG r[ r� r rW r� r r r r r& rt rn rB rB ri rC r
� s �� 5 r
c s& e Zd Zd� fdd� Zdd� Z� ZS )�UploadArchiveHandlerNc s( t t| ��|||� |�|d �| _d S )Nr )rc r( r[ r> r� )r@ rp r� rq rr ri rB rC r[ s
�z UploadArchiveHandler.__init__c s( � fdd�}g }� j �� D ]8}|�dd�\}}|dkrBtd| ��|�|�d�� qd}d }d
}� jj} |t|�k r�|| }
|
d kr�|d7 }|| }n:|
d kr�|d7 }|| � d
�}n� jj
|
} | | | j } |d7 }qh� j � d� � j � d � t
| | t�� ||d�D ]}
||
� �q� j � d � d S )Nc s � j �t| �S rY r� r� rL rB rC r� $ s z*UploadArchiveHandler.handle.<locals>.writerz r� s argumentzunknown command %sr� ry Ztarr s --prefixs --formatr� s ACK)�mtime�prefix�format)rq Z read_pkt_seqr� r
r� r� r� rU r� �decoderV �treer� r �time)r@ r� Z argumentsZpkt�keyr� r* r+ r� r� ZargumentZ
commit_shar- �chunkrB rL rC rt # s:
�
zUploadArchiveHandler.handle)N)rD rE rF r[ rt rn rB rB ri rC r( s r( )s git-upload-packs git-receive-packs git-upload-archivec @ s e Zd Zdd� Zdd� ZdS )�TCPGitRequestHandlerc O s$ || _ tjj| g|�R i |�� d S rY )�handlers� socketserver�StreamRequestHandlerr[ )r@ r2 r� �kwargsrB rB rC r[ N s z TCPGitRequestHandler.__init__c C sj t | jj| jj�}|�� \}}t�d||� | j� |d �}t
|�sNt d| ��|| j j
||�}|�� d S )Nz Handling %s request, args=%szInvalid service %s)r- Z
connectionr Zwfiler� Zread_cmdr\ r~ r2 r� �callabler
�serverrp rt )r@ rq r� r� r r� rB rB rC rt R s zTCPGitRequestHandler.handleN)rD rE rF r[ rt rB rB rB rC r1 L s r1 c @ s> e Zd ZdZejjZdd� Ze dfdd�Z
dd� Z d d
� Z dS ) � TCPGitServerTc O s t | jg|�R i |��S rY )r1 r2 )r@ r� r5 rB rB rC �
_make_handlerc s zTCPGitServer._make_handlerNc C sL t t�| _|d ur | j�|� || _t�d||� tj� | ||f| j
� d S )Nz&Listening for TCP connections on %s:%d) r_ �DEFAULT_HANDLERSr2 r� rp r\ r~ r3 � TCPServerr[ r9 )r@ rp Z listen_addr�portr2 rB rB rC r[ f s
��zTCPGitServer.__init__c C s t �d|� dS )NzHandling request from %sT)r\ r~ �r@ ZrequestZclient_addressrB rB rC �verify_requestp s zTCPGitServer.verify_requestc C s t �d|� d S )Nz7Exception happened during processing of request from %s)r\ Z exceptionr= rB rB rC � handle_errort s �zTCPGitServer.handle_error)
rD rE rF Zallow_reuse_addressr3 r; �
serve_foreverZserver9 r2 r[ r> r? rB rB rB rC r8 ^ s
r8 c C s� ddl }|�� }|jdddddd� |jd d
d ttd d
� |�| �\}}t�� t|�dkrf|d }nd}t |�}t
||j |j �}|�
� dS )z*Entry point for starting a TCP git server.r Nz-lz--listen_address�listen_addressZ localhostzBinding IP address.)�dest�default�helpz-pz--portr< zBinding TCP port.)rB �typerC rD r� �.)�optparseZ OptionParserZ
add_optionr� r2 �
parse_argsr Zdefault_logging_configr� ra r8 rA r< r@ )�argvrG �parserZoptionsr� Zgitdirrp r7 rB rB rC �mainy s$
� �
rK c sF |du rt � }� fdd�}t|j|�}| ||dd� |�}|�� dS )a� Serve a single command.
This is mostly useful for the implementation of commands used by e.g.
git+ssh.
Args:
handler_cls: `Handler` class to use for the request
argv: execv-style command-line arguments. Defaults to sys.argv.
backend: `Backend` to use
inf: File-like object to read from, defaults to standard input.
outf: File-like object to write to, defaults to standard output.
Returns: Exit code for use with sys.exit. 0 on success, 1 on failure.
Nc s � � | � � �� d S rY )r� r )�data��outfrB rC �send_fn� s
z serve_command.<locals>.send_fnr� r )ra r+ r rt )Z handler_clsrI rp �infrN rO rq r� rB rM rC �
serve_command� s rQ c C s | � � }t|| j�S )zGenerate an info refs file.)rM r9 rU )r� rV rB rB rC �generate_info_refs� s rR c c s* | j jD ] }dt�|jj� d V qdS )z Generate an index for for packs.s P r� N)rU �packsrd �fsencoderL �filename)r� ZpackrB rB rC �generate_objects_info_packs� s
��rV c C sF | � tj�dd�d�t| ��� | � tj�ddd�d�t| ��� dS )z�Generate server info for dumb file access.
This generates info/refs and objects/info/packs,
similar to "git update-server-info".
r~ rV ry ZobjectsrS N)Z_put_named_filerd rA rk rR rV )r� rB rB rC �update_server_info� s � �rW �__main__)nrG r� rd r �sysr. �typingr r r r r r r3 Zdulwich.archiver Zdulwich.errorsr r r
r r r
r Zdulwichr Zdulwich.objectsr r Z dulwich.packr Zdulwich.protocolr r r r r r r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 Z dulwich.refsr8 r9 Z dulwich.repor: r; Z getLoggerrD r\ �objectr< rH rX ra ro ru r� r� r� r� r� r� r r� r� r� r
r( r: r4 r1 r; r8 rI rK �stdin�stdoutrQ rR rV rW rB rB rB rC �<module> sh $ �'
+ 7q') Q
,;A *�
�