![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to commandlineinterface.cpp CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [truecrypt] / truecrypt / main|
Return to commandlineinterface.cpp CVS log | Up to [truecrypt] / truecrypt / main |
1.1 ! root 1: /* ! 2: Copyright (c) 2008 TrueCrypt Foundation. All rights reserved. ! 3: ! 4: Governed by the TrueCrypt License 2.4 the full text of which is contained ! 5: in the file License.txt included in TrueCrypt binary and source code ! 6: distribution packages. ! 7: */ ! 8: ! 9: #include "System.h" ! 10: #include <wx/cmdline.h> ! 11: #include <wx/tokenzr.h> ! 12: #include "Core/Core.h" ! 13: #include "CommandLineInterface.h" ! 14: #include "LanguageStrings.h" ! 15: #include "UserInterfaceException.h" ! 16: ! 17: namespace TrueCrypt ! 18: { ! 19: CommandLineInterface::CommandLineInterface (wxCmdLineParser &parser, UserInterfaceType::Enum interfaceType) : ! 20: ArgCommand (CommandId::None), ! 21: StartBackgroundTask (false) ! 22: { ! 23: parser.SetSwitchChars (L"-"); ! 24: ! 25: parser.AddOption (L"", L"auto-mount", _("Auto mount device-hosted/favorite volumes")); ! 26: parser.AddSwitch (L"", L"background-task", _("Start Background Task")); ! 27: parser.AddSwitch (L"d", L"dismount", _("Dismount volume")); ! 28: parser.AddSwitch (L"", L"cache", _("Cache passwords and keyfiles")); ! 29: parser.AddSwitch (L"C", L"change", _("Change password or keyfiles")); ! 30: parser.AddSwitch (L"", L"explore", _("Open explorer window for mounted volume")); ! 31: parser.AddSwitch (L"f", L"force", _("Force mount/dismount/overwrite")); ! 32: parser.AddSwitch (L"h", L"help", _("Display help"), wxCMD_LINE_OPTION_HELP); ! 33: parser.AddOption (L"k", L"keyfiles", _("Keyfiles")); ! 34: parser.AddOption (L"", L"filesystem", _("Filesystem type")); ! 35: #if !defined(TC_WINDOWS) && !defined(TC_MACOSX) ! 36: parser.AddOption (L"", L"fs-options", _("Filesystem options")); ! 37: #endif ! 38: parser.AddSwitch (L"l", L"list", _("List mounted volumes")); ! 39: parser.AddOption (L"m", L"mount-options", _("Mount options")); ! 40: parser.AddOption (L"", L"new-keyfiles", _("New keyfiles")); ! 41: parser.AddOption (L"", L"new-password", _("New password")); ! 42: parser.AddSwitch (L"", L"non-interactive", _("Do not interact with user")); ! 43: parser.AddOption (L"p", L"password", _("Password")); ! 44: parser.AddSwitch (L"", L"load-preferences", _("Load user preferences")); ! 45: parser.AddSwitch (L"", L"protect-hidden", _("Protect hidden volume")); ! 46: parser.AddOption (L"", L"protection-keyfiles", _("Keyfiles for protected hidden volume")); ! 47: parser.AddOption (L"", L"protection-password", _("Password for protected hidden volume")); ! 48: parser.AddSwitch (L"v", L"verbose", _("Enable verbose output")); ! 49: parser.AddOption (L"", L"slot", _("Volume slot number")); ! 50: parser.AddSwitch (L"", L"test", _("Test internal algorithms")); ! 51: parser.AddSwitch (L"t", L"text", _("Use text user interface")); ! 52: parser.AddSwitch (L"", L"version", _("Display version information")); ! 53: parser.AddParam ( _("Volume path"), wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL); ! 54: parser.AddParam ( _("Mount point"), wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL); ! 55: ! 56: wxString str; ! 57: bool param1IsVolume = false; ! 58: bool param1IsMountedVolumeSpec = false; ! 59: bool param1IsMountPoint = false; ! 60: ! 61: if (parser.Parse () > 0) ! 62: throw_err (_("Incorrect command line specified.")); ! 63: ! 64: if (parser.Found (L"help")) ! 65: { ! 66: ArgCommand = CommandId::Help; ! 67: return; ! 68: } ! 69: ! 70: if (parser.Found (L"text") && interfaceType != UserInterfaceType::Text) ! 71: { ! 72: wstring msg = wstring (_("Option -t or --text must be specified as the first argument.")); ! 73: wcerr << msg << endl; ! 74: throw_err (msg); ! 75: } ! 76: ! 77: if (parser.Found (L"version")) ! 78: { ! 79: ArgCommand = CommandId::DisplayVersion; ! 80: return; ! 81: } ! 82: ! 83: // Preferences ! 84: if (parser.Found (L"load-preferences")) ! 85: { ! 86: // Load preferences first to allow command line options to override them ! 87: Preferences.Load(); ! 88: ArgMountOptions = Preferences.DefaultMountOptions; ! 89: } ! 90: ! 91: // Commands ! 92: if (parser.Found (L"auto-mount", &str)) ! 93: { ! 94: CheckCommandSingle(); ! 95: ! 96: wxStringTokenizer tokenizer (str, L","); ! 97: while (tokenizer.HasMoreTokens()) ! 98: { ! 99: wxString token = tokenizer.GetNextToken(); ! 100: ! 101: if (token == L"devices") ! 102: { ! 103: if (ArgCommand == CommandId::AutoMountFavorites) ! 104: ArgCommand = CommandId::AutoMountDevicesFavorites; ! 105: else ! 106: ArgCommand = CommandId::AutoMountDevices; ! 107: ! 108: param1IsMountPoint = true; ! 109: } ! 110: else if (token == L"favorites") ! 111: { ! 112: if (ArgCommand == CommandId::AutoMountDevices) ! 113: ArgCommand = CommandId::AutoMountDevicesFavorites; ! 114: else ! 115: ArgCommand = CommandId::AutoMountFavorites; ! 116: } ! 117: else ! 118: { ! 119: throw_err (LangString["UNKNOWN_OPTION"] + L": " + token); ! 120: } ! 121: } ! 122: } ! 123: ! 124: if (parser.Found (L"change")) ! 125: { ! 126: CheckCommandSingle(); ! 127: ArgCommand = CommandId::ChangePassword; ! 128: param1IsVolume = true; ! 129: } ! 130: ! 131: if (parser.Found (L"dismount")) ! 132: { ! 133: CheckCommandSingle(); ! 134: ArgCommand = CommandId::DismountVolumes; ! 135: param1IsMountedVolumeSpec = true; ! 136: } ! 137: ! 138: if (parser.Found (L"list")) ! 139: { ! 140: CheckCommandSingle(); ! 141: ArgCommand = CommandId::ListVolumes; ! 142: param1IsMountedVolumeSpec = true; ! 143: } ! 144: ! 145: if (parser.Found (L"test")) ! 146: { ! 147: CheckCommandSingle(); ! 148: ArgCommand = CommandId::Test; ! 149: } ! 150: ! 151: // Options ! 152: if (parser.Found (L"background-task")) ! 153: StartBackgroundTask = true; ! 154: ! 155: if (parser.Found (L"cache")) ! 156: ArgMountOptions.CachePassword = true; ! 157: ! 158: if (parser.Found (L"explore")) ! 159: Preferences.OpenExplorerWindowAfterMount = true; ! 160: ! 161: if (parser.Found (L"filesystem", &str)) ! 162: { ! 163: if (str.IsSameAs (L"none", false)) ! 164: ArgMountOptions.NoFilesystem = true; ! 165: else ! 166: ArgMountOptions.FilesystemType = wstring (str); ! 167: } ! 168: ! 169: if (parser.Found (L"force")) ! 170: ArgForce = true; ! 171: ! 172: if (parser.Found (L"keyfiles", &str)) ! 173: ArgKeyfiles = ToKeyfileList (str); ! 174: ! 175: if (parser.Found (L"mount-options", &str)) ! 176: { ! 177: wxStringTokenizer tokenizer (str, L","); ! 178: while (tokenizer.HasMoreTokens()) ! 179: { ! 180: wxString token = tokenizer.GetNextToken(); ! 181: ! 182: if (token == L"readonly" || token == L"ro") ! 183: ArgMountOptions.Protection = VolumeProtection::ReadOnly; ! 184: else if (token == L"timestamp" || token == L"ts") ! 185: ArgMountOptions.PreserveTimestamps = false; ! 186: #ifdef TC_WINDOWS ! 187: else if (token == L"removable" || token == L"rm") ! 188: ArgMountOptions.Removable = true; ! 189: #endif ! 190: else ! 191: throw_err (LangString["UNKNOWN_OPTION"] + L": " + token); ! 192: } ! 193: } ! 194: ! 195: if (parser.Found (L"new-keyfiles", &str)) ! 196: ArgNewKeyfiles = ToKeyfileList (str); ! 197: ! 198: if (parser.Found (L"new-password", &str)) ! 199: ArgNewPassword.reset (new VolumePassword (wstring (str))); ! 200: ! 201: if (parser.Found (L"non-interactive")) ! 202: Preferences.NonInteractive = true; ! 203: ! 204: if (parser.Found (L"password", &str)) ! 205: ArgPassword.reset (new VolumePassword (wstring (str))); ! 206: ! 207: if (parser.Found (L"protect-hidden") && ArgMountOptions.Protection != VolumeProtection::ReadOnly) ! 208: ArgMountOptions.Protection = VolumeProtection::HiddenVolumeReadOnly; ! 209: ! 210: if (parser.Found (L"protection-keyfiles", &str)) ! 211: { ! 212: ArgMountOptions.ProtectionKeyfiles = ToKeyfileList (str); ! 213: ArgMountOptions.Protection = VolumeProtection::HiddenVolumeReadOnly; ! 214: } ! 215: ! 216: if (parser.Found (L"protection-password", &str)) ! 217: { ! 218: ArgMountOptions.ProtectionPassword.reset (new VolumePassword (wstring (str))); ! 219: ArgMountOptions.Protection = VolumeProtection::HiddenVolumeReadOnly; ! 220: } ! 221: ! 222: if (parser.Found (L"slot", &str)) ! 223: { ! 224: unsigned long number; ! 225: if (!str.ToULong (&number) || number < Core->GetFirstSlotNumber() || number > Core->GetLastSlotNumber()) ! 226: throw_err (LangString["PARAMETER_INCORRECT"] + L": " + str); ! 227: ! 228: ArgMountOptions.SlotNumber = number; ! 229: ! 230: if (param1IsMountedVolumeSpec) ! 231: { ! 232: shared_ptr <VolumeInfo> volume = Core->GetMountedVolume (number); ! 233: if (!volume) ! 234: throw_err (_("No such volume is mounted.")); ! 235: ! 236: ArgVolumes.push_back (volume); ! 237: param1IsMountedVolumeSpec = false; ! 238: } ! 239: } ! 240: ! 241: if (parser.Found (L"verbose")) ! 242: Preferences.Verbose = true; ! 243: ! 244: // Parameters ! 245: if (parser.GetParamCount() > 0) ! 246: { ! 247: if (ArgCommand == CommandId::None) ! 248: { ! 249: ArgCommand = CommandId::MountVolume; ! 250: param1IsVolume = true; ! 251: } ! 252: ! 253: if (param1IsVolume) ! 254: { ! 255: wxFileName volPath (parser.GetParam (0)); ! 256: ! 257: #ifdef TC_WINDOWS ! 258: if (!parser.GetParam (0).StartsWith (L"\\Device\\")) ! 259: #endif ! 260: volPath.Normalize (wxPATH_NORM_ABSOLUTE | wxPATH_NORM_DOTS); ! 261: ! 262: ArgVolumePath.reset (new VolumePath (wstring (volPath.GetFullPath()))); ! 263: } ! 264: ! 265: if (param1IsMountPoint || parser.GetParamCount() >= 2) ! 266: { ! 267: wstring s (parser.GetParam (param1IsMountPoint ? 0 : 1)); ! 268: wxFileName mountPoint (wstring (Directory::AppendSeparator (s))); ! 269: mountPoint.Normalize (wxPATH_NORM_ABSOLUTE | wxPATH_NORM_DOTS); ! 270: ArgMountPoint.reset (new DirectoryPath (wstring (mountPoint.GetPath()))); ! 271: } ! 272: } ! 273: ! 274: if (param1IsMountedVolumeSpec) ! 275: ArgVolumes = GetMountedVolumes (parser.GetParamCount() > 0 ? parser.GetParam (0) : wxString()); ! 276: } ! 277: ! 278: CommandLineInterface::~CommandLineInterface () ! 279: { ! 280: } ! 281: ! 282: void CommandLineInterface::CheckCommandSingle () const ! 283: { ! 284: if (ArgCommand != CommandId::None) ! 285: throw_err (_("Only a single command can be specified at a time.")); ! 286: } ! 287: ! 288: shared_ptr <KeyfileList> CommandLineInterface::ToKeyfileList (const wxString &arg) const ! 289: { ! 290: wxStringTokenizer tokenizer (arg, L",", wxTOKEN_RET_EMPTY_ALL); ! 291: ! 292: // Handle escaped separator ! 293: wxArrayString arr; ! 294: bool prevEmpty = false; ! 295: while (tokenizer.HasMoreTokens()) ! 296: { ! 297: wxString token = tokenizer.GetNextToken(); ! 298: ! 299: if (prevEmpty && token.empty() && tokenizer.HasMoreTokens()) ! 300: { ! 301: token = tokenizer.GetNextToken(); ! 302: if (!token.empty()) ! 303: { ! 304: arr.Add (token); ! 305: prevEmpty = true; ! 306: continue; ! 307: } ! 308: } ! 309: ! 310: if (token.empty() && !tokenizer.HasMoreTokens()) ! 311: break; ! 312: ! 313: if (prevEmpty || token.empty()) ! 314: { ! 315: if (arr.Count() < 1) ! 316: { ! 317: arr.Add (L""); ! 318: continue; ! 319: } ! 320: arr.Last() += token.empty() ? L',' : token; ! 321: } ! 322: else ! 323: arr.Add (token); ! 324: ! 325: prevEmpty = token.empty(); ! 326: } ! 327: ! 328: make_shared_auto (KeyfileList, keyfileList); ! 329: for (size_t i = 0; i < arr.GetCount(); i++) ! 330: { ! 331: if (!arr[i].empty()) ! 332: keyfileList->push_back (make_shared <Keyfile> (wstring (arr[i]))); ! 333: } ! 334: ! 335: return keyfileList; ! 336: } ! 337: ! 338: VolumeInfoList CommandLineInterface::GetMountedVolumes (const wxString &mountedVolumeSpec) const ! 339: { ! 340: VolumeInfoList volumes = Core->GetMountedVolumes (); ! 341: VolumeInfoList filteredVolumes; ! 342: ! 343: wxFileName pathFilter; ! 344: if (!mountedVolumeSpec.empty()) ! 345: { ! 346: pathFilter = mountedVolumeSpec; ! 347: pathFilter.Normalize (wxPATH_NORM_ABSOLUTE | wxPATH_NORM_DOTS); ! 348: } ! 349: else ! 350: return volumes; ! 351: ! 352: foreach (shared_ptr <VolumeInfo> volume, volumes) ! 353: { ! 354: if (mountedVolumeSpec.empty()) ! 355: { ! 356: filteredVolumes.push_back (volume); ! 357: } ! 358: else if (wxString (volume->Path) == pathFilter.GetFullPath()) ! 359: { ! 360: filteredVolumes.push_back (volume); ! 361: } ! 362: else if (wxString (volume->MountPoint) == pathFilter.GetFullPath() ! 363: || (wxString (volume->MountPoint) + wxFileName::GetPathSeparator()) == pathFilter.GetFullPath()) ! 364: { ! 365: filteredVolumes.push_back (volume); ! 366: } ! 367: } ! 368: ! 369: if (!mountedVolumeSpec.IsEmpty() && filteredVolumes.size() < 1) ! 370: throw_err (_("No such volume is mounted.")); ! 371: ! 372: return filteredVolumes; ! 373: } ! 374: ! 375: auto_ptr <CommandLineInterface> CmdLine; ! 376: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.