Implement adjusting file permissions for shared repositories.

Florian Zschocke

2013-08-20 e85277e1de9f59ac45df5ffd84c5d9be0e4d20d2

 src/main/java/com/gitblit/utils/JGitUtils.java

@@ -32,6 +32,7 @@
import java.util.Map.Entry;

import java.util.regex.Pattern;



import org.apache.commons.io.filefilter.TrueFileFilter;

import org.eclipse.jgit.api.CloneCommand;

import org.eclipse.jgit.api.FetchCommand;

import org.eclipse.jgit.api.Git;

@@ -58,6 +59,7 @@
import org.eclipse.jgit.lib.RefUpdate.Result;

import org.eclipse.jgit.lib.Repository;

import org.eclipse.jgit.lib.RepositoryCache.FileKey;

import org.eclipse.jgit.lib.StoredConfig;

import org.eclipse.jgit.lib.TreeFormatter;

import org.eclipse.jgit.revwalk.RevBlob;

import org.eclipse.jgit.revwalk.RevCommit;

@@ -259,14 +261,152 @@
    * @return Repository

    */

   public static Repository createRepository(File repositoriesFolder, String name) {

      return createRepository(repositoriesFolder, name, "FALSE");

   }



   /**

    * Creates a bare, shared repository.

    * 
    * @param repositoriesFolder

    * @param name

    * @param shared

    *          the setting for the --shared option of "git init".

    * @return Repository

    */

   public static Repository createRepository(File repositoriesFolder, String name, String shared) {

      try {

         Git git = Git.init().setDirectory(new File(repositoriesFolder, name)).setBare(true).call();

         return git.getRepository();

      } catch (GitAPIException e) {

         Repository repo = null;

         try {

            Git git = Git.init().setDirectory(new File(repositoriesFolder, name)).setBare(true).call();

            repo = git.getRepository();

         } catch (GitAPIException e) {

            throw new RuntimeException(e);

         }



         GitConfigSharedRepository sharedRepository = new GitConfigSharedRepository(shared);

         if (sharedRepository.isShared()) {

            StoredConfig config = repo.getConfig();

            config.setString("core", null, "sharedRepository", sharedRepository.getValue());

            config.setBoolean("receive", null, "denyNonFastforwards", true);

            config.save();



            if (! JnaUtils.isWindows()) {

               Iterator<File> iter = org.apache.commons.io.FileUtils.iterateFilesAndDirs(repo.getDirectory(),

                     TrueFileFilter.INSTANCE, TrueFileFilter.INSTANCE);

               // Adjust permissions on file/directory

               while (iter.hasNext()) {

                  adjustSharedPerm(iter.next(), sharedRepository);

               }

            }

         }



         return repo;

      } catch (IOException e) {

         throw new RuntimeException(e);

      }

   }



   private enum GitConfigSharedRepositoryValue

   {

      UMASK("0", 0), FALSE("0", 0), OFF("0", 0), NO("0", 0),

      GROUP("1", 0660), TRUE("1", 0660), ON("1", 0660), YES("1", 0660),

      ALL("2", 0664), WORLD("2", 0664), EVERYBODY("2", 0664),

      Oxxx(null, -1);



      private String configValue;

      private int permValue;

      private GitConfigSharedRepositoryValue(String config, int perm) { configValue = config; permValue = perm; };



      public String getConfigValue() { return configValue; };

      public int getPerm() { return permValue; };



   }



   private static class GitConfigSharedRepository

   {

      private int intValue;

      private GitConfigSharedRepositoryValue enumValue;



      GitConfigSharedRepository(String s) {

         if ( s == null || s.trim().isEmpty() ) {

            enumValue = GitConfigSharedRepositoryValue.GROUP;

         }

         else {

            try {

               // Try one of the string values

               enumValue = GitConfigSharedRepositoryValue.valueOf(s.trim().toUpperCase());

            } catch (IllegalArgumentException  iae) {

               try {

                  // Try if this is an octal number

                  int i = Integer.parseInt(s, 8);

                  if ( (i & 0600) != 0600 ) {

                     String msg = String.format("Problem with core.sharedRepository filemode value (0%03o).\nThe owner of files must always have read and write permissions.", i);

                     throw new IllegalArgumentException(msg);

                  }

                  intValue = i & 0666;

                  enumValue = GitConfigSharedRepositoryValue.Oxxx;

               } catch (NumberFormatException nfe) {

                  throw new IllegalArgumentException("Bad configuration value for 'shared': '" + s + "'");

               }

            }

         }

      }



      String getValue() {

         if ( enumValue == GitConfigSharedRepositoryValue.Oxxx ) return Integer.toOctalString(intValue);

         return enumValue.getConfigValue();

      }



      int getPerm() {

         if ( enumValue == GitConfigSharedRepositoryValue.Oxxx ) return intValue;

         return enumValue.getPerm();

      }



      boolean isCustom() {

         return enumValue == GitConfigSharedRepositoryValue.Oxxx;

      }



      boolean isShared() {

         return (enumValue.getPerm() > 0) || enumValue == GitConfigSharedRepositoryValue.Oxxx;

      }

   }





   public static int adjustSharedPerm(File path, String configShared) {

      return adjustSharedPerm(path, new GitConfigSharedRepository(configShared));

   }





   public static int adjustSharedPerm(File path, GitConfigSharedRepository configShared) {

      if (! configShared.isShared()) return 0;



      int perm = configShared.getPerm();

      int mode = JnaUtils.getFilemode(path);

      if (mode < 0) return -1;



      // If the owner has no write access, delete it from group and other, too.

      if ((mode & JnaUtils.S_IWUSR) == 0) perm &= ~0222;

      // If the owner has execute access, set it for all blocks that have read access.

      if ((mode & JnaUtils.S_IXUSR) == JnaUtils.S_IXUSR) perm |= (perm & 0444) >> 2;



      if (configShared.isCustom()) {

         // Use the custom value for access permissions.

         mode |= (mode & ~0777) | perm;

      }

      else {

         // Just add necessary bits to existing permissions.

         mode |= perm;

      }



      if (path.isDirectory()) {

         mode |= (mode & 0444) >> 2;

         mode |= JnaUtils.S_ISGID;

      }



      return JnaUtils.setFilemode(path, mode);

   }





   /**

    * Returns a list of repository names in the specified folder.

    *