Skip to content

Commit

Permalink
Remove unused modules (#72)
Browse files Browse the repository at this point in the history
  • Loading branch information
AstaFrode authored Aug 29, 2023
1 parent ce7d0ae commit d80058f
Show file tree
Hide file tree
Showing 7 changed files with 7 additions and 415 deletions.
99 changes: 0 additions & 99 deletions core/file.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,9 @@ package core

import (
"context"
"io"
"os"
"path/filepath"

"github.com/CESSProject/p2p-go/pb"
"github.com/libp2p/go-libp2p/core/network"
"github.com/libp2p/go-libp2p/core/peer"
"github.com/libp2p/go-libp2p/core/protocol"
"github.com/libp2p/go-msgio/pbio"
Expand All @@ -29,7 +26,6 @@ type FileProtocol struct {

func (n *Node) NewFileProtocol() *FileProtocol {
e := FileProtocol{Node: n}
n.SetStreamHandler(protocol.ID(n.protocolPrefix+FILE_PROTOCOL), e.onFileRequest)
return &e
}

Expand Down Expand Up @@ -85,98 +81,3 @@ func (e *protocols) FileReq(peerId peer.ID, filehash string, filetype pb.FileTyp

return respMsg.Code, nil
}

// remote peer requests handler
func (e *FileProtocol) onFileRequest(s network.Stream) {
defer s.Close()

var resp = &pb.Response{}
var putResp = &pb.PutResponse{
Code: 1,
}
var respMsg = &pb.Response_PutResponse{
PutResponse: putResp,
}
resp.Response = respMsg

var reqMsg = &pb.Request{}

r := pbio.NewDelimitedReader(s, FileProtocolMsgBuf)
w := pbio.NewDelimitedWriter(s)
err := r.ReadMsg(reqMsg)
if err != nil {
w.WriteMsg(resp)
return
}

switch reqMsg.GetRequest().(type) {
case *pb.Request_PutRequest:
putReq := reqMsg.GetPutRequest()
switch putReq.Type {
case pb.FileType_IdleData:
fpath := filepath.Join(e.FileProtocol.GetDirs().IdleDataDir, putReq.Hash)
err = saveFileStream(r, w, reqMsg, resp, fpath, putReq.Size, putReq.Data)
if err == nil {
putResp.Code = 0
}
e.FileProtocol.putIdleDataCh(fpath)
default:
}
default:
}
w.WriteMsg(resp)
return
}

func saveFileStream(r pbio.ReadCloser, w pbio.WriteCloser, reqMsg *pb.Request, resp *pb.Response, fpath string, size uint64, data []byte) error {
f, err := os.OpenFile(fpath, os.O_CREATE|os.O_APPEND|os.O_WRONLY, os.ModePerm)
if err != nil {
return errors.Wrapf(err, "[OpenFile]")
}
defer f.Close()

_, err = f.Write(data)
if err != nil {
return errors.Wrapf(err, "[f.Write]")
}

putReq := &pb.PutRequest{}
for bytesRead := uint64(0); bytesRead < size; {
// Receive bytes
err := r.ReadMsg(reqMsg)
if err != nil {
if err != io.EOF {
return errors.Wrapf(err, "[ReadMsg]")
}
break
}

err = w.WriteMsg(resp)
if err != nil {
return errors.Wrapf(err, "[WriteMsg]")
}

putReq = reqMsg.GetPutRequest()

// Write bytes to the file
_, err = f.Write(putReq.Data)
if err != nil {
return errors.Wrapf(err, "[f.Write]")
}
}

err = f.Sync()
if err != nil {
return errors.Wrapf(err, "[f.Sync]")
}

fstat, err := f.Stat()
if err != nil {
return errors.Wrapf(err, "[f.Stat]")
}

if uint64(fstat.Size()) != size {
return errors.New("invalid size")
}
return nil
}
176 changes: 0 additions & 176 deletions core/filetransfer.go

This file was deleted.

63 changes: 0 additions & 63 deletions core/idle.go

This file was deleted.

Loading

0 comments on commit d80058f

Please sign in to comment.