mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 16:40:24 +08:00 
			
		
		
		
	The dump cmd did not check the return value of the z.AddFile or z.AddDir when building the final archive. It caused the dump command to succeed even if an error occurred. The resulted dump archive could be corrupted/empty. (errors could be various: removal by a concurrent process, disk full, bugs in the dump cmd itself)
		
			
				
	
	
		
			83 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			83 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2014 The Gogs Authors. All rights reserved.
 | 
						|
// Use of this source code is governed by a MIT-style
 | 
						|
// license that can be found in the LICENSE file.
 | 
						|
 | 
						|
package cmd
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"log"
 | 
						|
	"os"
 | 
						|
	"path"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"github.com/Unknwon/cae/zip"
 | 
						|
	"github.com/codegangsta/cli"
 | 
						|
 | 
						|
	"github.com/gogits/gogs/models"
 | 
						|
	"github.com/gogits/gogs/modules/setting"
 | 
						|
)
 | 
						|
 | 
						|
var CmdDump = cli.Command{
 | 
						|
	Name:  "dump",
 | 
						|
	Usage: "Dump Gogs files and database",
 | 
						|
	Description: `Dump compresses all related files and database into zip file.
 | 
						|
It can be used for backup and capture Gogs server image to send to maintainer`,
 | 
						|
	Action: runDump,
 | 
						|
	Flags: []cli.Flag{
 | 
						|
		stringFlag("config, c", "custom/conf/app.ini", "Custom configuration file path"),
 | 
						|
		boolFlag("verbose, v", "show process details"),
 | 
						|
	},
 | 
						|
}
 | 
						|
 | 
						|
func runDump(ctx *cli.Context) {
 | 
						|
	if ctx.IsSet("config") {
 | 
						|
		setting.CustomConf = ctx.String("config")
 | 
						|
	}
 | 
						|
	setting.NewContext()
 | 
						|
	models.LoadConfigs()
 | 
						|
	models.SetEngine()
 | 
						|
 | 
						|
	log.Printf("Dumping local repositories...%s", setting.RepoRootPath)
 | 
						|
	zip.Verbose = ctx.Bool("verbose")
 | 
						|
	defer os.Remove("gogs-repo.zip")
 | 
						|
	if err := zip.PackTo(setting.RepoRootPath, "gogs-repo.zip", true); err != nil {
 | 
						|
		log.Fatalf("Fail to dump local repositories: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	log.Printf("Dumping database...")
 | 
						|
	defer os.Remove("gogs-db.sql")
 | 
						|
	if err := models.DumpDatabase("gogs-db.sql"); err != nil {
 | 
						|
		log.Fatalf("Fail to dump database: %v", err)
 | 
						|
	}
 | 
						|
 | 
						|
	fileName := fmt.Sprintf("gogs-dump-%d.zip", time.Now().Unix())
 | 
						|
	log.Printf("Packing dump files...")
 | 
						|
	z, err := zip.Create(fileName)
 | 
						|
	if err != nil {
 | 
						|
		os.Remove(fileName)
 | 
						|
		log.Fatalf("Fail to create %s: %v", fileName, err)
 | 
						|
	}
 | 
						|
 | 
						|
	workDir, _ := setting.WorkDir()
 | 
						|
	if err := z.AddFile("gogs-repo.zip", path.Join(workDir, "gogs-repo.zip")); err !=nil {
 | 
						|
		log.Fatalf("Fail to include gogs-repo.zip: %v", err)
 | 
						|
	}
 | 
						|
	if err := z.AddFile("gogs-db.sql", path.Join(workDir, "gogs-db.sql")); err !=nil {
 | 
						|
		log.Fatalf("Fail to include gogs-db.sql: %v", err)
 | 
						|
	}
 | 
						|
	if err := z.AddDir("custom", path.Join(workDir, "custom")); err !=nil {
 | 
						|
		log.Fatalf("Fail to include custom: %v", err)
 | 
						|
	}
 | 
						|
	if err := z.AddDir("log", path.Join(workDir, "log")); err !=nil {
 | 
						|
		log.Fatalf("Fail to include log: %v", err)
 | 
						|
	}
 | 
						|
	// FIXME: SSH key file.
 | 
						|
	if err = z.Close(); err != nil {
 | 
						|
		os.Remove(fileName)
 | 
						|
		log.Fatalf("Fail to save %s: %v", fileName, err)
 | 
						|
	}
 | 
						|
 | 
						|
	log.Println("Finish dumping!")
 | 
						|
}
 |