From a1040ae657f9852baded093f59268360b9848c4e Mon Sep 17 00:00:00 2001
From: "Frederick F. Kautz IV" <fkautz@alumni.cmu.edu>
Date: Wed, 10 Dec 2014 20:59:49 -0800
Subject: [PATCH] erasure-demo put now takes 1 or 2 arguments

---
 cmd/erasure-demo/put.go | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/cmd/erasure-demo/put.go b/cmd/erasure-demo/put.go
index db41f71a9..923566089 100644
--- a/cmd/erasure-demo/put.go
+++ b/cmd/erasure-demo/put.go
@@ -12,9 +12,15 @@ func put(c *cli.Context) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	filePath := c.Args().Get(0)
-	if len(filePath) == 0 {
-		log.Fatal("Please specify a valid object name \n # erasure-demo put <OBJECTNAME>")
+	objectPath := c.Args().Get(0)
+	var filePath string
+	switch len(c.Args()) {
+	case 1:
+		filePath = objectPath
+	case 2:
+		filePath = c.Args().Get(1)
+	default:
+		log.Fatal("Please specify a valid object name \n # erasure-demo put [OBJECTNAME] [FILENAME]")
 	}
 	inputFile, err := os.Open(filePath)
 	if err != nil {