00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
#include <config.h>
00024
00025
#include <sys/types.h>
00026
00027
#ifdef HAVE_SYS_STAT_H
00028
#include <sys/stat.h>
00029
#endif
00030
00031
#include <fcntl.h>
00032
#include <stdlib.h>
00033
#include <unistd.h>
00034
00035
#ifdef HAVE_TEST
00036
#include <test.h>
00037
#endif
00038
#ifdef HAVE_PATHS_H
00039
#include <paths.h>
00040
#endif
00041
00042
#ifndef _PATH_TMP
00043
#define _PATH_TMP "/tmp"
00044
#endif
00045
00046
#include <qdatetime.h>
00047
#include <qfile.h>
00048
#include <qdatastream.h>
00049
#include <qtextstream.h>
00050
00051
#include "kglobal.h"
00052
#include "kapplication.h"
00053
#include "kinstance.h"
00054
#include "ktempfile.h"
00055
#include "kstandarddirs.h"
00056
00057
00058
00059 KTempFile::KTempFile(
QString filePrefix,
QString fileExtension,
int mode)
00060 {
00061 bAutoDelete =
false;
00062 mFd = -1;
00063 mStream = 0;
00064 mFile = 0;
00065 mTextStream = 0;
00066 mDataStream = 0;
00067 mError = 0;
00068 bOpen =
false;
00069
if (fileExtension.isEmpty())
00070 fileExtension =
".tmp";
00071
if (filePrefix.isEmpty())
00072 {
00073 filePrefix = locateLocal(
"tmp", KGlobal::instance()->instanceName());
00074 }
00075 (
void) create(filePrefix, fileExtension, mode);
00076 }
00077
00078 KTempFile::KTempFile(
bool)
00079 {
00080 bAutoDelete =
false;
00081 mFd = -1;
00082 mStream = 0;
00083 mFile = 0;
00084 mTextStream = 0;
00085 mDataStream = 0;
00086 mError = 0;
00087 bOpen =
false;
00088 }
00089
00090
bool
00091 KTempFile::create(
const QString &filePrefix,
const QString &fileExtension,
00092
int mode)
00093 {
00094
00095 (
void)
KApplication::random();
00096
00097
QCString ext = QFile::encodeName(fileExtension);
00098
QCString nme = QFile::encodeName(filePrefix) +
"XXXXXX" + ext;
00099
if((mFd = mkstemps(nme.data(), ext.length())) < 0)
00100 {
00101
00102
QCString nme = QFile::encodeName(filePrefix) +
"XXXXXX" + ext;
00103 qWarning(
"KTempFile: Error trying to create %s: %s", nme.data(), strerror(errno));
00104 mError = errno;
00105 mTmpName = QString::null;
00106
return false;
00107 }
00108
00109
00110 mTmpName = QFile::decodeName(nme);
00111 mode_t tmp = 0;
00112 mode_t umsk = umask(tmp);
00113 umask(umsk);
00114 fchmod(mFd, mode&(~umsk));
00115
00116
00117 bOpen =
true;
00118
00119
00120 fchown(mFd, getuid(), getgid());
00121
00122
00123 fcntl(mFd, F_SETFD, FD_CLOEXEC);
00124
00125
return true;
00126 }
00127
00128 KTempFile::~KTempFile()
00129 {
00130
close();
00131
if (bAutoDelete)
00132
unlink();
00133 }
00134
00135
int
00136 KTempFile::status()
const
00137
{
00138
return mError;
00139 }
00140
00141
QString
00142 KTempFile::name()
const
00143
{
00144
return mTmpName;
00145 }
00146
00147
int
00148 KTempFile::handle()
const
00149
{
00150
return mFd;
00151 }
00152
00153 FILE *
00154 KTempFile::fstream()
00155 {
00156
if (mStream)
return mStream;
00157
if (mFd < 0)
return 0;
00158
00159
00160 mStream = fdopen(mFd,
"r+");
00161
if (!mStream) {
00162 qWarning(
"KTempFile: Error trying to open %s: %s", mTmpName.latin1(), strerror(errno));
00163 mError = errno;
00164 }
00165
return mStream;
00166 }
00167
00168
QFile *
00169 KTempFile::file()
00170 {
00171
if (mFile)
return mFile;
00172
if ( !
fstream() )
return 0;
00173
00174 mFile =
new QFile();
00175 mFile->setName(
name() );
00176 mFile->open(IO_ReadWrite, mStream);
00177
return mFile;
00178 }
00179
00180
QTextStream *
00181 KTempFile::textStream()
00182 {
00183
if (mTextStream)
return mTextStream;
00184
if ( !
file() )
return 0;
00185
00186 mTextStream =
new QTextStream( mFile );
00187
return mTextStream;
00188 }
00189
00190
QDataStream *
00191 KTempFile::dataStream()
00192 {
00193
if (mDataStream)
return mDataStream;
00194
if ( !
file() )
return 0;
00195
00196 mDataStream =
new QDataStream( mFile );
00197
return mDataStream;
00198 }
00199
00200
void
00201 KTempFile::unlink()
00202 {
00203
if (!mTmpName.isEmpty())
00204 QFile::remove( mTmpName );
00205 mTmpName = QString::null;
00206 }
00207
00208
#if defined(_POSIX_SYNCHRONIZED_IO) && _POSIX_SYNCHRONIZED_IO > 0
00209
#define FDATASYNC fdatasync
00210
#else
00211
#define FDATASYNC fsync
00212
#endif
00213
00214
bool
00215 KTempFile::sync()
00216 {
00217
int result = 0;
00218
00219
if (mStream)
00220 {
00221
do {
00222 result = fflush(mStream);
00223 }
00224
while ((result == -1) && (errno == EINTR));
00225
00226
if (result)
00227 {
00228 qWarning(
"KTempFile: Error trying to flush %s: %s", mTmpName.latin1(), strerror(errno));
00229 mError = errno;
00230 }
00231 }
00232
00233
if (mFd >= 0)
00234 {
00235 result = FDATASYNC(mFd);
00236
if (result)
00237 {
00238 qWarning(
"KTempFile: Error trying to sync %s: %s", mTmpName.latin1(), strerror(errno));
00239 mError = errno;
00240 }
00241 }
00242
00243
return (mError == 0);
00244 }
00245
00246
#undef FDATASYNC
00247
00248
bool
00249 KTempFile::close()
00250 {
00251
int result = 0;
00252
delete mTextStream; mTextStream = 0;
00253
delete mDataStream; mDataStream = 0;
00254
delete mFile; mFile = 0;
00255
00256
if (mStream)
00257 {
00258 result = ferror(mStream);
00259
if (result)
00260 mError = ENOSPC;
00261
00262 result = fclose(mStream);
00263 mStream = 0;
00264 mFd = -1;
00265
if (result != 0) {
00266 qWarning(
"KTempFile: Error trying to close %s: %s", mTmpName.latin1(), strerror(errno));
00267 mError = errno;
00268 }
00269 }
00270
00271
00272
if (mFd >= 0)
00273 {
00274 result = ::close(mFd);
00275 mFd = -1;
00276
if (result != 0) {
00277 qWarning(
"KTempFile: Error trying to close %s: %s", mTmpName.latin1(), strerror(errno));
00278 mError = errno;
00279 }
00280 }
00281
00282 bOpen =
false;
00283
return (mError == 0);
00284 }
00285