update writefile (#93)

CESSProject · Nov 27, 2023 · e40b326 · e40b326
1 parent 42284f0
commit e40b326
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/core/config.go b/core/config.go
@@ -28,3 +28,5 @@ const DirMode = 0644
 const BufferSize = 64 * SIZE_1KiB
 const SegmentSize = 64 * SIZE_1MiB
 const FragmentSize = 16 * SIZE_1MiB
+
+const ZeroFileHash_16M = "080acf35a507ac9849cfcba47dc2ad83e01b75663a516279c8b9d243b719643e"
diff --git a/core/writefile.go b/core/writefile.go
@@ -210,6 +210,23 @@ func (e *WriteFileProtocol) onWriteFileRequest(s network.Stream) {
 
 	fpath := filepath.Join(dir, data.Datahash)
 
+	if data.Datahash == ZeroFileHash_16M {
+		f, err := os.Create(fpath)
+		if err != nil {
+			s.Reset()
+			return
+		}
+		defer f.Close()
+		_, err = f.Write(make([]byte, FragmentSize))
+		if err != nil {
+			s.Reset()
+			return
+		}
+		resp.Code = P2PResponseFinish
+		e.SendProtoMessage(s.Conn().RemotePeer(), protocol.ID(e.ProtocolPrefix+writeFileResponse), resp)
+		return
+	}
+
 	fstat, err = os.Stat(fpath)
 	if err == nil {
 		size = fstat.Size()