From 36b817d34fc4366d6cb2db16e2e2c5848913f349 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Germ=C3=A1n=20Mart=C3=ADnez?= <germaga92@gmail.com>
Date: Thu, 5 Sep 2019 11:55:08 +0200
Subject: [PATCH] Add ethereum contract address validator

---
 src/components/forms/AddressInput/index.jsx               | 4 ++++
 src/components/forms/validator.js                         | 8 ++++++++
 .../Balances/SendModal/screens/SendCustomTx/index.jsx     | 1 +
 3 files changed, 13 insertions(+)

diff --git a/src/components/forms/AddressInput/index.jsx b/src/components/forms/AddressInput/index.jsx
index 71217c95..1ad3ea3a 100644
--- a/src/components/forms/AddressInput/index.jsx
+++ b/src/components/forms/AddressInput/index.jsx
@@ -7,6 +7,7 @@ import {
   composeValidators,
   required,
   mustBeEthereumAddress,
+  mustBeEthereumContractAddress,
 } from '~/components/forms/validator'
 import { getAddressFromENS } from '~/logic/wallets/getWeb3'
 
@@ -19,6 +20,7 @@ type Props = {
   testId?: string,
   validators?: Function[],
   inputAdornment?: React.Element,
+  mustBeContract?: boolean,
 }
 
 const isValidEnsName = (name) => /^([\w-]+\.)+(eth|test|xyz|luxe)$/.test(name)
@@ -35,6 +37,7 @@ const AddressInput = ({
   testId,
   inputAdornment,
   validators = [],
+  mustBeContract,
 }: Props): React.Element<*> => (
   <>
     <Field
@@ -44,6 +47,7 @@ const AddressInput = ({
       validate={composeValidators(
         required,
         mustBeEthereumAddress,
+        mustBeContract && mustBeEthereumContractAddress,
         ...validators,
       )}
       inputAdornment={inputAdornment}
diff --git a/src/components/forms/validator.js b/src/components/forms/validator.js
index 0c0fd6a2..119058ad 100644
--- a/src/components/forms/validator.js
+++ b/src/components/forms/validator.js
@@ -66,6 +66,14 @@ export const mustBeEthereumAddress = simpleMemoize((address: Field) => {
   return isAddress ? undefined : 'Address should be a valid Ethereum address or ENS name'
 })
 
+export const mustBeEthereumContractAddress = simpleMemoize(async (address: Field) => {
+  const contractCode: string = await getWeb3().eth.getCode(address)
+
+  return (!contractCode || contractCode.replace('0x', '').replace(/0/g, '') === '')
+    ? 'Address should be a valid Ethereum contract address or ENS name'
+    : undefined
+})
+
 export const minMaxLength = (minLen: string | number, maxLen: string | number) => (value: string) => (value.length >= +minLen && value.length <= +maxLen ? undefined : `Should be ${minLen} to ${maxLen} symbols`)
 
 export const ADDRESS_REPEATED_ERROR = 'Address already introduced'
diff --git a/src/routes/safe/components/Balances/SendModal/screens/SendCustomTx/index.jsx b/src/routes/safe/components/Balances/SendModal/screens/SendCustomTx/index.jsx
index c580078d..76cc8d46 100644
--- a/src/routes/safe/components/Balances/SendModal/screens/SendCustomTx/index.jsx
+++ b/src/routes/safe/components/Balances/SendModal/screens/SendCustomTx/index.jsx
@@ -99,6 +99,7 @@ const SendCustomTx = ({
                       text="Recipient*"
                       className={classes.addressInput}
                       fieldMutator={mutators.setRecipient}
+                      mustBeContract
                     />
                   </Col>
                 </Row>