Commit 1005f542 authored by Corey Hickey's avatar Corey Hickey

- Add new file internal.h for common internal-use-only functions.

- Add new function av_tempfile() for creating temporary files; contains
  workaround for MinGW.
- Make XviD stuff use av_tempfile().

Originally committed as revision 5245 to svn://svn.ffmpeg.org/ffmpeg/trunk
parent 06ab9cff
#ifndef INTERNAL_H
#define INTERNAL_H
/**
* @file internal.h
* common functions for internal libavcodec use
*/
int av_tempfile(char *prefix, char **filename);
#endif /* INTERNAL_H */
...@@ -33,6 +33,9 @@ ...@@ -33,6 +33,9 @@
#include <stdarg.h> #include <stdarg.h>
#include <limits.h> #include <limits.h>
#include <float.h> #include <float.h>
#ifdef CONFIG_WIN32
#include <fcntl.h>
#endif
const uint8_t ff_reverse[256]={ const uint8_t ff_reverse[256]={
0x00,0x80,0x40,0xC0,0x20,0xA0,0x60,0xE0,0x10,0x90,0x50,0xD0,0x30,0xB0,0x70,0xF0, 0x00,0x80,0x40,0xC0,0x20,0xA0,0x60,0xE0,0x10,0x90,0x50,0xD0,0x30,0xB0,0x70,0xF0,
...@@ -1349,3 +1352,39 @@ unsigned int av_xiphlacing(unsigned char *s, unsigned int v) ...@@ -1349,3 +1352,39 @@ unsigned int av_xiphlacing(unsigned char *s, unsigned int v)
n++; n++;
return n; return n;
} }
/* Wrapper to work around the lack of mkstemp() on mingw/cygin.
* Also, tries to create file in /tmp first, if possible.
* *prefix can be a character constant; *filename will be allocated internally.
* Returns file descriptor of opened file (or -1 on error)
* and opened file name in **filename. */
int av_tempfile(char *prefix, char **filename) {
int fd=-1;
#ifdef CONFIG_WIN32
*filename = tempnam(".", prefix);
#else
size_t len = strlen(prefix) + 12; /* room for "/tmp/" and "XXXXXX\0" */
*filename = av_malloc(len * sizeof(char));
#endif
/* -----common section-----*/
if (*filename == NULL) {
av_log(NULL, AV_LOG_ERROR, "ff_tempfile: Cannot allocate file name\n");
return -1;
}
#ifdef CONFIG_WIN32
fd = open(*filename, _O_RDWR | _O_BINARY | _O_CREAT, 0444);
#else
snprintf(*filename, len, "/tmp/%sXXXXXX", prefix);
fd = mkstemp(*filename);
if (fd < 0) {
snprintf(*filename, len, "./%sXXXXXX", prefix);
fd = mkstemp(*filename);
}
#endif
/* -----common section-----*/
if (fd < 0) {
av_log(NULL, AV_LOG_ERROR, "ff_tempfile: Cannot open temporary file %s\n", *filename);
return -1;
}
return fd; /* success */
}
...@@ -21,6 +21,7 @@ ...@@ -21,6 +21,7 @@
#include <xvid.h> #include <xvid.h>
#include <unistd.h> #include <unistd.h>
#include "avcodec.h" #include "avcodec.h"
#include "internal.h"
//#include "dsputil.h" //#include "dsputil.h"
#include "mpegvideo.h" #include "mpegvideo.h"
...@@ -37,11 +38,10 @@ int ff_xvid_rate_control_init(MpegEncContext *s){ ...@@ -37,11 +38,10 @@ int ff_xvid_rate_control_init(MpegEncContext *s){
//xvid_debug=-1; //xvid_debug=-1;
tmp_name= av_strdup("/tmp/xvidrc.XXXXXX"); fd=av_tempfile("xvidrc.", &tmp_name);
fd = mkstemp(tmp_name); if (fd == -1) {
if(fd < 0){ av_log(NULL, AV_LOG_ERROR, "Can't create temporary pass2 file.\n");
strcpy(tmp_name, "./xvidrc.XXXXXX"); return -1;
fd = mkstemp(tmp_name);
} }
for(i=0; i<s->rc_context.num_entries; i++){ for(i=0; i<s->rc_context.num_entries; i++){
......
...@@ -27,9 +27,7 @@ ...@@ -27,9 +27,7 @@
#include <unistd.h> #include <unistd.h>
#include "common.h" #include "common.h"
#include "avcodec.h" #include "avcodec.h"
#ifdef CONFIG_WIN32 #include "internal.h"
#include <fcntl.h>
#endif
/** /**
* Buffer management macros. * Buffer management macros.
...@@ -229,39 +227,7 @@ int ff_xvid_encode_init(AVCodecContext *avctx) { ...@@ -229,39 +227,7 @@ int ff_xvid_encode_init(AVCodecContext *avctx) {
rc2pass2.version = XVID_VERSION; rc2pass2.version = XVID_VERSION;
rc2pass2.bitrate = avctx->bit_rate; rc2pass2.bitrate = avctx->bit_rate;
#ifdef CONFIG_WIN32 /* Ugly work around */ fd = av_tempfile("xvidff.", &(x->twopassfile));
{
char *tempname;
tempname = tempnam(".", "xvidff");
fd = -1;
if( tempname &&
(fd = open(tempname, _O_RDWR | _O_BINARY)) != -1 ) {
x->twopassfile = av_strdup(tempname);
#undef free
free(tempname);
#define free please_use_av_free
if( x->twopassfile == NULL ) {
av_log(avctx, AV_LOG_ERROR,
"XviD: Cannot allocate 2-pass buffer\n");
return -1;
}
}
}
#else
x->twopassfile = av_malloc(BUFFER_SIZE);
if( x->twopassfile == NULL ) {
av_log(avctx, AV_LOG_ERROR,
"XviD: Cannot allocate 2-pass buffer\n");
return -1;
}
strcpy(x->twopassfile, "/tmp/xvidff.XXXXXX");
fd = mkstemp(x->twopassfile);
if(fd < 0){
strcpy(x->twopassfile, "./xvidff.XXXXXX");
fd = mkstemp(x->twopassfile);
}
#endif
if( fd == -1 ) { if( fd == -1 ) {
av_log(avctx, AV_LOG_ERROR, av_log(avctx, AV_LOG_ERROR,
"XviD: Cannot write 2-pass pipe\n"); "XviD: Cannot write 2-pass pipe\n");
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment