54 lines
1.1 KiB
C
54 lines
1.1 KiB
C
|
#ifndef BTSTREAM_H
|
|||
|
|
#define BTSTREAM_H
|
|||
|
|
|
|||
|
|
#include "./def.h"
|
|||
|
|
#include "./bufio.h"
|
|||
|
|
|
|||
|
|
#ifdef WINDOWS
|
|||
|
|
#include "Winsock2.h"
|
|||
|
|
#else
|
|||
|
|
|
|||
|
|
#include "unistd.h"
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
class btStream
|
|||
|
|
{
|
|||
|
|
private:
|
|||
|
|
SOCKET sock;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
BufIo in_buffer;
|
|||
|
|
BufIo out_buffer;
|
|||
|
|
|
|||
|
|
btStream() { sock = INVALID_SOCKET ;}
|
|||
|
|
~btStream() {if( INVALID_SOCKET != sock) CLOSE_SOCKET(sock);}
|
|||
|
|
|
|||
|
|
|
|||
|
|
SOCKET GetSocket() {return sock;}
|
|||
|
|
void SetSocket(SOCKET sk){ sock = sk; }
|
|||
|
|
|
|||
|
|
void Close(){
|
|||
|
|
if( INVALID_SOCKET != sock ){ CLOSE_SOCKET(sock); sock = INVALID_SOCKET;}
|
|||
|
|
in_buffer.Close();
|
|||
|
|
out_buffer.Close();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ssize_t PickMessage(); //<2F>Ƴ<EFBFBD><C6B3><EFBFBD><EFBFBD>ջ<EFBFBD><D5BB><EFBFBD><EFBFBD>е<EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD>Ϣ
|
|||
|
|
ssize_t Feed();
|
|||
|
|
|
|||
|
|
int HaveMessage(); // <20><><EFBFBD><EFBFBD>ֵ 1: <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ 0: <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ -1: ʧ<><CAA7>
|
|||
|
|
|
|||
|
|
ssize_t Send_Keepalive();
|
|||
|
|
ssize_t Send_State(unsigned char state);
|
|||
|
|
ssize_t Send_Have(size_t idx);
|
|||
|
|
ssize_t Send_Piece(size_t idx,size_t off,char *piece_buf,size_t len);
|
|||
|
|
ssize_t Send_Bitfield(char *bit_buf,size_t len);
|
|||
|
|
ssize_t Send_Request(size_t idx,size_t off,size_t len);
|
|||
|
|
ssize_t Send_Cancel(size_t idx,size_t off,size_t len);
|
|||
|
|
ssize_t Send_Buffer(char *buf,size_t len);
|
|||
|
|
|
|||
|
|
ssize_t Flush();
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
#endif
|