fuse_trash.c
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:4k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. #include <hdfs.h>
  19. #include <strings.h>
  20. #include "fuse_dfs.h"
  21. #include "fuse_trash.h"
  22. #include "fuse_context_handle.h"
  23. const char *const TrashPrefixDir = "/user/root/.Trash";
  24. const char *const TrashDir = "/user/root/.Trash/Current";
  25. #define TRASH_RENAME_TRIES  100
  26. //
  27. // NOTE: this function is a c implementation of org.apache.hadoop.fs.Trash.moveToTrash(Path path).
  28. //
  29. int move_to_trash(const char *item, hdfsFS userFS) {
  30.   // retrieve dfs specific data
  31.   dfs_context *dfs = (dfs_context*)fuse_get_context()->private_data;
  32.   // check params and the context var
  33.   assert(item);
  34.   assert(dfs);
  35.   assert('/' == *item);
  36.   assert(rindex(item,'/') >= 0);
  37.   char fname[4096]; // or last element of the directory path
  38.   char parent_directory[4096]; // the directory the fname resides in
  39.   if (strlen(item) > sizeof(fname) - strlen(TrashDir)) {
  40.     syslog(LOG_ERR, "ERROR: internal buffer too small to accomodate path of length %d %s:%dn", (int)strlen(item), __FILE__, __LINE__);
  41.     return -EIO;
  42.   }
  43.   // separate the file name and the parent directory of the item to be deleted
  44.   {
  45.     int length_of_parent_dir = rindex(item, '/') - item ;
  46.     int length_of_fname = strlen(item) - length_of_parent_dir - 1; // the '/'
  47.     // note - the below strncpys should be safe from overflow because of the check on item's string length above.
  48.     strncpy(parent_directory, item, length_of_parent_dir);
  49.     parent_directory[length_of_parent_dir ] = 0;
  50.     strncpy(fname, item + length_of_parent_dir + 1, strlen(item));
  51.     fname[length_of_fname + 1] = 0;
  52.   }
  53.   // create the target trash directory
  54.   char trash_dir[4096];
  55.   if (snprintf(trash_dir, sizeof(trash_dir), "%s%s",TrashDir,parent_directory) >= sizeof trash_dir) {
  56.     syslog(LOG_ERR, "move_to_trash error target is not big enough to hold new name for %s %s:%dn",item, __FILE__, __LINE__);
  57.     return -EIO;
  58.   }
  59.   // create the target trash directory in trash (if needed)
  60.   if ( hdfsExists(userFS, trash_dir)) {
  61.     // make the directory to put it in in the Trash - NOTE
  62.     // hdfsCreateDirectory also creates parents, so Current will be created if it does not exist.
  63.     if (hdfsCreateDirectory(userFS, trash_dir)) {
  64.       return -EIO;
  65.     }
  66.   }
  67.   //
  68.   // if the target path in Trash already exists, then append with
  69.   // a number. Start from 1.
  70.   //
  71.   char target[4096];
  72.   int j ;
  73.   if ( snprintf(target, sizeof target,"%s/%s",trash_dir, fname) >= sizeof target) {
  74.     syslog(LOG_ERR, "move_to_trash error target is not big enough to hold new name for %s %s:%dn",item, __FILE__, __LINE__);
  75.     return -EIO;
  76.   }
  77.   // NOTE: this loop differs from the java version by capping the #of tries
  78.   for (j = 1; ! hdfsExists(userFS, target) && j < TRASH_RENAME_TRIES ; j++) {
  79.     if (snprintf(target, sizeof target,"%s/%s.%d",trash_dir, fname, j) >= sizeof target) {
  80.       syslog(LOG_ERR, "move_to_trash error target is not big enough to hold new name for %s %s:%dn",item, __FILE__, __LINE__);
  81.       return -EIO;
  82.     }
  83.   }
  84.   if (hdfsRename(userFS, item, target)) {
  85.     syslog(LOG_ERR,"ERROR: hdfs trying to rename %s to %s",item, target);
  86.     return -EIO;
  87.   }
  88.   return 0;
  89. int hdfsDeleteWithTrash(hdfsFS userFS, const char *path, int useTrash) {
  90.   // move the file to the trash if this is enabled and its not actually in the trash.
  91.   if (useTrash && strncmp(path, TrashPrefixDir, strlen(TrashPrefixDir)) != 0) {
  92.     int ret= move_to_trash(path, userFS);
  93.     return ret;
  94.   }
  95.   if (hdfsDelete(userFS, path)) {
  96.     syslog(LOG_ERR,"ERROR: hdfs trying to delete the file %s",path);
  97.     return -EIO;
  98.   }
  99.   return 0;
  100. }