开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Ftp客户端 > 查看源码
chmod.cpp
资源名称：warftpd.zip　[点击查看]
上传用户：surprise9
上传日期：2007-01-04
资源大小：426k
文件大小：9k
源码类别：

Ftp客户端

开发平台：

Visual C++

chmod.cpp：源码内容
  1. // This is part of the WAR SOFTWARE SERIES initiated by Jarle Aase
  2. // Copyright 1996 by Jarle Aase. All rights reserved.
  3. // See the "War Software Series Licende Agreement" for details concerning 
  4. // use and distribution.
  5. // ---
  6. // This source code, executables and programs containing source code or
  7. // binaries or proprietetary technology from the War Software Series are
  8. // NOT alloed used, viewed or tested by any governmental agencies in
  9. // any countries. This includes the government, departments, police, 
  10. // military etc.
  11. // ---
  12. // This file is intended for use with Tab space = 2
  13. // Created and maintained in MSVC Developer Studio
  14. // ---
  15. // NAME : chmod.cpp
  16. // PURPOSE : extended UNIX chmod command
  17. // PROGRAM : 
  18. // DATE : Oct. 13 1996
  19. // AUTHOR : Jarle Aase
  20. // ---
  21. // REVISION HISTORY
  22. // 
  24. #include "stdafx.h"
  25. // For now, we don't support non-daemon programs
  26. //#include "WarSoftware.h"
  27. #include "WarDaemon.h"
  28. #include "WarFsys.h"
  29. #include "VfFSys.h"
  30. #include "UnixFsysTypes.h"
  32. #ifdef _DEBUG
  33. #define new DEBUG_NEW
  34. #undef THIS_FILE
  35. static char THIS_FILE[] = __FILE__;
  36. #endif
  38. CChmod::CChmod(CFsysThread *pFsysThread)
  39. {
  40. set = NULL;
  41. m_Owner = -1;
  42. m_Group = -1;
  43. m_Comment = NULL;
  44. m_DefDirPerms = -1;
  45. m_DefFilePerms = -1;
  46. omode = -1;
  47. m_SetMode = FALSE;
  48. m_pFsysThread = pFsysThread;
  49. m_SpecialNewFilePerm = -1;
  50. }
  52. CChmod::~CChmod()
  53. {
  54. if (set)
  55. free(set);
  56. }
  59. int CChmod::main(int argc, char **argv)
  60. {
  61. char *ep, *mode;
  62. BOOL Alternate = FALSE;
  63. CFileList MasterFileList;
  64. int Rval;
  66. set = NULL;
  67. omode = 0;
  68. Hflag = Lflag = Pflag = Rflag = fflag = hflag = 0;
  70. while ((ch = getopt(argc, argv, "N:O:G:C:D:F:HLPRXfgorstuwx")) != EOF)
  71. switch (ch) 
  72. {
  73. case 'N': // SpecialNewFilePerm (octal)
  74. if (!optarg || !*optarg)
  75. goto badarg;
  76. sscanf(optarg,"%o", &m_SpecialNewFilePerm);
  77. Alternate = TRUE;
  78. break;
  80. case 'O': // chown
  81. if (!optarg || !*optarg)
  82. goto badarg;
  83. if (atoi(optarg))
  84. m_Owner = atoi(optarg);
  85. else if ((m_Owner = CUsr::FindUser(UT_USER, optarg)) == 0)
  86. {
  87. printf("Unknown user name '%s'n", optarg);
  88. return -1;
  89. }
  91. Alternate = TRUE;
  92. break;
  94. case 'G': // chown
  95. if (!optarg || !*optarg)
  96. goto badarg;
  97. if (!optarg || !*optarg)
  98. goto badarg;
  99. if (atoi(optarg))
  100. m_Group = atoi(optarg);
  101. else if ((m_Group = CUsr::FindUser(UT_CLASS, optarg)) == 0)
  102. {
  103. printf("Unknown class name '%s'n", optarg);
  104. return -1;
  105. }
  106. Alternate = TRUE;
  107. break;
  109. case 'C': // change comment
  110. if (!optarg)
  111. goto badarg;
  112. m_Comment = optarg;
  113. Alternate = TRUE;
  114. break;
  115. case 'D':
  116. if (!optarg || !*optarg || (*optarg < '0') || (*optarg > '8'))
  117. goto badarg;
  118. m_DefDirPerms = strtol(optarg, &ep, 8);
  119. Alternate = TRUE;
  120. break;
  121. case 'F':
  122. if (!optarg || !*optarg || (*optarg < '0') || (*optarg > '8'))
  123. goto badarg;
  124. m_DefFilePerms = strtol(optarg, &ep, 8);
  125. Alternate = TRUE;
  126. break;
  127. case 'H':
  128. Hflag = 1;
  129. Lflag = Pflag = 0;
  130. break;
  131. case 'L':
  132. Lflag = 1;
  133. Hflag = Pflag = 0;
  134. break;
  135. case 'P':
  136. Pflag = 1;
  137. Hflag = Lflag = 0;
  138. break;
  139. case 'R':
  140. Rflag = 1;
  141. break;
  142. case 'f': /* XXX: undocumented. */
  143. fflag = 1;
  144. break;
  145. case 'h':
  146. /*
  147.  * In System V (and probably POSIX.2) the -h option
  148.  * causes chmod to change the mode of the symbolic
  149.  * link.  4.4BSD's symbolic links don't have modes,
  150.  * so it's an undocumented noop.  Do syntax checking,
  151.  * though.
  152.  */
  153. hflag = 1;
  154. break;
  155. /*
  156.  * XXX
  157.  * "-[rwx]" are valid mode commands.  If they are the entire
  158.  * argument, getopt has moved past them, so decrement optind.
  159.  * Regardless, we're done argument processing.
  160.  */
  161. case 'g': case 'o': case 'r': case 's':
  162. case 't': case 'u': case 'w': case 'X': case 'x':
  163. if (argv[optind - 1][0] == '-' &&
  164.     argv[optind - 1][1] == ch &&
  165.     argv[optind - 1][2] == '')
  166. --optind;
  167. Alternate = FALSE;
  168. goto done;
  169. case '?':
  170. default:
  171. return usage();
  172. }
  173. done:
  174. argv += optind;
  175. argc -= optind;
  177. if (Rflag) 
  178. {
  179. if (hflag)
  180. {
  181. printf("the -R and -h options may not be specified together.");
  182. return usage();
  183. }
  184. }
  186. if (!Alternate)
  187. {
  188. if (argc < 2)
  189. return usage();
  191. m_SetMode = TRUE;
  192. mode = *argv;
  193. --argc;
  194. ++argv;
  195. if (*mode >= '0' && *mode <= '7') 
  196. {
  197. val = strtol(mode, &ep, 8);
  198. if (val > INT_MAX || val < 0)
  199. {
  200. printf("invalid file mode: %s", mode);
  201. return 1;
  202. }
  203. if (*ep)
  204. {
  205. printf("invalid file mode: %s", mode);
  206. return 1;
  207. }
  208. omode = val;
  209. oct = 1;
  210. } 
  211. else 
  212. {
  213. if ((set = setmode(mode)) == NULL)
  214. {
  215. printf("invalid file mode: %s", mode);
  216. return 1;
  217. }
  218. oct = 0;
  219. }
  220. }
  222. pLock->WriteLock();
  223. Rval = Process(argc, argv, 0, MasterFileList);
  224. pLock->UnlockWrite();
  225. return Rval;
  227. badarg:
  228. return usage();
  229. }
  231. int CChmod::Process(int argc, char **argv, int Level, CFileList& MasterFileList)
  232. {
  233. int argcSave = argc;
  234. char **argvSave = argv;
  236. for(;argc ; argc--, argv++)
  237. {
  238. // Prepere file system
  239. CString Path = *argv;
  240. m_pFsysThread->ParsePath(Path);
  241. m_pFsysThread->ScanPath(Path, FALSE, Rflag, FALSE, FALSE);
  242. }
  244. int Rval = _Process(argcSave, argvSave, Level, MasterFileList);
  246. return Rval;
  247. }
  249. int CChmod::_Process(int argc, char **argv, int Level, CFileList& MasterFileList)
  250. {
  251. CString cShowPath, cCurrentPath;
  252. CLinkedListItem *Item;
  254. ASSERT(argc >= 1);
  256. for(;argc ; argc--, argv++)
  257. {
  258. CFileList FileList;
  259. ASSERT(AfxIsValidString(*argv));
  261. if (m_VisitedDirs.Find(*argv))
  262. continue; // Already been here
  264. cCurrentPath = cShowPath = *argv;
  265. m_pFsysThread->BldFileInfoList(FileList, *argv, Rflag);
  267. if (Rflag)
  268. {
  269. m_VisitedDirs.AddHead(*argv);
  271. int Rval;
  272. for(Item = FileList.First(); Item; Item = FileList.Next(Item))
  273. {
  274. FILE_NODE *pNode = (FILE_NODE *)FileList.Ptr(Item);
  275. if ((pNode->Perms & NODE_DIR) 
  276. && strcmp(pNode->Name,".") && strcmp(pNode->Name,".."))
  277. {
  278. CString cBuf;
  280. // We need to modify the current path
  281. LPSTR p = cCurrentPath.GetBuffer(1);
  282. LPSTR pp = strrchr(p, '/');
  283. if (pp)
  284. *pp = 0;
  285. else
  286. *p = 0;
  287. cCurrentPath.ReleaseBuffer();
  290. if (cCurrentPath.IsEmpty())
  291. cBuf = pNode->Name;
  292. else
  293. cBuf.Format("%s/%s", cCurrentPath, pNode->Name);
  295. cBuf += "/*";
  297. LPSTR MyArgs[1];
  298. MyArgs[0] = cBuf.GetBuffer(1);
  300. if (Rval = Process(1, MyArgs, Level +1, MasterFileList))
  301. return Rval; // Error
  302. }
  303. }
  304. }
  305. MasterFileList += FileList;
  306. }
  308. if (!Level)
  309. {
  310. if (!ChmodFileList(MasterFileList, cShowPath))
  311. return -1;
  313. // Flush
  314. for(Item = MasterFileList.First(); Item; Item = MasterFileList.Next(Item))
  315. {
  316. FILE_NODE *pNode = (FILE_NODE *)MasterFileList.Ptr(Item);
  317. m_pFsysThread->Flush(pNode->Father);
  318. }
  319. }
  321. return 0;
  322. }
  325. BOOL CChmod::ChmodFileList(CFileList& FileList, LPCSTR Path)
  326. {
  327. FILE_NODE *pNode;
  328. LPCSTR FileName;
  329. CLinkedListItem *Item;
  331. for(Item = FileList.First(); Item; Item = FileList.Next(Item))
  332. {
  333. pNode = (FILE_NODE *)FileList.Ptr(Item);
  334. FileName = pNode->Name;
  335. ASSERT(AfxIsValidString(FileName));
  337. if (!strcmp(FileName,".") || !strcmp(FileName,".."))
  338. continue;
  340. // Change the mode
  341. if (__chmod(
  342. pNode, 
  343. m_SetMode ? oct ? omode : getmode(set,pNode->Perms) : -1,
  344. m_Owner, m_Group, m_Comment, m_DefDirPerms, m_DefFilePerms, m_SpecialNewFilePerm))
  345. {
  346. printf("chmod: %s - mode change failed %s/%sn", Path, FileName, GetLastErrorText());
  347. }
  348. }
  349. return TRUE;
  350. }
  353. int CChmod::usage()
  354. {
  355. printf("usage: chmod [-R [-H | -L | -P]] mode file ...n");
  356. printf("  or   chmod [-O Owner] [-G Class] [-C "Comment"] [-D DefDirPerms] [-F DefFilePerms] file ...n");
  357. printf("  or   chmod -N SpecialNewFilePerms directory ... (value is octal, default is 022)n");
  358. return 1;
  359. }
  362. // Chmod on a single file
  363. // Return 0 on success
  364. int CChmod::__chmod(FILE_NODE *pNode, int SetMode, int SetOwner, 
  365. int SetClass, LPCSTR SetComment, int DefDirMode, int DefFileMode,
  366. int SpecialNewFilePerm)
  367. {
  368. int rval = 0;
  369. DWORD PreserveFlags = pNode->Perms & ~NODE_SETTABLE;
  371. // Check permission
  372. if (!m_Fsys->m_IsAdmin && (m_Fsys->m_User != pNode->User))
  373. {
  374. SetLastError(ERROR_ACCESS_DENIED);
  375. return -1;
  376. }
  378. // Set the new permissions
  379. if (SetMode != -1)
  380. pNode->Perms = (SetMode & NODE_SETTABLE) | PreserveFlags;
  382. if (SetOwner != -1)
  383. pNode->User = SetOwner;
  385. if (SetClass != -1)
  386. pNode->Class = SetClass;
  388. if ((SpecialNewFilePerm != -1) && (pNode->Perms & NODE_DIR))
  389. ((DIR_NODE *)pNode)->SpecialNewFilePerm = SpecialNewFilePerm;
  391. if (SetComment)
  392. ChangeNodeInformation(pNode, NULL, NULL, NULL, SetComment);
  394. DIR_NODE *pDir = (DIR_NODE *)pSmem->__Ptr(pNode->Father);
  395. if (pDir)
  396. pDir->f_Dirty = TRUE;
  398. return 0;
  399. }