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

Gearbox updates, FLV parser rewrite, buffer fixes...

Browse source
This commit is contained in:
Thulinma 2010-11-07 14:49:45 +01:00
parent 3fb3d79ba0
commit 731a212bb0
1 changed files with 65 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

93
util/flv.cpp
View file

@ -8,43 +8,80 @@ struct FLV_Pack {
};//FLV_Pack };//FLV_Pack
char FLVHeader[13]; char FLVHeader[13];
bool All_Hell_Broke_Loose = false;
//reads full length from a file descriptor //checks FLV Header for correctness
void Magic_Read(char * buf, int len, int file){
int i = 0;
while (i < len) i += read(file, buf, len-i);
}
//reads a FLV header and checks for correctness
//returns true if everything is alright, false otherwise //returns true if everything is alright, false otherwise
bool FLV_Readheader(){ bool FLV_Checkheader(char * header){
fread(FLVHeader,1,13,stdin); if (header[0] != 'F') return false;
if (FLVHeader[0] != 'F') return false; if (header[1] != 'L') return false;
if (FLVHeader[1] != 'L') return false; if (header[2] != 'V') return false;
if (FLVHeader[2] != 'V') return false; if (header[8] != 0x09) return false;
if (FLVHeader[8] != 0x09) return false; if (header[9] != 0) return false;
if (FLVHeader[9] != 0) return false; if (header[10] != 0) return false;
if (FLVHeader[10] != 0) return false; if (header[11] != 0) return false;
if (FLVHeader[11] != 0) return false; if (header[12] != 0) return false;
if (FLVHeader[12] != 0) return false;
return true; return true;
}//FLV_Readheader }//FLV_Checkheader
//returns true if header is an FLV header
bool FLV_Isheader(char * header){
if (header[0] != 'F') return false;
if (header[1] != 'L') return false;
if (header[2] != 'V') return false;
return true;
}//FLV_Isheader
bool ReadUntil(char * buffer, unsigned int count, unsigned int & sofar){
if (sofar >= count){return true;}
int r = 0;
r = fread(buffer + sofar,1,count-sofar,stdin);
if (r < 0){All_Hell_Broke_Loose = true; return false;}
sofar += r;
if (sofar >= count){return true;}
return false;
}
//gets a packet, storing in given FLV_Pack pointer. //gets a packet, storing in given FLV_Pack pointer.
//will assign pointer if null //will assign pointer if null
//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){
static bool done = true;
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));}
if (p->buf < 15){p->data = (char*)realloc(p->data, 15); p->buf = 15;} if (p->buf < 15){p->data = (char*)realloc(p->data, 15); p->buf = 15;}
if (fread(p->data,1,11,stdin) != 11){return false;}
p->len = p->data[3] + 15; if (done){
p->len += (p->data[2] << 8); //read a header
p->len += (p->data[1] << 16); if (ReadUntil(p->data, 11, sofar)){
if (p->buf < p->len){p->data = (char*)realloc(p->data, p->len);p->buf = p->len;} //if its a correct FLV header, throw away and read tag header
if (fread(p->data+11,1,p->len-11,stdin) != (unsigned int)(p->len-11)){return false;} if (FLV_Isheader(p->data)){
p->isKeyframe = false; if (ReadUntil(p->data, 13, sofar)){
if ((p->data[0] == 0x09) && (((p->data[11] & 0xf0) >> 4) == 1)){p->isKeyframe = true;} if (FLV_Checkheader(p->data)){
return true; sofar = 0;
memcpy(FLVHeader, p->data, 13);
}else{All_Hell_Broke_Loose = true;}
}
}else{
//if a tag header, calculate length and read tag body
p->len = p->data[3] + 15;
p->len += (p->data[2] << 8);
p->len += (p->data[1] << 16);
if (p->buf < p->len){p->data = (char*)realloc(p->data, p->len);p->buf = p->len;}
done = false;
}
}
}else{
//read tag body
if (ReadUntil(p->data, p->len, sofar)){
//calculate keyframeness, next time read header again, return true
p->isKeyframe = false;
if ((p->data[0] == 0x09) && (((p->data[11] & 0xf0) >> 4) == 1)){p->isKeyframe = true;}
done = true;
return true;
}
}
return false;
}//FLV_GetPacket }//FLV_GetPacket