2021-11-09 15:36:23 +08:00
|
|
|
// Copyright 2021 The Gitea 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 activitypub
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"crypto/rsa"
|
|
|
|
"crypto/x509"
|
|
|
|
"encoding/pem"
|
|
|
|
"fmt"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
user_model "code.gitea.io/gitea/models/user"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"github.com/go-fed/activity/pub"
|
|
|
|
"github.com/go-fed/httpsig"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2021-11-10 20:35:02 +08:00
|
|
|
// ActivityStreamsContentType const
|
|
|
|
ActivityStreamsContentType = "application/ld+json; profile=\"https://www.w3.org/ns/activitystreams\""
|
2021-11-09 15:36:23 +08:00
|
|
|
)
|
|
|
|
|
2021-11-10 20:35:02 +08:00
|
|
|
func containsRequiredHTTPHeaders(method string, headers []string) error {
|
2021-11-09 15:36:23 +08:00
|
|
|
var hasRequestTarget, hasDate, hasDigest bool
|
|
|
|
for _, header := range headers {
|
|
|
|
hasRequestTarget = hasRequestTarget || header == httpsig.RequestTarget
|
|
|
|
hasDate = hasDate || header == "Date"
|
|
|
|
hasDigest = method == "GET" || hasDigest || header == "Digest"
|
|
|
|
}
|
|
|
|
if !hasRequestTarget {
|
|
|
|
return fmt.Errorf("missing http header for %s: %s", method, httpsig.RequestTarget)
|
|
|
|
} else if !hasDate {
|
|
|
|
return fmt.Errorf("missing http header for %s: Date", method)
|
|
|
|
} else if !hasDigest {
|
|
|
|
return fmt.Errorf("missing http header for %s: Digest", method)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2021-11-10 20:35:02 +08:00
|
|
|
// Client struct
|
2021-11-09 15:36:23 +08:00
|
|
|
type Client struct {
|
|
|
|
clock pub.Clock
|
|
|
|
client *http.Client
|
|
|
|
algs []httpsig.Algorithm
|
|
|
|
digestAlg httpsig.DigestAlgorithm
|
|
|
|
getHeaders []string
|
|
|
|
postHeaders []string
|
|
|
|
priv *rsa.PrivateKey
|
2021-11-10 20:35:02 +08:00
|
|
|
pubID string
|
2021-11-09 15:36:23 +08:00
|
|
|
}
|
|
|
|
|
2021-11-10 20:35:02 +08:00
|
|
|
// NewClient function
|
|
|
|
func NewClient(user *user_model.User, pubID string) (c *Client, err error) {
|
|
|
|
if err = containsRequiredHTTPHeaders(http.MethodGet, setting.Federation.GetHeaders); err != nil {
|
2021-11-09 15:36:23 +08:00
|
|
|
return
|
2021-11-10 20:35:02 +08:00
|
|
|
} else if err = containsRequiredHTTPHeaders(http.MethodPost, setting.Federation.PostHeaders); err != nil {
|
2021-11-09 15:36:23 +08:00
|
|
|
return
|
|
|
|
} else if !httpsig.IsSupportedDigestAlgorithm(setting.Federation.DigestAlgorithm) {
|
|
|
|
err = fmt.Errorf("unsupported digest algorithm: %s", setting.Federation.DigestAlgorithm)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
algos := make([]httpsig.Algorithm, len(setting.Federation.Algorithms))
|
|
|
|
for i, algo := range setting.Federation.Algorithms {
|
|
|
|
algos[i] = httpsig.Algorithm(algo)
|
|
|
|
}
|
|
|
|
clock, err := NewClock()
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
priv, err := GetPrivateKey(user)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
privPem, _ := pem.Decode([]byte(priv))
|
|
|
|
privParsed, err := x509.ParsePKCS1PrivateKey(privPem.Bytes)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
c = &Client{
|
|
|
|
clock: clock,
|
|
|
|
client: &http.Client{},
|
|
|
|
algs: algos,
|
|
|
|
digestAlg: httpsig.DigestAlgorithm(setting.Federation.DigestAlgorithm),
|
|
|
|
getHeaders: setting.Federation.GetHeaders,
|
|
|
|
postHeaders: setting.Federation.PostHeaders,
|
|
|
|
priv: privParsed,
|
2021-11-10 20:35:02 +08:00
|
|
|
pubID: pubID,
|
2021-11-09 15:36:23 +08:00
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-11-10 20:35:02 +08:00
|
|
|
// NewRequest function
|
|
|
|
func (c *Client) NewRequest(b []byte, to string) (req *http.Request, err error) {
|
2021-11-09 15:36:23 +08:00
|
|
|
byteCopy := make([]byte, len(b))
|
|
|
|
copy(byteCopy, b)
|
|
|
|
buf := bytes.NewBuffer(byteCopy)
|
|
|
|
req, err = http.NewRequest(http.MethodPost, to, buf)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
2021-11-10 20:35:02 +08:00
|
|
|
req.Header.Add("Content-Type", ActivityStreamsContentType)
|
2021-11-09 15:36:23 +08:00
|
|
|
req.Header.Add("Accept-Charset", "utf-8")
|
|
|
|
req.Header.Add("Date", fmt.Sprintf("%s GMT", c.clock.Now().UTC().Format("Mon, 02 Jan 2006 15:04:05")))
|
|
|
|
|
|
|
|
signer, _, err := httpsig.NewSigner(c.algs, c.digestAlg, c.postHeaders, httpsig.Signature, 60)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
2021-11-10 20:35:02 +08:00
|
|
|
err = signer.SignRequest(c.priv, c.pubID, req, b)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Post function
|
|
|
|
func (c *Client) Post(b []byte, to string) (resp *http.Response, err error) {
|
|
|
|
var req *http.Request
|
|
|
|
if req, err = c.NewRequest(b, to); err != nil {
|
2021-11-09 15:36:23 +08:00
|
|
|
return
|
|
|
|
}
|
|
|
|
resp, err = c.client.Do(req)
|
|
|
|
return
|
|
|
|
}
|