hydrar/mistserver
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Buffer crash fix

Browse source
This commit is contained in:
Thulinma 2010-11-08 03:12:24 +01:00
parent cd0ed3be75
commit 84c7c9c275
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
util/flv.cpp
View file

@ -1,4 +1,5 @@
#include <unistd.h> //for read() #include <unistd.h> //for read()
#include <fcntl.h>
struct FLV_Pack { struct FLV_Pack {
int len; int len;
@ -47,6 +48,9 @@ bool ReadUntil(char * buffer, unsigned int count, unsigned int & sofar){
//resizes FLV_Pack data field bigger if data doesn't fit //resizes FLV_Pack data field bigger if data doesn't fit
// (does not auto-shrink for speed!) // (does not auto-shrink for speed!)
bool FLV_GetPacket(FLV_Pack *& p){ bool FLV_GetPacket(FLV_Pack *& p){
int preflags = fcntl(fileno(stdin), F_GETFL, 0);
int postflags = preflags | O_NONBLOCK;
fcntl(fileno(stdin), F_SETFL, postflags);
static bool done = true; static bool done = true;
static unsigned int sofar = 0; static unsigned int sofar = 0;
if (!p){p = (FLV_Pack*)calloc(1, sizeof(FLV_Pack));} if (!p){p = (FLV_Pack*)calloc(1, sizeof(FLV_Pack));}
@ -80,9 +84,11 @@ bool FLV_GetPacket(FLV_Pack *& p){
if ((p->data[0] == 0x09) && (((p->data[11] & 0xf0) >> 4) == 1)){p->isKeyframe = true;} if ((p->data[0] == 0x09) && (((p->data[11] & 0xf0) >> 4) == 1)){p->isKeyframe = true;}
done = true; done = true;
sofar = 0; sofar = 0;
fcntl(fileno(stdin), F_SETFL, preflags);
return true; return true;
} }
} }
fcntl(fileno(stdin), F_SETFL, preflags);
return false; return false;
}//FLV_GetPacket }//FLV_GetPacket