folkslinux
/
rpm
mirror of https://github.com/rpm-software-management/rpm.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
rpm/rpmio/rpmio_internal.h

551 lines
10 KiB
C
Raw Blame History

#ifndef H_RPMIO_INTERNAL
#define H_RPMIO_INTERNAL
/** \ingroup rpmio
* \file rpmio/rpmio_internal.h
*/
#include <assert.h>
#include <rpm/rpmio.h>
#include <rpm/rpmurl.h>
#include <rpm/rpmpgp.h>
#include <rpm/rpmsw.h>
#include "rpmio/digest.h"
/** \ingroup rpmio
*/
typedef struct _FDSTACK_s {
FDIO_t io;
void * fp;
int fdno;
} FDSTACK_t;
/** \ingroup rpmio
* Cumulative statistics for a descriptor.
*/
typedef struct {
struct rpmop_s ops[FDSTAT_MAX]; /*!< Cumulative statistics. */
} * FDSTAT_t;
/** \ingroup rpmio
*/
typedef struct _FDDIGEST_s {
pgpHashAlgo hashalgo;
DIGEST_CTX hashctx;
} * FDDIGEST_t;
/** \ingroup rpmio
* The FD_t File Handle data structure.
*/
struct _FD_s {
int nrefs;
int flags;
#define RPMIO_DEBUG_IO 0x40000000
#define RPMIO_DEBUG_REFS 0x20000000
int magic;
#define FDMAGIC 0x04463138
int nfps;
FDSTACK_t fps[8];
int urlType; /* ufdio: */
int rd_timeoutsecs; /* ufdRead: per FD_t timer */
ssize_t bytesRemain; /* ufdio: */
ssize_t contentLength; /* ufdio: */
int persist; /* ufdio: */
int wr_chunked; /* ufdio: */
int syserrno; /* last system errno encountered */
const void *errcookie; /* gzdio/bzdio/ufdio: */
FDSTAT_t stats; /* I/O statistics */
int ndigests;
#define FDDIGEST_MAX 4
struct _FDDIGEST_s digests[FDDIGEST_MAX];
unsigned int firstFree; /* fadio: */
long int fileSize; /* fadio: */
long int fd_cpioPos; /* cpio: */
};
/** \ingroup rpmio
* \name RPMIO Vectors.
*/
/**
*/
typedef ssize_t (*fdio_read_function_t) (void *cookie, char *buf, size_t nbytes);
/**
*/
typedef ssize_t (*fdio_write_function_t) (void *cookie, const char *buf, size_t nbytes);
/**
*/
typedef int (*fdio_seek_function_t) (void *cookie, _libio_pos_t pos, int whence);
/**
*/
typedef int (*fdio_close_function_t) (void *cookie);
/**
*/
typedef FD_t (*fdio_ref_function_t) ( void * cookie, const char * msg);
/**
*/
typedef FD_t (*fdio_deref_function_t) ( FD_t fd, const char * msg);
/**
*/
typedef FD_t (*fdio_new_function_t) (const char * msg);
/**
*/
typedef int (*fdio_fileno_function_t) (void * cookie);
/**
*/
typedef FD_t (*fdio_open_function_t) (const char * path, int flags, mode_t mode);
/**
*/
typedef FD_t (*fdio_fopen_function_t) (const char * path, const char * fmode);
/**
*/
typedef void * (*fdio_ffileno_function_t) (FD_t fd);
/**
*/
typedef int (*fdio_fflush_function_t) (FD_t fd);
/** \ingroup rpmrpc
* \name RPMRPC Vectors.
*/
/**
*/
typedef int (*fdio_mkdir_function_t) (const char * path, mode_t mode);
/**
*/
typedef int (*fdio_chdir_function_t) (const char * path);
/**
*/
typedef int (*fdio_rmdir_function_t) (const char * path);
/**
*/
typedef int (*fdio_rename_function_t) (const char * oldpath, const char * newpath);
/**
*/
typedef int (*fdio_unlink_function_t) (const char * path);
/**
*/
typedef int (*fdio_stat_function_t) (const char * path, struct stat * st);
/**
*/
typedef int (*fdio_lstat_function_t) (const char * path, struct stat * st);
/**
*/
typedef int (*fdio_access_function_t) (const char * path, int amode);
/** \ingroup rpmio
*/
struct FDIO_s {
fdio_read_function_t read;
fdio_write_function_t write;
fdio_seek_function_t seek;
fdio_close_function_t close;
fdio_ref_function_t _fdref;
fdio_deref_function_t _fdderef;
fdio_new_function_t _fdnew;
fdio_fileno_function_t _fileno;
fdio_open_function_t _open;
fdio_fopen_function_t _fopen;
fdio_ffileno_function_t _ffileno;
fdio_fflush_function_t _fflush;
fdio_mkdir_function_t _mkdir;
fdio_chdir_function_t _chdir;
fdio_rmdir_function_t _rmdir;
fdio_rename_function_t _rename;
fdio_unlink_function_t _unlink;
};
/**
*/
extern FDIO_t fdio;
/**
*/
extern FDIO_t fpio;
/**
*/
extern FDIO_t ufdio;
/**
*/
extern FDIO_t gzdio;
/**
*/
extern FDIO_t bzdio;
/**
*/
extern FDIO_t fadio;
#define FDSANE(fd) assert(fd && fd->magic == FDMAGIC)
extern int _rpmio_debug;
#define DBG(_f, _m, _x) \
\
if ((_rpmio_debug | ((_f) ? ((FD_t)(_f))->flags : 0)) & (_m)) fprintf _x \
#define DBGIO(_f, _x) DBG((_f), RPMIO_DEBUG_IO, _x)
#define DBGREFS(_f, _x) DBG((_f), RPMIO_DEBUG_REFS, _x)
#ifdef __cplusplus
extern "C" {
#endif
/** \ingroup rpmio
*/
int ufdClose( void * cookie);
/** \ingroup rpmio
*/
static inline
FDIO_t fdGetIo(FD_t fd)
{
FDSANE(fd);
return fd->fps[fd->nfps].io;
}
/** \ingroup rpmio
*/
/* FIX: io may be NULL */
static inline
void fdSetIo(FD_t fd, FDIO_t io)
{
FDSANE(fd);
fd->fps[fd->nfps].io = io;
}
/** \ingroup rpmio
*/
static inline
void * fdGetFp(FD_t fd)
{
FDSANE(fd);
return fd->fps[fd->nfps].fp;
}
/** \ingroup rpmio
*/
/* FIX: fp may be NULL */
static inline
void fdSetFp(FD_t fd, void * fp)
{
FDSANE(fd);
fd->fps[fd->nfps].fp = fp;
}
/** \ingroup rpmio
*/
static inline
int fdGetFdno(FD_t fd)
{
FDSANE(fd);
return fd->fps[fd->nfps].fdno;
}
/** \ingroup rpmio
*/
static inline
void fdSetFdno(FD_t fd, int fdno)
{
FDSANE(fd);
fd->fps[fd->nfps].fdno = fdno;
}
/** \ingroup rpmio
*/
static inline
void fdSetContentLength(FD_t fd, ssize_t contentLength)
{
FDSANE(fd);
fd->contentLength = fd->bytesRemain = contentLength;
}
/** \ingroup rpmio
*/
static inline
void fdPush(FD_t fd, FDIO_t io, void * fp, int fdno)
{
FDSANE(fd);
if (fd->nfps >= (sizeof(fd->fps)/sizeof(fd->fps[0]) - 1))
return;
fd->nfps++;
fdSetIo(fd, io);
fdSetFp(fd, fp);
fdSetFdno(fd, fdno);
}
/** \ingroup rpmio
*/
static inline
void fdPop(FD_t fd)
{
FDSANE(fd);
if (fd->nfps < 0) return;
fdSetIo(fd, NULL);
fdSetFp(fd, NULL);
fdSetFdno(fd, -1);
fd->nfps--;
}
/** \ingroup rpmio
*/
static inline
void fdstat_enter(FD_t fd, int opx)
{
if (fd == NULL) return;
if (fd->stats != NULL)
(void) rpmswEnter(fdOp(fd, opx), (ssize_t) 0);
}
/** \ingroup rpmio
*/
static inline
void fdstat_exit(FD_t fd, int opx, ssize_t rc)
{
if (fd == NULL) return;
if (rc == -1)
fd->syserrno = errno;
else if (rc > 0 && fd->bytesRemain > 0)
switch (opx) {
case FDSTAT_READ:
case FDSTAT_WRITE:
fd->bytesRemain -= rc;
break;
default:
break;
}
if (fd->stats != NULL)
(void) rpmswExit(fdOp(fd, opx), rc);
}
/** \ingroup rpmio
*/
static inline
void fdstat_print(FD_t fd, const char * msg, FILE * fp)
{
static int usec_scale = (1000*1000);
int opx;
if (fd == NULL || fd->stats == NULL) return;
for (opx = 0; opx < 4; opx++) {
rpmop op = &fd->stats->ops[opx];
if (op->count <= 0) continue;
switch (opx) {
case FDSTAT_READ:
if (msg) fprintf(fp, "%s:", msg);
fprintf(fp, "%8d reads, %8ld total bytes in %d.%06d secs\n",
op->count, (long)op->bytes,
(int)(op->usecs/usec_scale), (int)(op->usecs%usec_scale));
break;
case FDSTAT_WRITE:
if (msg) fprintf(fp, "%s:", msg);
fprintf(fp, "%8d writes, %8ld total bytes in %d.%06d secs\n",
op->count, (long)op->bytes,
(int)(op->usecs/usec_scale), (int)(op->usecs%usec_scale));
break;
case FDSTAT_SEEK:
break;
case FDSTAT_CLOSE:
break;
}
}
}
/** \ingroup rpmio
*/
static inline
void fdSetSyserrno(FD_t fd, int syserrno, const void * errcookie)
{
FDSANE(fd);
fd->syserrno = syserrno;
fd->errcookie = errcookie;
}
/** \ingroup rpmio
*/
static inline
int fdGetRdTimeoutSecs(FD_t fd)
{
FDSANE(fd);
return fd->rd_timeoutsecs;
}
/** \ingroup rpmio
*/
static inline
long int fdGetCpioPos(FD_t fd)
{
FDSANE(fd);
return fd->fd_cpioPos;
}
/** \ingroup rpmio
*/
static inline
void fdSetCpioPos(FD_t fd, long int cpioPos)
{
FDSANE(fd);
fd->fd_cpioPos = cpioPos;
}
/** \ingroup rpmio
*/
static inline
FD_t c2f(void * cookie)
{
FD_t fd = (FD_t) cookie;
FDSANE(fd);
return fd;
}
/** \ingroup rpmio
* Attach digest to fd.
*/
static inline
void fdInitDigest(FD_t fd, pgpHashAlgo hashalgo, int flags)
{
FDDIGEST_t fddig = fd->digests + fd->ndigests;
if (fddig != (fd->digests + FDDIGEST_MAX)) {
fd->ndigests++;
fddig->hashalgo = hashalgo;
fdstat_enter(fd, FDSTAT_DIGEST);
fddig->hashctx = rpmDigestInit(hashalgo, flags);
fdstat_exit(fd, FDSTAT_DIGEST, (ssize_t) 0);
}
}
/** \ingroup rpmio
* Update digest(s) attached to fd.
*/
static inline
void fdUpdateDigests(FD_t fd, const unsigned char * buf, size_t buflen)
{
int i;
if (buf != NULL && buflen > 0)
for (i = fd->ndigests - 1; i >= 0; i--) {
FDDIGEST_t fddig = fd->digests + i;
if (fddig->hashctx == NULL)
continue;
fdstat_enter(fd, FDSTAT_DIGEST);
(void) rpmDigestUpdate(fddig->hashctx, buf, buflen);
fdstat_exit(fd, FDSTAT_DIGEST, (ssize_t) buflen);
}
}
/** \ingroup rpmio
*/
static inline
void fdFiniDigest(FD_t fd, pgpHashAlgo hashalgo,
void ** datap,
size_t * lenp,
int asAscii)
{
int imax = -1;
int i;
for (i = fd->ndigests - 1; i >= 0; i--) {
FDDIGEST_t fddig = fd->digests + i;
if (fddig->hashctx == NULL)
continue;
if (i > imax) imax = i;
if (fddig->hashalgo != hashalgo)
continue;
fdstat_enter(fd, FDSTAT_DIGEST);
(void) rpmDigestFinal(fddig->hashctx, datap, lenp, asAscii);
fdstat_exit(fd, FDSTAT_DIGEST, (ssize_t) 0);
fddig->hashctx = NULL;
break;
}
if (i < 0) {
if (datap) *datap = NULL;
if (lenp) *lenp = 0;
}
fd->ndigests = imax;
if (i < imax)
fd->ndigests++; /* convert index to count */
}
/* XXX Steal the digest-in-progress from the file handle. */
static inline
void fdStealDigest(FD_t fd, pgpDig dig)
{
int i;
for (i = fd->ndigests - 1; i >= 0; i--) {
FDDIGEST_t fddig = fd->digests + i;
if (fddig->hashctx != NULL)
switch (fddig->hashalgo) {
case PGPHASHALGO_MD5:
assert(dig->md5ctx == NULL);
dig->md5ctx = fddig->hashctx;
fddig->hashctx = NULL;
break;
case PGPHASHALGO_SHA1:
case PGPHASHALGO_SHA256:
case PGPHASHALGO_SHA384:
case PGPHASHALGO_SHA512:
assert(dig->sha1ctx == NULL);
dig->sha1ctx = fddig->hashctx;
fddig->hashctx = NULL;
break;
default:
break;
}
}
}
/**
* Read an entire file into a buffer.
* @param fn file name to read
* @retval *bp (malloc'd) buffer address
* @retval *blenp (malloc'd) buffer length
* @return 0 on success
*/
int rpmioSlurp(const char * fn,
uint8_t ** bp, ssize_t * blenp);
#ifdef __cplusplus
}
#endif
#endif /* H_RPMIO_INTERNAL */
Reference in New Issue View Git Blame Copy Permalink