Skip to content

Commit

Permalink
Temporary files use ioutil.TempDir
Browse files Browse the repository at this point in the history
  • Loading branch information
Immutability, LLC committed Dec 22, 2017
1 parent 79f42d8 commit fad7d18
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 32 deletions.
8 changes: 4 additions & 4 deletions ethereum/path_accounts.go
Original file line number Diff line number Diff line change
Expand Up @@ -243,7 +243,7 @@ func (b *backend) pathAccountsCreate(req *logical.Request, data *framework.Field
} else if passphrase == "" {
return nil, fmt.Errorf("must provide a passphrase to encrypt the keystore")
}
tmpDir, err := b.createTemporaryKeystoreDirectory(req.Path)
tmpDir, err := b.createTemporaryKeystoreDirectory()
if err != nil {
return nil, err
}
Expand Down Expand Up @@ -273,7 +273,7 @@ func (b *backend) pathAccountsCreate(req *logical.Request, data *framework.Field
if err != nil {
return nil, err
}
b.removeTemporaryKeystore(req.Path)
b.removeTemporaryKeystore(tmpDir)
return &logical.Response{
Data: map[string]interface{}{
"account": accountJSON.Address,
Expand Down Expand Up @@ -301,7 +301,7 @@ func (b *backend) pathAccountsUpdate(req *logical.Request, data *framework.Field
if err != nil {
return nil, err
}
tmpDir, err := b.createTemporaryKeystoreDirectory(req.Path)
tmpDir, err := b.createTemporaryKeystoreDirectory()
if err != nil {
return nil, err
}
Expand All @@ -311,7 +311,7 @@ func (b *backend) pathAccountsUpdate(req *logical.Request, data *framework.Field
}

jsonKeystore, err := b.rekeyJSONKeystore(keystorePath, account.Passphrase, passphrase)
b.removeTemporaryKeystore(req.Path)
b.removeTemporaryKeystore(tmpDir)
if err != nil {
return nil, err
} else {
Expand Down
23 changes: 7 additions & 16 deletions ethereum/util.go
Original file line number Diff line number Diff line change
Expand Up @@ -39,22 +39,13 @@ func (b *backend) writeTemporaryKeystoreFile(path string, filename string, data
return keystorePath, err
}

func (b *backend) createTemporaryKeystoreDirectory(name string) (string, error) {
file, _ := os.Open(PathTempDir + name)
if file != nil {
file.Close()
return "", fmt.Errorf("account already exists at %s", PathTempDir+name)
}
return PathTempDir + name, os.MkdirAll(PathTempDir+name, os.FileMode(0522))
func (b *backend) createTemporaryKeystoreDirectory() (string, error) {
dir, err := ioutil.TempDir("", "keystore")
return dir, err
}

func (b *backend) removeTemporaryKeystore(name string) error {
file, _ := os.Open(PathTempDir + name)
if file != nil {
return os.RemoveAll(PathTempDir + name)
} else {
return fmt.Errorf("keystore doesn't exist at %s", PathTempDir+name)
}
func (b *backend) removeTemporaryKeystore(path string) error {
return os.RemoveAll(path)
}

func convertMapToStringValue(initial map[string]interface{}) map[string]string {
Expand Down Expand Up @@ -177,7 +168,7 @@ func (b *backend) NewTransactor(key *ecdsa.PrivateKey) *bind.TransactOpts {
}

func (b *backend) getAccountPrivateKey(path string, account Account) (*keystore.Key, error) {
tmpDir, err := b.createTemporaryKeystoreDirectory(path)
tmpDir, err := b.createTemporaryKeystoreDirectory()
if err != nil {
return nil, err
}
Expand All @@ -190,7 +181,7 @@ func (b *backend) getAccountPrivateKey(path string, account Account) (*keystore.
if err != nil {
return nil, err
}
err = b.removeTemporaryKeystore(path)
err = b.removeTemporaryKeystore(tmpDir)
return key, err
}

Expand Down
12 changes: 0 additions & 12 deletions scripts/cycle.sh

This file was deleted.

0 comments on commit fad7d18

Please sign in to comment.