add letter to task3

main
Nikita Tokarchuk 5 years ago
parent 8fb2789461
commit 2408336bdd
  1. 43
      task3-backend/lib/server.go

@ -6,7 +6,10 @@ import (
"crypto/ecdsa"
"errors"
"fmt"
"io"
"net"
"os"
"path"
"sync/atomic"
"time"
@ -17,7 +20,9 @@ import (
"github.com/docker/docker/api/types/network"
"github.com/docker/docker/client"
"github.com/go-redis/redis/v7"
"github.com/google/uuid"
log "github.com/sirupsen/logrus"
"golang.org/x/sys/unix"
)
type Server struct {
@ -174,6 +179,11 @@ func (s *Server) readKey(conn net.Conn) (key string, err error) {
func (s *Server) createJail(now int64, claims *jwtgo.StandardClaims) (id string, err error) {
letterPath, letterFullPath, err := createLetterProtected(s.LetterPath)
if err != nil {
return
}
name := fmt.Sprintf("%d-%s", now, claims.Id)
cfg := &container.Config{
NetworkDisabled: true,
@ -188,9 +198,8 @@ func (s *Server) createJail(now int64, claims *jwtgo.StandardClaims) (id string,
}
letter := mount.Mount{
Type: mount.TypeBind,
Source: s.LetterPath,
Target: "/letter",
ReadOnly: true,
Source: letterFullPath,
Target: fmt.Sprintf("/%s", letterPath),
}
hcfg := &container.HostConfig{
NetworkMode: "none",
@ -307,3 +316,31 @@ func createProxyChan(conn net.Conn, closed *uint32) (out chan []byte) {
return
}
func createLetterProtected(sourcePath string) (letterPath string, letterFullPath string, err error) {
source, err := os.Open(sourcePath)
if err != nil {
return
}
defer source.Close()
tmpPath := os.TempDir()
letterPath = fmt.Sprintf("%s.letter", uuid.New().String())
letterFullPath = path.Join(tmpPath, letterPath)
letter, err := os.OpenFile(letterFullPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0000)
if err != nil {
return
}
defer letter.Close()
_, err = io.Copy(letter, source)
if err != nil {
return
}
_ = unix.Sync()
return
}

Loading…
Cancel
Save