Method Upload

share/file.go:161–227

(route string, targetUUID string, identity int)

}

func (file *MyFile) Upload(route string, targetUUID string, identity int) {
	var sMessage protocol.Message
	if identity == ADMIN {
		sMessage = protocol.PrepareAndDecideWhichSProtoToLower(global.G_Component.Conn, global.G_Component.Secret, global.G_Component.UUID)
	} else {
		sMessage = protocol.PrepareAndDecideWhichSProtoToUpper(global.G_Component.Conn, global.G_Component.Secret, global.G_Component.UUID)
	}

	dataHeader := &protocol.Header{
		Sender:      global.G_Component.UUID,
		Accepter:    targetUUID,
		MessageType: protocol.FILEDATA,
		RouteLen:    uint32(len([]byte(route))),
		Route:       route,
	}

	errHeader := &protocol.Header{
		Sender:      global.G_Component.UUID,
		Accepter:    targetUUID,
		MessageType: protocol.FILEERR,
		RouteLen:    uint32(len([]byte(route))),
		Route:       route,
	}

	fileErrMess := &protocol.FileErr{
		Error: 1,
	}

	if identity == ADMIN {
		fmt.Println("\n[*] File transmitting, please wait...")
		file.StatusChan <- &Status{Stat: START}
	}

	buffer := make([]byte, 30720)

	defer func() {
		if identity == ADMIN {
			file.StatusChan <- &Status{Stat: DONE}
		}
		runtime.GC()
		file.Handler.Close()
	}()

	for {
		length, err := file.Handler.Read(buffer)
		if err != nil && err != io.EOF {
			protocol.ConstructMessage(sMessage, errHeader, fileErrMess, false)
			sMessage.SendMessage()
			return
		} else if err != nil && err == io.EOF {
			return
		}

		fileDataMess := &protocol.FileData{
			DataLen: uint64(length),
			Data:    buffer[:length],
		}