mirror of
https://github.com/kasmtech/KasmVNC.git
synced 2024-11-08 17:24:13 +01:00
259f1055cb
Provides safety against them accidentally becoming negative because of bugs in the calculations. Also does the same to CharArray and friends as they were strongly connection to the stream objects.
80 lines
2.1 KiB
C++
80 lines
2.1 KiB
C++
/* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
|
|
*
|
|
* This is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This software 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this software; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
|
|
* USA.
|
|
*/
|
|
|
|
//
|
|
// FdInStream streams from a file descriptor.
|
|
//
|
|
|
|
#ifndef __RDR_FDINSTREAM_H__
|
|
#define __RDR_FDINSTREAM_H__
|
|
|
|
#include <rdr/InStream.h>
|
|
|
|
namespace rdr {
|
|
|
|
class FdInStreamBlockCallback {
|
|
public:
|
|
virtual void blockCallback() = 0;
|
|
virtual ~FdInStreamBlockCallback() {}
|
|
};
|
|
|
|
class FdInStream : public InStream {
|
|
|
|
public:
|
|
|
|
FdInStream(int fd, int timeoutms=-1, size_t bufSize=0,
|
|
bool closeWhenDone_=false);
|
|
FdInStream(int fd, FdInStreamBlockCallback* blockCallback,
|
|
size_t bufSize=0);
|
|
virtual ~FdInStream();
|
|
|
|
void setTimeout(int timeoutms);
|
|
void setBlockCallback(FdInStreamBlockCallback* blockCallback);
|
|
int getFd() { return fd; }
|
|
size_t pos();
|
|
void readBytes(void* data, size_t length);
|
|
|
|
void startTiming();
|
|
void stopTiming();
|
|
unsigned int kbitsPerSecond();
|
|
unsigned int timeWaited() { return timeWaitedIn100us; }
|
|
|
|
protected:
|
|
size_t overrun(size_t itemSize, size_t nItems, bool wait);
|
|
|
|
private:
|
|
size_t readWithTimeoutOrCallback(void* buf, size_t len, bool wait=true);
|
|
|
|
int fd;
|
|
bool closeWhenDone;
|
|
int timeoutms;
|
|
FdInStreamBlockCallback* blockCallback;
|
|
|
|
bool timing;
|
|
unsigned int timeWaitedIn100us;
|
|
unsigned int timedKbits;
|
|
|
|
size_t bufSize;
|
|
size_t offset;
|
|
U8* start;
|
|
};
|
|
|
|
} // end of namespace rdr
|
|
|
|
#endif
|