Add support for customizable user (#7569)
Harshavardhana
6 years ago
committed by
Nitish Tiwari
parent
1008c2c069
commit
91ceae23d0
@ -0,0 +1,136 @@ |
|||||||
|
// +build ignore
|
||||||
|
|
||||||
|
/* |
||||||
|
* MinIO Cloud Storage, (C) 2019 MinIO, Inc. |
||||||
|
* |
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||||
|
* you may not use this file except in compliance with the License. |
||||||
|
* You may obtain a copy of the License at |
||||||
|
* |
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* |
||||||
|
* Unless required by applicable law or agreed to in writing, software |
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
* See the License for the specific language governing permissions and |
||||||
|
* limitations under the License. |
||||||
|
*/ |
||||||
|
|
||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"log" |
||||||
|
"os" |
||||||
|
"os/exec" |
||||||
|
"os/user" |
||||||
|
"syscall" |
||||||
|
|
||||||
|
"github.com/minio/cli" |
||||||
|
minio "github.com/minio/minio/cmd" |
||||||
|
) |
||||||
|
|
||||||
|
var defaultUserGroup string |
||||||
|
|
||||||
|
func init() { |
||||||
|
username := os.Getenv("MINIO_USERNAME") |
||||||
|
groupname := os.Getenv("MINIO_GROUPNAME") |
||||||
|
defaultUserGroup = username + ":" + groupname |
||||||
|
} |
||||||
|
|
||||||
|
func getUserGroup(path string) (string, error) { |
||||||
|
fi, err := os.Stat(minio.PathJoin(path, ".minio.sys")) |
||||||
|
if err != nil { |
||||||
|
// Fresh directory we should default to what was requested by user.
|
||||||
|
if os.IsNotExist(err) { |
||||||
|
cmd := exec.Command("chown", "-R", defaultUserGroup, path) |
||||||
|
if err = cmd.Run(); err != nil { |
||||||
|
return "", err |
||||||
|
} |
||||||
|
return defaultUserGroup, nil |
||||||
|
} |
||||||
|
return "", err |
||||||
|
} |
||||||
|
stat, ok := fi.Sys().(*syscall.Stat_t) |
||||||
|
if !ok { |
||||||
|
// Unable to figure out uid/gid, default to defaultUserGroup
|
||||||
|
return defaultUserGroup, nil |
||||||
|
} |
||||||
|
u, err := user.LookupId(fmt.Sprintf("%d", stat.Uid)) |
||||||
|
if err != nil { |
||||||
|
return fmt.Sprintf("%d:%d", stat.Uid, stat.Gid), nil |
||||||
|
} |
||||||
|
g, err := user.LookupGroupId(fmt.Sprintf("%d", stat.Gid)) |
||||||
|
if err != nil { |
||||||
|
return fmt.Sprintf("%d:%d", stat.Uid, stat.Gid), nil |
||||||
|
} |
||||||
|
return fmt.Sprintf("%s:%s", u.Username, g.Name), nil |
||||||
|
} |
||||||
|
|
||||||
|
func main() { |
||||||
|
app := cli.NewApp() |
||||||
|
app.Flags = append(minio.ServerFlags, minio.GlobalFlags...) |
||||||
|
app.Action = func(ctx *cli.Context) { |
||||||
|
// Fetch address option
|
||||||
|
serverAddr := ctx.GlobalString("address") |
||||||
|
if serverAddr == "" || serverAddr == ":9000" { |
||||||
|
serverAddr = ctx.String("address") |
||||||
|
} |
||||||
|
if ctx.Args().First() == "help" { |
||||||
|
cli.ShowCommandHelpAndExit(ctx, "check-user", 1) |
||||||
|
} |
||||||
|
if ctx.Args().First() != "minio" { |
||||||
|
cli.ShowCommandHelpAndExit(ctx, "check-user", 1) |
||||||
|
} |
||||||
|
args := cli.Args(ctx.Args().Tail()) |
||||||
|
if !args.Present() { |
||||||
|
cli.ShowCommandHelpAndExit(ctx, "check-user", 1) |
||||||
|
} |
||||||
|
var ug string |
||||||
|
var err error |
||||||
|
switch args.First() { |
||||||
|
case "gateway": |
||||||
|
args = cli.Args(args.Tail()) |
||||||
|
if args.First() != "nas" { |
||||||
|
fmt.Println(defaultUserGroup) |
||||||
|
return |
||||||
|
} |
||||||
|
args = cli.Args(args.Tail()) |
||||||
|
if args.First() == "" { |
||||||
|
fmt.Println("") |
||||||
|
return |
||||||
|
} |
||||||
|
ug, err = getUserGroup(args.First()) |
||||||
|
if err != nil { |
||||||
|
log.Fatalln(err) |
||||||
|
} |
||||||
|
case "server": |
||||||
|
var setArgs [][]string |
||||||
|
setArgs, err = minio.GetAllSets(args.Tail()...) |
||||||
|
if err != nil { |
||||||
|
log.Fatalln(err) |
||||||
|
} |
||||||
|
var endpoints minio.EndpointList |
||||||
|
_, endpoints, _, err = minio.CreateEndpoints(serverAddr, setArgs...) |
||||||
|
if err != nil { |
||||||
|
log.Fatalln(err) |
||||||
|
} |
||||||
|
for _, endpoint := range endpoints { |
||||||
|
if !endpoint.IsLocal { |
||||||
|
continue |
||||||
|
} |
||||||
|
ug, err = getUserGroup(endpoint.Path) |
||||||
|
if err != nil { |
||||||
|
log.Fatalln(err) |
||||||
|
} |
||||||
|
break |
||||||
|
} |
||||||
|
default: |
||||||
|
cli.ShowCommandHelpAndExit(ctx, "check-user", 1) |
||||||
|
} |
||||||
|
fmt.Println(ug) |
||||||
|
} |
||||||
|
if err := app.Run(os.Args); err != nil { |
||||||
|
log.Fatalln(err) |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue