コード例 #1
0
ファイル: session.go プロジェクト: Eriken/gypo
func (fs *FileSession) pushHandler(incomingPacket packet.PacketStd) {

	fmt.Println("processing push")

	//getting pushList from socket
	incomingBytes, _ := strconv.Atoi(incomingPacket.Data)
	pushListB := make([]byte, incomingBytes)
	i := 0
	for i < incomingBytes {
		chunkSize, _ := fs.Connection.Read(pushListB[i:])
		i = i + chunkSize
		fmt.Println("read so far: ", i)
	}

	//unmarshal data from socket to PacketList instance
	var client_pushList packet.PacketList
	client_pushList.ContentType = "push"
	client_pushList.Name = "client"
	flsize, _ := strconv.Atoi(incomingPacket.Name)
	flObj := make([]packet.PacketStd, flsize)
	client_pushList.Data = flObj
	error := json.Unmarshal(pushListB, &client_pushList)

	if error == nil {
		fmt.Println("push recieved, " + incomingPacket.Name + " entries\n")
	} else {
		fmt.Println(error.Error())
	}

	//tells client push was recieved
	newPacket := packet.PacketStd{packet.Packet{"push", ""}, "PUSH_RECIEVED"}
	go newPacket.SendPacket(fs.Connection)

	//PacketList instance for server files(db)
	var server_fileList packet.PacketList

	db, error := sql.Open("sqlite3", "../gypo.db")

	if error != nil {
		fmt.Println(error.Error())
	} else {
		//get number of files from db to use for file array size
		var dbFileCount int
		error = db.QueryRow("SELECT count(*) AS number FROM files WHERE user_id = ?", fs.User.Id).Scan(&dbFileCount)

		if error != nil {
			fmt.Println(error.Error())
		} else {
			server_fileList.Data = make([]packet.PacketStd, dbFileCount)

			//fetches files from db and adds them to server_fileList
			rows, error := db.Query("SELECT filename, path, hash, is_dir FROM files WHERE user_id = ?", fs.User.Id)
			if error != nil {
				fmt.Println(error.Error())
			} else {
				var is_dir int
				var rowContentType, rowName, rowData string
				var rowPacket packet.PacketStd

				//populates server_fileList.Data
				i = 0
				for rows.Next() {
					error = rows.Scan(&rowContentType, &rowName, &rowData, &is_dir)
					//fmt.Println(i, " | ", is_dir)
					if error != nil {
						fmt.Println(error.Error())
					}
					if is_dir == 1 {
						rowPacket.ContentType = "folder"
						rowPacket.Name = rowName
						rowPacket.Data = ""
					} else {
						rowPacket.ContentType = "hash"
						rowPacket.Name = rowName + rowContentType
						rowPacket.Data = rowData
					}
					server_fileList.Data[i] = rowPacket
					i++
				}

				fmt.Println("server_fileList size: ", len(server_fileList.Data))
				fmt.Println("client_pushList size: ", len(client_pushList.Data))

				/*
				   i = 0
				   for i < len(server_fileList.Data) {
				       fmt.Println("ContentType: " + server_fileList.Data[i].ContentType + " | Name: " + server_fileList.Data[i].Name + " | Data: " + server_fileList.Data[i].Data)
				       i++
				   }

				   fmt.Println("\n\n\n\n")

				   i = 0
				   for i < len(client_pushList.Data) {
				       fmt.Println("ContentType: " + client_pushList.Data[i].ContentType + " | Name: " + client_pushList.Data[i].Name + " | Data: " + server_fileList.Data[i].Data)
				       i++
				   }*/

				//compares client_pushList to server_fileList
				nilPacket := packet.PacketStd{packet.Packet{"", ""}, ""} //to flag list entries as solved
				i = 0
				j := 0
				for i < len(client_pushList.Data) {
					//check if i is allready solved
					if len(client_pushList.Data[i].ContentType) > 0 {
						for j < len(server_fileList.Data) {
							//file
							if client_pushList.Data[i].ContentType == "hash" {
								//match
								if client_pushList.Data[i].Name == server_fileList.Data[j].Name {
									//update file if needed
									if client_pushList.Data[i].Data != server_fileList.Data[j].Data {

										client_pushList.Data[i].Data = "1" + client_pushList.Data[i].Data //adds '1' to indicate UPDATE(existing file)
										fs.tempHash[client_pushList.Data[i].Name] = client_pushList.Data[i]
										fmt.Println("hash for: " + client_pushList.Data[i].Name + " added to tempHash")

										//fmt.Println("path and filename the same, different hash | client: " + client_pushList.Data[i].Data + " server: " + server_fileList.Data[j].Data)
										newPacket := packet.PacketStd{packet.Packet{"hash", client_pushList.Data[i].Name}, "SEND_FILE_TRUE"}
										go newPacket.SendPacket(fs.Connection)
										fmt.Println(client_pushList.Data[i].Name + " exist, different hash, send\n")

										//flag list entries as solved
										client_pushList.Data[i] = nilPacket
										server_fileList.Data[j] = nilPacket
									} else {
										//files are identical, flag list entries as solved
										client_pushList.Data[i] = nilPacket
										server_fileList.Data[j] = nilPacket
									}
								}
								//path
							} else if client_pushList.Data[i].ContentType == "folder" {
								//match
								if client_pushList.Data[i].Name == server_fileList.Data[j].Name {
									//if both paths exist on server and client, flag list entries as solved
									client_pushList.Data[i] = nilPacket
									server_fileList.Data[j] = nilPacket
								}
							}
							j++ //increase server_fileList index
						}
					}
					j = 0 //resets server_fileList index
					i++   //increase client_pushList index
				}

				//client_pushList entries not flagged as handled are treated as new files and folders and will be downloaded an/or created
				i = 0
				for i < len(client_pushList.Data) {

					if client_pushList.Data[i].ContentType == "hash" { //file

						//adds hash to tempHash, if file recieve is successful; tempHash is dumped to db
						client_pushList.Data[i].Data = "0" + client_pushList.Data[i].Data //adds '0' to indicate INSERT(new file)
						fs.tempHash[client_pushList.Data[i].Name] = client_pushList.Data[i]
						fmt.Println("hash for: " + client_pushList.Data[i].Name + " added to tempHash")

						newPacket := packet.PacketStd{packet.Packet{"hash", client_pushList.Data[i].Name}, "SEND_FILE_TRUE"}
						go newPacket.SendPacket(fs.Connection)
						fmt.Println(client_pushList.Data[i].Name + " not exist, send\n")

						//flag list entry as solved
						client_pushList.Data[i] = nilPacket

					} else if client_pushList.Data[i].ContentType == "folder" { //folder
						error = os.MkdirAll(client_pushList.Data[i].Name[1:], 07777) //liberal chmod

						if error != nil {
							fmt.Println(error.Error())
							//adds folder to db
						} else {
							_, error := db.Exec("INSERT INTO files (path, is_dir, user_id) VALUES (?, ?, ?)", client_pushList.Data[i].Name, 1, fs.User.Id)
							if error != nil {
								fmt.Println(error.Error())
							} else {
								fmt.Println("folder: " + client_pushList.Data[i].Name + " created and added to db")
							}
						}
					}
					i++
				}

				fmt.Println("tempHash size: ", len(fs.tempHash))

				//server_fileList entries not flagged as handled are threated as deleted files and gets deleted
				i = 0
				for i < len(server_fileList.Data) {
					if server_fileList.Data[i].ContentType == "hash" { //file
						//delete db entry
						_, error := db.Exec("DELETE FROM files WHERE path = ? AND hash = ? AND user_id = ?", server_fileList.Data[i].Name[:strings.LastIndex(server_fileList.Data[i].Name, "/")+1], server_fileList.Data[i].Data, fs.User.Id)

						if error != nil {
							fmt.Println(error.Error())
						} else {
							//delete file
							_ = os.RemoveAll(fs.RootDir + server_fileList.Data[i].Name)
							/*if error != nil {
							    fmt.Println(error.Error())
							}*/
						}
						//flag list entry as solved
						server_fileList.Data[i] = nilPacket

					} else if server_fileList.Data[i].ContentType == "folder" {
						//delete db entry
						_, error := db.Exec("DELETE FROM files WHERE path = ? AND is_dir = ? AND user_id = ?", server_fileList.Data[i].Name, 1, fs.User.Id)

						if error != nil {
							fmt.Println(error.Error())
						} else {
							//delete folder
							_ = os.RemoveAll(fs.RootDir + server_fileList.Data[i].Name)
							/*if error != nil {
							    fmt.Println(error.Error())
							}*/
						}
						//flag list entry as solved
						server_fileList.Data[i] = nilPacket
					}
					i++
				}
			}
		}
	}
}
コード例 #2
0
ファイル: session.go プロジェクト: Eriken/gypo
func (fs *FileSession) pullHandler(incomingPacket packet.PacketStd) {

	if incomingPacket.Name == "REQ_FILELIST" {
		//db initializing
		db, error := sql.Open("sqlite3", "../gypo.db")

		//tells client pull request recieved
		newPacket := packet.PacketStd{packet.Packet{"pull", "REQ_FILELIST_RECV"}, ""}
		go newPacket.SendPacket(fs.Connection)

		//creates list of server files and sends back to client
		if error != nil {
			fmt.Println(error.Error())
		} else {
			//PacketList instance for server files(db)
			var server_fileList packet.PacketList

			//get number of files from db to use for file server_fileList.Data size
			var dbFileCount int
			error = db.QueryRow("SELECT count(*) AS number FROM files WHERE user_id = ?", fs.User.Id).Scan(&dbFileCount)

			if error != nil {
				fmt.Println(error.Error())
			} else {
				//sets PacketList instance variables
				server_fileList.Data = make([]packet.PacketStd, dbFileCount)
				server_fileList.ContentType = "pull"
				server_fileList.Name = "RESP_FILELIST"

				//get fileList from db
				rows, error := db.Query("SELECT filename, path, hash, is_dir FROM files WHERE user_id = ?", fs.User.Id)

				if error != nil {
					fmt.Println(error.Error())
				} else {
					var rowPacket packet.PacketStd
					var filename, path, hash string
					var is_dir int
					i := 0

					for rows.Next() {
						error = rows.Scan(&filename, &path, &hash, &is_dir)
						if error != nil {
							fmt.Println(error.Error())
						} else {
							if is_dir == 1 {
								rowPacket.ContentType = "folder"
								rowPacket.Name = path
								rowPacket.Data = ""
							} else {
								rowPacket.ContentType = "hash"
								rowPacket.Name = path + filename
								rowPacket.Data = hash
							}
						}
						server_fileList.Data[i] = rowPacket
						i++
					}
					go server_fileList.SendPacket(fs.Connection)
					fmt.Println("server_fileList size: ", len(server_fileList.Data))
				}
			}
		}
	} else if incomingPacket.Name == "REQ_FILE" {
		//wraps file in packet and sends it
		newPacket := packet.PacketStd{packet.Packet{"pull", "RESP_FILE"}, incomingPacket.Data}
		go newPacket.SendFilePacket(fs.Connection)
	}

}
コード例 #3
0
ファイル: server.go プロジェクト: Eriken/gypo
func (ws WebServer) handler(w http.ResponseWriter, r *http.Request) {

	var currentSessionCookie *http.Cookie
	var error error
	var html []byte //respons data array

	//check cookie for auth
	currentSessionCookie, error = r.Cookie("gypoSession")

	//if /login requests
	if r.URL.Path[1:] == "login" {

		//respons packet
		var newPacket packet.PacketStd

		//if no cookie, authentiate
		if error != nil {
			newWebSession := session.WebSession{}
			newWebSession.User = user.User{-1}
			newWebSession.RootDir = ws.RootDir
			newWebSession.SessionChan = ws.SessionsChan

			var loginCode int
			var loginCodeStr string

			//authenticates form input
			loginCode = newWebSession.WebSession_Authenticate(r.FormValue("username"), r.FormValue("password"))
			loginCodeStr = strconv.Itoa(loginCode)

			//makes respons packets from loginCode return value
			if loginCode < 0 { //not success
				newPacket = packet.PacketStd{packet.Packet{"text", "loginCode"}, loginCodeStr}

			} else { //succes
				newPacket = packet.PacketStd{packet.Packet{"text", "loginCode"}, loginCodeStr}
				newWebSession.User.Id = loginCode

				//sets client cookie
				newSessionCookie := &http.Cookie{Name: "gypoSession", Value: strconv.Itoa(newWebSession.User.Id), Domain: ".localhost.com", Expires: time.Now().AddDate(0, 0, 1)} //expires in 24h - time.Now().Add(120*60*1000)
				http.SetCookie(w, newSessionCookie)
			}

			//if cookie, prolong expiration
		} else {
			newSessionCookie := &http.Cookie{Name: "gypoSession", Value: currentSessionCookie.Value, Domain: ".localhost.com", Expires: time.Now().AddDate(0, 0, 1)} //expires in 24h - time.Now().Add(120*60*1000)
			http.SetCookie(w, newSessionCookie)
		}

		//writes response packets to client with json
		html, _ = json.Marshal(newPacket)
		w.Header().Set("Content-Type", "application/json")

		//everything but /login
	} else {
		//if currentSessionCookie missing, no auth
		if error != nil {
			if r.URL.Path == "/" || r.URL.Path == "" || r.URL.Path == "/favicon.ico" { //Browsers automatically request the favicon.ico file by default when you issue a request for a web page. The favicon.ico file is the small icon that appears in the URL bar of your browser.
				//trys to load login for client
				html, error = ws.loadPage("index")

				if error != nil {
					html = []byte("could not load index")
				}
			} else {
				html = []byte("you need to login first")
			}

		} else {
			//refreshes cookie
			newSessionCookie := &http.Cookie{Name: "gypoSession", Value: currentSessionCookie.Value, Domain: ".localhost.com", Expires: time.Now().AddDate(0, 0, 1)} //expires in 24h - time.Now().Add(120*60*1000)
			http.SetCookie(w, newSessionCookie)
			userId, _ := strconv.Atoi(currentSessionCookie.Value)

			//redirects to overview as auth is ok
			if r.URL.Path == "/" || r.URL.Path == "" || r.URL.Path == "/favicon.ico" {
				//trys to load login for client
				html, error = ws.loadPage("overview")

				if error != nil {
					html = []byte("could not load overview")
				}

				//overview jason
			} else if r.URL.Path[1:] == "getOverview" {

				db, error := sql.Open("sqlite3", ws.RootDir+"/gypo.db")

				if error != nil {
					html = []byte(error.Error())

				} else {
					//get number of files to populate fileList array
					var dbFileCount int
					error = db.QueryRow("SELECT count(*) AS number FROM files WHERE user_id = ?", userId).Scan(&dbFileCount)

					if error != nil {
						html = []byte(error.Error())
					} else {
						rows, error := db.Query("SELECT id, filename, path, is_dir FROM files WHERE user_id = ? ", userId)

						if error != nil {
							html = []byte(error.Error())

						} else {
							var id int
							var filename, path string
							var is_dir bool
							var fileList packet.PacketList
							var file packet.PacketStd

							//set array size
							fileList.ContentType = "json"
							fileList.Name = "fileList"
							fileList.Data = make([]packet.PacketStd, dbFileCount)

							//populate array with file objects
							i := 0
							for rows.Next() {
								error = rows.Scan(&id, &filename, &path, &is_dir)

								if error != nil {
									file = packet.PacketStd{packet.Packet{"file", "error"}, error.Error()}
								} else {
									if is_dir == true {
										file = packet.PacketStd{packet.Packet{"folder", path}, strconv.Itoa(id)}
									} else {
										mimeStr := mime.TypeByExtension(strings.ToLower(filename[strings.LastIndex(filename, "."):])) //to lower case to ensure equal parse
										file = packet.PacketStd{packet.Packet{"file|" + mimeStr, path + "|" + filename}, strconv.Itoa(id)}
									}
								}
								fileList.Data[i] = file
								i++
							}

							//marshals fileList to json and sets proper content-type header
							html, _ = json.Marshal(fileList)
							w.Header().Set("Content-Type", "application/json")
						}
						db.Close()
					}
				}
			} else if strings.Index(r.URL.Path, "getFile") == 1 {

				db, error := sql.Open("sqlite3", ws.RootDir+"/gypo.db")
				defer db.Close()

				if error != nil {
					html = []byte(error.Error())

				} else {

					var path, filename string
					fileId, _ := strconv.Atoi(r.URL.RawQuery[strings.Index(r.URL.RawQuery, "=")+1:])

					error = db.QueryRow("SELECT path, filename number FROM files WHERE id = ?", fileId).Scan(&path, &filename)

					if error != nil {
						html = []byte(error.Error())
					} else {
						pathfile := ws.RootDir + "/user-" + strconv.Itoa(userId) + path + filename
						file, error := os.Open(pathfile)

						if error != nil {
							html = []byte(error.Error())
						} else {
							defer file.Close()
							//get filesize
							fi, _ := file.Stat()
							fileSize := int(fi.Size())

							//reads file and writes it to buffer array
							html = make([]byte, fileSize)
							bytesWritten := 0
							i := 0
							fmt.Println("File to be written: " + fi.Name() + " | size: " + strconv.Itoa(fileSize))

							//set headers accoring to file
							mimeStr := mime.TypeByExtension(filename[strings.LastIndex(filename, "."):])
							fmt.Println("setting content-type: " + mimeStr)
							w.Header().Set("Content-Type", mimeStr)
							w.Header().Set("Content-Length", strconv.Itoa(fileSize))
							w.Header().Set("Content-Disposition", "attachment; filename="+filename)

							for bytesWritten < fileSize && error == nil {
								i, error = file.Read(html[bytesWritten:])

								if error != nil {
									html = []byte(error.Error())
								} else {
									bytesWritten = bytesWritten + i
								}
							}
						}
					}
				}

			} else if strings.Index(r.URL.Path, "getImage") == 1 {

				fmt.Println("getImage")

				db, error := sql.Open("sqlite3", ws.RootDir+"/gypo.db")
				defer db.Close()

				if error != nil {
					html = []byte(error.Error())

				} else {

					var path, filename string
					if strings.Index(r.URL.RawQuery, "&width=") == -1 || strings.Index(r.URL.RawQuery, "id=") == -1 {
						html = []byte("bad url")
					} else {
						fileId, error := strconv.Atoi(r.URL.RawQuery[strings.Index(r.URL.RawQuery, "id=")+3 : strings.Index(r.URL.RawQuery, "&width=")])

						if error != nil {
							html = []byte(error.Error())
						} else {
							error = db.QueryRow("SELECT path, filename number FROM files WHERE id = ?", fileId).Scan(&path, &filename)

							if error != nil {
								html = []byte(error.Error())
							} else {
								pathfile := ws.RootDir + "/user-" + strconv.Itoa(userId) + path + filename
								image, error := magick.NewFromFile(pathfile)

								if error != nil {
									html = []byte(error.Error())
								} else {
									defer image.Destroy()

									//if image wider than 50 % of screen; resize
									screenWidth, _ := strconv.Atoi(r.URL.RawQuery[strings.LastIndex(r.URL.RawQuery, "=")+1:])
									var resizeFactor float32
									resizeFactor = float32(screenWidth) * 50.0 / 100.0

									if int(resizeFactor) < image.Width() {
										resizeFactor = resizeFactor / float32(image.Width())
										error = image.Resize(strconv.Itoa(int(float32(image.Width())*resizeFactor)) + "x" + strconv.Itoa(int(float32(image.Height())*resizeFactor)))

										if error != nil {
											fmt.Println(error.Error())
										} else {
											fmt.Println("image resized: " + filename)
										}
										fmt.Println("resizeFactor: ", resizeFactor)
									}

									if error != nil {
										html = []byte(error.Error())

									} else {
										//reads image and writes it to buffer array
										html, error = image.ToBlob(filename[strings.LastIndex(filename, "."):])

										if error != nil {
											html = []byte(error.Error())
											fmt.Println(error.Error())
										} else {

											//set headers according to file
											mimeStr := mime.TypeByExtension(filename[strings.LastIndex(filename, "."):])
											fmt.Println("setting content-type: " + mimeStr)
											w.Header().Set("Content-Type", mimeStr)
											w.Header().Set("Content-Length", strconv.Itoa(len(html)))
											w.Header().Set("Content-Disposition", "inline; filename="+filename)
											fmt.Println("image sent, size: ", len(html), "\n")
										}
									}
								}
							}
						}
					}
				}

			} else { //unknown query
				html, error = ws.loadPage(r.URL.Path[1:])

				if error != nil {
					html = []byte("could not find page")
				}
			}
		}
	}

	//writes output to client
	j, error := w.Write(html)
	if error != nil {
		fmt.Println(error.Error())
	} else {
		fmt.Println("WebServer | bytes written: ", j) //writes amount written to client
	}
}