# This file is part of ssh-python.

# Copyright (C) 2018 Panos Kittenis

#

# This library is free software; you can redistribute it and/or

# modify it under the terms of the GNU Lesser General Public

# License as published by the Free Software Foundation, version 2.1.

#

# This library is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

# Lesser General Public License for more details.

#

# You should have received a copy of the GNU Lesser General Public

# License along with this library; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-130

from libc.stdlib cimport malloc, free

from sftp_attributes cimport SFTPAttributes

from sftp_statvfs cimport SFTPStatVFS

from .exceptions import SFTPError, SFTPHandleError

from c_ssh cimport uint32_t, uint64_t, ssh_get_error, SSH_ERROR, SSH_AGAIN

cimport c_sftp

cdef class SFTPFile:

@staticmethod

cdef SFTPFile from_ptr(c_sftp.sftp_file _file, SFTP sftp):

cdef SFTPFile _fh = SFTPFile.__new__(SFTPFile, sftp)

_fh._file = _file

_fh.sftp = sftp

return _fh

def __enter__(self):

return self

def __exit__(self, *args):

self.close()

def __iter__(self):

return self

def __next__(self):

size, data = self.read()

if size > 0:

return size, data

raise StopIteration

def fstat(self):

cdef SFTPAttributes _attrs

cdef c_sftp.sftp_attributes c_attrs

with nogil:

c_attrs = c_sftp.sftp_fstat(self._file)

if c_attrs is NULL:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

_attrs = SFTPAttributes.from_ptr(c_attrs, self.sftp)

return _attrs

def close(self):

cdef int rc

if self.closed:

return 0

with nogil:

rc = c_sftp.sftp_close(self._file)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

self.closed = True

return rc

def set_nonblocking(self):

with nogil:

c_sftp.sftp_file_set_nonblocking(self._file)

def set_blocking(self):

with nogil:

c_sftp.sftp_file_set_blocking(self._file)

def read(self, size_t size=1048576):

cdef ssize_t _size

cdef bytes buf = b''

cdef char *c_buf

with nogil:

c_buf = <char *>malloc(sizeof(char) * size)

if c_buf is NULL:

with gil:

raise MemoryError

_size = c_sftp.sftp_read(self._file, c_buf, size)

try:

if _size > 0:

buf = c_buf[:_size]

finally:

free(c_buf)

return _size, buf

def async_read_begin(self, uint32_t length=1048576):

cdef int rc

with nogil:

rc = c_sftp.sftp_async_read_begin(self._file, length)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

return rc

def async_read(self, uint32_t _id, uint32_t length=1048576):

cdef int size

cdef bytes buf = b''

cdef char *c_buf

with nogil:

c_buf = <char *>malloc(sizeof(char) * length)

if c_buf is NULL:

with gil:

raise MemoryError

size = c_sftp.sftp_async_read(self._file, c_buf, length, _id)

try:

if size > 0:

buf = c_buf[:size]

elif size < 0:

if size == SSH_ERROR:

raise SFTPError(ssh_get_error(self.sftp.session._session))

elif size == SSH_AGAIN:

return SSH_AGAIN, buf

finally:

free(c_buf)

return size, buf

def write(self, bytes data):

cdef ssize_t rc

cdef const char *c_data = data

cdef size_t data_len = len(data)

with nogil:

rc = c_sftp.sftp_write(self._file, c_data, data_len)

if rc < 0:

raise SFTPError(ssh_get_error(self.sftp.session._session))

return rc

def seek(self, uint32_t offset):

cdef int rc

with nogil:

rc = c_sftp.sftp_seek(self._file, offset)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

return rc

def seek64(self, uint64_t offset):

cdef int rc

with nogil:

rc = c_sftp.sftp_seek64(self._file, offset)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

return rc

def tell(self):

cdef unsigned long rc

with nogil:

rc = c_sftp.sftp_tell(self._file)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

return rc

def tell64(self):

cdef uint64_t rc

with nogil:

rc = c_sftp.sftp_tell64(self._file)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

return rc

def rewind(self):

with nogil:

c_sftp.sftp_rewind(self._file)

def fstatvfs(self):

cdef SFTPStatVFS vfs

cdef c_sftp.sftp_statvfs_t c_vfs

with nogil:

c_vfs = c_sftp.sftp_fstatvfs(self._file)

if c_vfs is NULL:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

vfs = SFTPStatVFS.from_ptr(c_vfs, self.sftp)

return vfs

def fsync(self):

cdef int rc

with nogil:

rc = c_sftp.sftp_fsync(self._file)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

return rc

cdef class SFTPDir:

def __dealloc__(self):

if not self.closed:

self.closedir()

@staticmethod

cdef SFTPDir from_ptr(c_sftp.sftp_dir _dir, SFTP sftp):

cdef SFTPDir _fh = SFTPDir.__new__(SFTPDir, sftp)

_fh._dir = _dir

_fh.sftp = sftp

return _fh

def __enter__(self):

return self

def __exit__(self, *args):

self.closedir()

def __iter__(self):

return self

def __next__(self):

cdef SFTPAttributes _attrs

_attrs = self.readdir()

while _attrs is not None:

return _attrs

raise StopIteration

@property

def sftp_session(self):

return self.sftp

def eof(self):

cdef bint rc

with nogil:

rc = c_sftp.sftp_dir_eof(self._dir)

return bool(rc)

def closedir(self):

cdef int rc

if self.closed:

return 0

with nogil:

rc = c_sftp.sftp_closedir(self._dir)

if rc < 0:

raise SFTPHandleError(ssh_get_error(self.sftp.session._session))

self.closed = True

return rc

cpdef SFTPAttributes readdir(self):

cdef SFTPAttributes _attrs

cdef c_sftp.sftp_attributes c_attrs

with nogil:

c_attrs = c_sftp.sftp_readdir(self.sftp._sftp, self._dir)

if c_sftp.sftp_dir_eof(self._dir) == 1:

return

elif c_attrs is NULL:

raise SFTPHandleError

_attrs = SFTPAttributes.from_ptr(c_attrs, self.sftp)

return _attrs