package repo import ( "bytes" "fmt" "io" "io/ioutil" "log" "net/http" "os" "os/exec" "path" "path/filepath" "regexp" "strconv" "strings" "time" "github.com/go-martini/martini" "github.com/gogits/gogs/models" "github.com/gogits/gogs/modules/base" "github.com/gogits/gogs/modules/middleware" ) func Http(ctx *middleware.Context, params martini.Params) { username := params["username"] reponame := params["reponame"] if strings.HasSuffix(reponame, ".git") { reponame = reponame[:len(reponame)-4] } var isPull bool service := ctx.Query("service") if service == "git-receive-pack" || strings.HasSuffix(ctx.Req.URL.Path, "git-receive-pack") { isPull = false } else if service == "git-upload-pack" || strings.HasSuffix(ctx.Req.URL.Path, "git-upload-pack") { isPull = true } else { isPull = (ctx.Req.Method == "GET") } repoUser, err := models.GetUserByName(username) if err != nil { ctx.Handle(500, "repo.GetUserByName", nil) return } repo, err := models.GetRepositoryByName(repoUser.Id, reponame) if err != nil { ctx.Handle(500, "repo.GetRepositoryByName", nil) return } // only public pull don't need auth var askAuth = !(!repo.IsPrivate && isPull) var authUser *models.User // check access if askAuth { baHead := ctx.Req.Header.Get("Authorization") if baHead == "" { // ask auth authRequired(ctx) return } auths := strings.Fields(baHead) // currently check basic auth // TODO: support digit auth if len(auths) != 2 || auths[0] != "Basic" { ctx.Handle(401, "no basic auth and digit auth", nil) return } authUsername, passwd, err := basicDecode(auths[1]) if err != nil { ctx.Handle(401, "no basic auth and digit auth", nil) return } authUser, err = models.GetUserByName(authUsername) if err != nil { ctx.Handle(401, "no basic auth and digit auth", nil) return } newUser := &models.User{Passwd: passwd, Salt: authUser.Salt} newUser.EncodePasswd() if authUser.Passwd != newUser.Passwd { ctx.Handle(401, "no basic auth and digit auth", nil) return } var tp = models.AU_WRITABLE if isPull { tp = models.AU_READABLE } has, err := models.HasAccess(authUsername, username+"/"+reponame, tp) if err != nil { ctx.Handle(401, "no basic auth and digit auth", nil) return } else if !has { if tp == models.AU_READABLE { has, err = models.HasAccess(authUsername, username+"/"+reponame, models.AU_WRITABLE) if err != nil || !has { ctx.Handle(401, "no basic auth and digit auth", nil) return } } else { ctx.Handle(401, "no basic auth and digit auth", nil) return } } } config := Config{base.RepoRootPath, "git", true, true, func(rpc string, input []byte) { if rpc == "receive-pack" { firstLine := bytes.IndexRune(input, '\n') if firstLine > -1 { fields := strings.Fields(string(input[:firstLine])) if len(fields) > 3 { oldCommitId := fields[0][4:] newCommitId := fields[1] refName := fields[2] models.Update(refName, oldCommitId, newCommitId, username, reponame, authUser.Id) } } } }} handler := HttpBackend(&config) handler(ctx.ResponseWriter, ctx.Req) /* Webdav dir := models.RepoPath(username, reponame) prefix := path.Join("/", username, params["reponame"]) server := webdav.NewServer( dir, prefix, true) server.ServeHTTP(ctx.ResponseWriter, ctx.Req) */ } type route struct { cr *regexp.Regexp method string handler func(handler) } type Config struct { ReposRoot string GitBinPath string UploadPack bool ReceivePack bool OnSucceed func(rpc string, input []byte) } type handler struct { *Config w http.ResponseWriter r *http.Request Dir string File string } var routes = []route{ {regexp.MustCompile("(.*?)/git-upload-pack$"), "POST", serviceUploadPack}, {regexp.MustCompile("(.*?)/git-receive-pack$"), "POST", serviceReceivePack}, {regexp.MustCompile("(.*?)/info/refs$"), "GET", getInfoRefs}, {regexp.MustCompile("(.*?)/HEAD$"), "GET", getTextFile}, {regexp.MustCompile("(.*?)/objects/info/alternates$"), "GET", getTextFile}, {regexp.MustCompile("(.*?)/objects/info/http-alternates$"), "GET", getTextFile}, {regexp.MustCompile("(.*?)/objects/info/packs$"), "GET", getInfoPacks}, {regexp.MustCompile("(.*?)/objects/info/[^/]*$"), "GET", getTextFile}, {regexp.MustCompile("(.*?)/objects/[0-9a-f]{2}/[0-9a-f]{38}$"), "GET", getLooseObject}, {regexp.MustCompile("(.*?)/objects/pack/pack-[0-9a-f]{40}\\.pack$"), "GET", getPackFile}, {regexp.MustCompile("(.*?)/objects/pack/pack-[0-9a-f]{40}\\.idx$"), "GET", getIdxFile}, } // Request handling function func HttpBackend(config *Config) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { //log.Printf("%s %s %s %s", r.RemoteAddr, r.Method, r.URL.Path, r.Proto) for _, route := range routes { if m := route.cr.FindStringSubmatch(r.URL.Path); m != nil { if route.method != r.Method { renderMethodNotAllowed(w, r) return } file := strings.Replace(r.URL.Path, m[1]+"/", "", 1) dir, err := getGitDir(config, m[1]) if err != nil { log.Print(err) renderNotFound(w) return } hr := handler{config, w, r, dir, file} route.handler(hr) return } } renderNotFound(w) return } } // Actual command handling functions func serviceUploadPack(hr handler) { serviceRpc("upload-pack", hr) } func serviceReceivePack(hr handler) { serviceRpc("receive-pack", hr) } func serviceRpc(rpc string, hr handler) { w, r, dir := hr.w, hr.r, hr.Dir access := hasAccess(r, hr.Config, dir, rpc, true) if access == false { renderNoAccess(w) return } input, _ := ioutil.ReadAll(r.Body) w.Header().Set("Content-Type", fmt.Sprintf("application/x-git-%s-result", rpc)) w.WriteHeader(http.StatusOK) args := []string{rpc, "--stateless-rpc", dir} cmd := exec.Command(hr.Config.GitBinPath, args...) cmd.Dir = dir in, err := cmd.StdinPipe() if err != nil { log.Print(err) return } stdout, err := cmd.StdoutPipe() if err != nil { log.Print(err) return } err = cmd.Start() if err != nil { log.Print(err) return } in.Write(input) io.Copy(w, stdout) cmd.Wait() if hr.Config.OnSucceed != nil { hr.Config.OnSucceed(rpc, input) } } func getInfoRefs(hr handler) { w, r, dir := hr.w, hr.r, hr.Dir serviceName := getServiceType(r) access := hasAccess(r, hr.Config, dir, serviceName, false) if access { args := []string{serviceName, "--stateless-rpc", "--advertise-refs", "."} refs := gitCommand(hr.Config.GitBinPath, dir, args...) hdrNocache(w) w.Header().Set("Content-Type", fmt.Sprintf("application/x-git-%s-advertisement", serviceName)) w.WriteHeader(http.StatusOK) w.Write(packetWrite("# service=git-" + serviceName + "\n")) w.Write(packetFlush()) w.Write(refs) } else { updateServerInfo(hr.Config.GitBinPath, dir) hdrNocache(w) sendFile("text/plain; charset=utf-8", hr) } } func getInfoPacks(hr handler) { hdrCacheForever(hr.w) sendFile("text/plain; charset=utf-8", hr) } func getLooseObject(hr handler) { hdrCacheForever(hr.w) sendFile("application/x-git-loose-object", hr) } func getPackFile(hr handler) { hdrCacheForever(hr.w) sendFile("application/x-git-packed-objects", hr) } func getIdxFile(hr handler) { hdrCacheForever(hr.w) sendFile("application/x-git-packed-objects-toc", hr) } func getTextFile(hr handler) { hdrNocache(hr.w) sendFile("text/plain", hr) } // Logic helping functions func sendFile(contentType string, hr handler) { w, r := hr.w, hr.r reqFile := path.Join(hr.Dir, hr.File) //fmt.Println("sendFile:", reqFile) f, err := os.Stat(reqFile) if os.IsNotExist(err) { renderNotFound(w) return } w.Header().Set("Content-Type", contentType) w.Header().Set("Content-Length", fmt.Sprintf("%d", f.Size())) w.Header().Set("Last-Modified", f.ModTime().Format(http.TimeFormat)) http.ServeFile(w, r, reqFile) } func getGitDir(config *Config, fPath string) (string, error) { root := config.ReposRoot if root == "" { cwd, err := os.Getwd() if err != nil { log.Print(err) return "", err } root = cwd } if !strings.HasSuffix(fPath, ".git") { fPath = fPath + ".git" } f := filepath.Join(root, fPath) if _, err := os.Stat(f); os.IsNotExist(err) { return "", err } return f, nil } func getServiceType(r *http.Request) string { serviceType := r.FormValue("service") if s := strings.HasPrefix(serviceType, "git-"); !s { return "" } return strings.Replace(serviceType, "git-", "", 1) } func hasAccess(r *http.Request, config *Config, dir string, rpc string, checkContentType bool) bool { if checkContentType { if r.Header.Get("Content-Type") != fmt.Sprintf("application/x-git-%s-request", rpc) { return false } } if !(rpc == "upload-pack" || rpc == "receive-pack") { return false } if rpc == "receive-pack" { return config.ReceivePack } if rpc == "upload-pack" { return config.UploadPack } return getConfigSetting(config.GitBinPath, rpc, dir) } func getConfigSetting(gitBinPath, serviceName string, dir string) bool { serviceName = strings.Replace(serviceName, "-", "", -1) setting := getGitConfig(gitBinPath, "http."+serviceName, dir) if serviceName == "uploadpack" { return setting != "false" } return setting == "true" } func getGitConfig(gitBinPath, configName string, dir string) string { args := []string{"config", configName} out := string(gitCommand(gitBinPath, dir, args...)) return out[0 : len(out)-1] } func updateServerInfo(gitBinPath, dir string) []byte { args := []string{"update-server-info"} return gitCommand(gitBinPath, dir, args...) } func gitCommand(gitBinPath, dir string, args ...string) []byte { command := exec.Command(gitBinPath, args...) command.Dir = dir out, err := command.Output() if err != nil { log.Print(err) } return out } // HTTP error response handling functions func renderMethodNotAllowed(w http.ResponseWriter, r *http.Request) { if r.Proto == "HTTP/1.1" { w.WriteHeader(http.StatusMethodNotAllowed) w.Write([]byte("Method Not Allowed")) } else { w.WriteHeader(http.StatusBadRequest) w.Write([]byte("Bad Request")) } } func renderNotFound(w http.ResponseWriter) { w.WriteHeader(http.StatusNotFound) w.Write([]byte("Not Found")) } func renderNoAccess(w http.ResponseWriter) { w.WriteHeader(http.StatusForbidden) w.Write([]byte("Forbidden")) } // Packet-line handling function func packetFlush() []byte { return []byte("0000") } func packetWrite(str string) []byte { s := strconv.FormatInt(int64(len(str)+4), 16) if len(s)%4 != 0 { s = strings.Repeat("0", 4-len(s)%4) + s } return []byte(s + str) } // Header writing functions func hdrNocache(w http.ResponseWriter) { w.Header().Set("Expires", "Fri, 01 Jan 1980 00:00:00 GMT") w.Header().Set("Pragma", "no-cache") w.Header().Set("Cache-Control", "no-cache, max-age=0, must-revalidate") } func hdrCacheForever(w http.ResponseWriter) { now := time.Now().Unix() expires := now + 31536000 w.Header().Set("Date", fmt.Sprintf("%d", now)) w.Header().Set("Expires", fmt.Sprintf("%d", expires)) w.Header().Set("Cache-Control", "public, max-age=31536000") } // Main /* func main() { http.HandleFunc("/", requestHandler()) err := http.ListenAndServe(":8080", nil) if err != nil { log.Fatal("ListenAndServe: ", err) } }*/