|
| 1 | +package fs9p |
| 2 | + |
| 3 | +import ( |
| 4 | + "fmt" |
| 5 | + "net" |
| 6 | + "os" |
| 7 | + "path/filepath" |
| 8 | + |
| 9 | + "github.com/DeedleFake/p9" |
| 10 | + "github.com/DeedleFake/p9/proto" |
| 11 | + "github.com/crc-org/crc/v2/pkg/crc/constants" |
| 12 | + "github.com/sirupsen/logrus" |
| 13 | +) |
| 14 | + |
| 15 | +type Server struct { |
| 16 | + // Listener this server is bound to |
| 17 | + Listener net.Listener |
| 18 | + // Plan9 Filesystem type that holds the exposed directory |
| 19 | + Filesystem p9.FileSystem |
| 20 | + // Directory this server exposes |
| 21 | + ExposedDir string |
| 22 | + // Errors from the server being started will come out here |
| 23 | + ErrChan chan error |
| 24 | +} |
| 25 | + |
| 26 | +// New9pServer exposes a single directory (and all children) via the given net.Listener |
| 27 | +// and returns the server struct. |
| 28 | +// Directory given must be an absolute path and must exist. |
| 29 | +func New9pServer(listener net.Listener, exposeDir string) (*Server, error) { |
| 30 | + // Verify that exposeDir makes sense. |
| 31 | + if !filepath.IsAbs(exposeDir) { |
| 32 | + return nil, fmt.Errorf("path to expose to machine must be absolute: %s", exposeDir) |
| 33 | + } |
| 34 | + stat, err := os.Stat(exposeDir) |
| 35 | + if err != nil { |
| 36 | + return nil, fmt.Errorf("cannot stat path to expose to machine: %w", err) |
| 37 | + } |
| 38 | + if !stat.IsDir() { |
| 39 | + return nil, fmt.Errorf("path to expose to machine must be a directory: %s", exposeDir) |
| 40 | + } |
| 41 | + |
| 42 | + fs := p9.FileSystem(p9.Dir(exposeDir)) |
| 43 | + errChan := make(chan error) |
| 44 | + |
| 45 | + toReturn := new(Server) |
| 46 | + toReturn.Listener = listener |
| 47 | + toReturn.Filesystem = fs |
| 48 | + toReturn.ExposedDir = exposeDir |
| 49 | + toReturn.ErrChan = errChan |
| 50 | + |
| 51 | + return toReturn, nil |
| 52 | +} |
| 53 | + |
| 54 | +// Start a server created by New9pServer. |
| 55 | +func (s *Server) Start() error { |
| 56 | + go func() { |
| 57 | + s.ErrChan <- proto.Serve(s.Listener, p9.Proto(), p9.FSConnHandler(s.Filesystem, constants.Plan9Msize)) |
| 58 | + close(s.ErrChan) |
| 59 | + }() |
| 60 | + |
| 61 | + // Just before returning, check to see if we got an error off server startup. |
| 62 | + select { |
| 63 | + case err := <-s.ErrChan: |
| 64 | + return fmt.Errorf("starting 9p server: %w", err) |
| 65 | + default: |
| 66 | + logrus.Infof("Successfully started 9p server on %s for directory %s", s.Listener.Addr().String(), s.ExposedDir) |
| 67 | + return nil |
| 68 | + } |
| 69 | +} |
| 70 | + |
| 71 | +// Stop a running server. |
| 72 | +// Please note that this does *BAD THINGS* to clients if they are still running |
| 73 | +// when the server stops. Processes get stuck in I/O deep sleep and zombify, and |
| 74 | +// nothing I do other than restarting the VM can remove the zombies. |
| 75 | +func (s *Server) Stop() error { |
| 76 | + if err := s.Listener.Close(); err != nil { |
| 77 | + return err |
| 78 | + } |
| 79 | + logrus.Infof("Successfully stopped 9p server for directory %s", s.ExposedDir) |
| 80 | + return nil |
| 81 | +} |
| 82 | + |
| 83 | +// WaitForError from a running server. |
| 84 | +func (s *Server) WaitForError() error { |
| 85 | + err := <-s.ErrChan |
| 86 | + return err |
| 87 | +} |
0 commit comments