www.pudn.com > ffmpeg_win32.rar > file.c

/* 
 * Buffered file io for ffmpeg system 
 * Copyright (c) 2001 Fabrice Bellard 
 * 
 * This file is part of FFmpeg. 
 * 
 * FFmpeg is free software; you can redistribute it and/or 
 * modify it under the terms of the GNU Lesser General Public 
 * License as published by the Free Software Foundation; either 
 * version 2.1 of the License, or (at your option) any later version. 
 * 
 * FFmpeg 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 
 * Lesser General Public License for more details. 
 * 
 * You should have received a copy of the GNU Lesser General Public 
 * License along with FFmpeg; if not, write to the Free Software 
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 
 */ 
#include "avformat.h" 
#include "avstring.h" 
#include  
//#include  
//#include  
//#include  
#include "os_support.h" 
 
#include  
//#include  
#include  
#include  
#include  
 
 
/* standard file protocol */ 
 
static int file_open(URLContext *h, const char *filename, int flags) 
{ 
    int access;/* 
	HANDLE hfile; 
	WCHAR wfilename[128]; 
	int result;*/ 
    int fd; 
 
    av_strstart(filename, "file:", &filename); 
 
    if (flags & URL_RDWR) { 
        access = O_CREAT | O_TRUNC | O_RDWR; 
		//access=GENERIC_READ | GENERIC_WRITE; 
    } else if (flags & URL_WRONLY) { 
        access = O_CREAT | O_TRUNC | O_WRONLY; 
		//access=GENERIC_WRITE; 
    } else { 
        access = O_RDONLY; 
		//access=GENERIC_READ; 
    } 
#ifdef O_BINARY 
    access |= O_BINARY; 
#endif 
 
	//memset(wfilename,0,sizeof(WCHAR)*128); 
	//result=MultiByteToWideChar(CP_ACP,0,filename,strlen(filename),wfilename,128); 
	//if(!result) 
	//	return AVERROR(ENOENT);		//wide char 
 
 
	//hfile=CreateFile(wfilename,access,FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL, NULL); 
 
    //if (hfile==INVALID_HANDLE_VALUE) 
    //    return AVERROR(ENOENT); 
 
    fd = open(filename, access, _S_IREAD | _S_IWRITE); 
	if (fd < 0) 
        return AVERROR(ENOENT); 
 
    h->priv_data = /*hfile; */(void *)(size_t)fd; 
    return 0; 
} 
 
static int file_read(URLContext *h, unsigned char *buf, int size) 
{ 
    int fd = (size_t)h->priv_data; 
    return read(fd, buf, size); 
 
	/*int result; 
	result = ReadFile(h->priv_data,buf,size,(DWORD*)&size,NULL); 
	if(!result) 
		return -1; 
	else 
		return size;*/ 
} 
 
static int file_write(URLContext *h, unsigned char *buf, int size) 
{ 
    int fd = (size_t)h->priv_data; 
	return write(fd,buf,size); 
 
	//int result; 
	//result=WriteFile(h->priv_data,buf,size,(DWORD*)&size,NULL); //(fd, buf, size); 
	//if(!result) 
	//	return -1; 
	//else 
	//	return size; 
} 
 
/* XXX: use llseek */ 
static offset_t file_seek(URLContext *h, offset_t pos, int whence) 
{ 
    //int32_t high,low,result; 
	//high=pos>>32; 
	//low=pos&0xFFFFFFFF;	 
	//result=SetFilePointer(file, low_part_pos, (LONG*)&__high_part_pos, FILE_BEGIN);  
	//if(result==INVALID_SET_FILE_POINTER){ 
	//	return -1; 
	//}else{ 
	//	return  
	//} 
 
	int fd = (size_t)h->priv_data; 
    return lseek(fd, pos, whence); 
} 
 
static int file_close(URLContext *h) 
{ 
    int fd = (size_t)h->priv_data; 
    return close(fd); 
} 
 
URLProtocol file_protocol = { 
    "file", 
    file_open, 
    file_read, 
    file_write, 
    file_seek, 
    file_close, 
}; 
 
/* pipe protocol */ 
 
static int pipe_open(URLContext *h, const char *filename, int flags) 
{ 
    int fd; 
    const char * final; 
    av_strstart(filename, "pipe:", &filename); 
 
    fd = strtol(filename, &final, 10); 
    if((filename == final) || *final ) {/* No digits found, or something like 10ab */ 
        if (flags & URL_WRONLY) { 
            fd = 1; 
        } else { 
            fd = 0; 
        } 
    } 
#ifdef O_BINARY 
    setmode(fd, O_BINARY); 
#endif 
    h->priv_data = (void *)(size_t)fd; 
    h->is_streamed = 1; 
    return 0; 
} 
 
URLProtocol pipe_protocol = { 
    "pipe", 
    pipe_open, 
    file_read, 
    file_write, 
};