OSDN Git Service

Fixed a possible use-after-free bug in initialization code.
[lamexp/LameXP.git] / src / Encoder_Wave.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2015 LoRd_MuldeR <MuldeR2@GMX.de>
4 //
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; either version 2 of the License, or
8 // (at your option) any later version, but always including the *additional*
9 // restrictions defined in the "License.txt" file.
10 //
11 // This program is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 // GNU General Public License for more details.
15 //
16 // You should have received a copy of the GNU General Public License along
17 // with this program; if not, write to the Free Software Foundation, Inc.,
18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 //
20 // http://www.gnu.org/licenses/gpl-2.0.txt
21 ///////////////////////////////////////////////////////////////////////////////
22
23 #include "Encoder_Wave.h"
24
25 #include "Global.h"
26 #include "Model_Settings.h"
27
28 #include <QDir>
29
30 //Windows includes
31 #define NOMINMAX
32 #define WIN32_LEAN_AND_MEAN
33 #include <Windows.h>
34 #include <Shellapi.h>
35
36 #define FIX_SEPARATORS(STR) for(int i = 0; STR[i]; i++) { if(STR[i] == L'/') STR[i] = L'\\'; }
37
38 ///////////////////////////////////////////////////////////////////////////////
39 // Encoder Info
40 ///////////////////////////////////////////////////////////////////////////////
41
42 class WaveEncoderInfo : public AbstractEncoderInfo
43 {
44 public:
45         virtual bool isModeSupported(int mode) const
46         {
47                 switch(mode)
48                 {
49                 case SettingsModel::VBRMode:
50                 case SettingsModel::ABRMode:
51                         return false;
52                         break;
53                 case SettingsModel::CBRMode:
54                         return true;
55                         break;
56                 default:
57                         MUTILS_THROW("Bad RC mode specified!");
58                 }
59         }
60
61         virtual int valueCount(int mode) const
62         {
63                 switch(mode)
64                 {
65                 case SettingsModel::VBRMode:
66                 case SettingsModel::ABRMode:
67                 case SettingsModel::CBRMode:
68                         return 0;
69                         break;
70                 default:
71                         MUTILS_THROW("Bad RC mode specified!");
72                 }
73         }
74
75         virtual int valueAt(int mode, int index) const
76         {
77                 switch(mode)
78                 {
79                 case SettingsModel::VBRMode:
80                 case SettingsModel::ABRMode:
81                 case SettingsModel::CBRMode:
82                         return -1;
83                         break;
84                 default:
85                         MUTILS_THROW("Bad RC mode specified!");
86                 }
87         }
88
89         virtual int valueType(int mode) const
90         {
91                 switch(mode)
92                 {
93                 case SettingsModel::VBRMode:
94                 case SettingsModel::ABRMode:
95                 case SettingsModel::CBRMode:
96                         return TYPE_UNCOMPRESSED;
97                         break;
98                 default:
99                         MUTILS_THROW("Bad RC mode specified!");
100                 }
101         }
102
103         virtual const char *description(void) const
104         {
105                 static const char* s_description = "Wave Audio (PCM)";
106                 return s_description;
107         }
108
109         virtual const char *extension(void) const
110         {
111                 static const char* s_extension = "wav";
112                 return s_extension;
113         }
114 }
115 static const g_waveEncoderInfo;
116
117 ///////////////////////////////////////////////////////////////////////////////
118 // Encoder implementation
119 ///////////////////////////////////////////////////////////////////////////////
120
121
122 WaveEncoder::WaveEncoder(void)
123 {
124 }
125
126 WaveEncoder::~WaveEncoder(void)
127 {
128 }
129
130 bool WaveEncoder::encode(const QString &sourceFile, const AudioFileModel_MetaInfo &metaInfo, const unsigned int duration, const QString &outputFile, volatile bool *abortFlag)
131 {
132         SHFILEOPSTRUCTW fileOperation;
133         memset(&fileOperation, 0, sizeof(SHFILEOPSTRUCTW));
134         fileOperation.fFlags = FOF_SILENT | FOF_NOCONFIRMATION | FOF_NOCONFIRMMKDIR | FOF_NOERRORUI | FOF_FILESONLY;
135
136         emit messageLogged(QString("Copy file \"%1\" to \"%2\"").arg(sourceFile, outputFile));
137         fileOperation.wFunc = FO_COPY;
138
139         /*
140         if(lamexp_temp_folder().compare(QFileInfo(sourceFile).canonicalPath(), Qt::CaseInsensitive) == 0)
141         {
142                 //If the source is in the TEMP folder take shortcut and move the file
143                 emit messageLogged(QString("Moving file \"%1\" to \"%2\"").arg(sourceFile, outputFile));
144                 fileOperation.wFunc = FO_MOVE;
145         }
146         else
147         {
148                 //...otherwise we actually copy the file in order to keep the source
149                 emit messageLogged(QString("Copy file \"%1\" to \"%2\"").arg(sourceFile, outputFile));
150                 fileOperation.wFunc = FO_COPY;
151         }
152         */
153         
154         size_t srcLen = wcslen(reinterpret_cast<const wchar_t*>(sourceFile.utf16())) + 3;
155         wchar_t *srcBuffer = new wchar_t[srcLen];
156         memset(srcBuffer, 0, srcLen * sizeof(wchar_t));
157         wcsncpy_s(srcBuffer, srcLen, reinterpret_cast<const wchar_t*>(sourceFile.utf16()), _TRUNCATE);
158         FIX_SEPARATORS (srcBuffer);
159         fileOperation.pFrom = srcBuffer;
160
161         size_t outLen = wcslen(reinterpret_cast<const wchar_t*>(outputFile.utf16())) + 3;
162         wchar_t *outBuffer = new wchar_t[outLen];
163         memset(outBuffer, 0, outLen * sizeof(wchar_t));
164         wcsncpy_s(outBuffer, outLen, reinterpret_cast<const wchar_t*>(outputFile.utf16()), _TRUNCATE);
165         FIX_SEPARATORS (outBuffer);
166         fileOperation.pTo = outBuffer;
167
168         emit statusUpdated(0);
169         int result = SHFileOperation(&fileOperation);
170         emit statusUpdated(100);
171
172         emit messageLogged(QString().sprintf("\nExited with code: 0x%04X", result));
173
174         delete [] srcBuffer;
175         delete [] outBuffer;
176
177         return (result == 0 && fileOperation.fAnyOperationsAborted == false);
178 }
179
180 bool WaveEncoder::isFormatSupported(const QString &containerType, const QString &containerProfile, const QString &formatType, const QString &formatProfile, const QString &formatVersion)
181 {
182         if(containerType.compare("Wave", Qt::CaseInsensitive) == 0)
183         {
184                 if(formatType.compare("PCM", Qt::CaseInsensitive) == 0)
185                 {
186                         return true;
187                 }
188         }
189         return false;
190 }
191
192 const AbstractEncoderInfo *WaveEncoder::getEncoderInfo(void)
193 {
194         return &g_waveEncoderInfo;
195 }