coderlv
diff --git a/‎mining_libs/stratum_listener.py
Lines changed: 20 additions & 3 deletions b/‎mining_libs/stratum_listener.py
Lines changed: 20 additions & 3 deletions
diff --git a/‎mining_proxy.py
Lines changed: 2 additions & 0 deletions b/‎mining_proxy.py
Lines changed: 2 additions & 0 deletions
@@ -80,6 +80,14 @@ class StratumProxyService(GenericService):
     service_vendor = 'mining_proxy'
     is_default = True
 
+    custom_user = None
+    custom_password = None
+    
+    @classmethod
+    def _set_custom_user(cls, custom_user, custom_password):
+        cls.custom_user = custom_user
+        cls.custom_password = custom_password
+    
     _f = None # Factory of upstream Stratum connection
     extranonce1 = None
     extranonce2_size = None
@@ -130,7 +138,11 @@ def _drop_tail(self, result, tail):
     def authorize(self, worker_name, worker_password, *args):
         if self._f.client == None or not self._f.client.connected:
             yield self._f.on_connect
-                        
+            
+        if self.custom_user:
+            worker_name = self.custom_user
+            worker_password = self.custom_password
+                            
         result = (yield self._f.rpc('mining.authorize', [worker_name, worker_password]))
         defer.returnValue(result)
 
@@ -163,7 +175,12 @@ def subscribe(self, *args):
     def submit(self, worker_name, job_id, extranonce2, ntime, nonce, *args):
         if self._f.client == None or not self._f.client.connected:
             raise SubmitException("Upstream not connected")
-
+        
+        if self.custom_user:
+            remote_name = self.custom_user
+        else:
+            remote_name = worker_name
+        
         session = self.connection_ref().get_session()
         tail = session.get('tail')
         if tail == None:
@@ -172,7 +189,7 @@ def submit(self, worker_name, job_id, extranonce2, ntime, nonce, *args):
         start = time.time()
 
         try:
-            result = (yield self._f.rpc('mining.submit', [worker_name, job_id, tail+extranonce2, ntime, nonce]))
+            result = (yield self._f.rpc('mining.submit', [remote_name, job_id, tail+extranonce2, ntime, nonce]))
         except RemoteServiceException as exc:
             response_time = (time.time() - start) * 1000
             log.info("[%dms] Share from '%s' REJECTED: %s" % (response_time, worker_name, str(exc)))
 
@@ -107,6 +107,8 @@ def on_connect(f, workers, job_registry):
     (_, extranonce1, extranonce2_size) = (yield f.rpc('mining.subscribe', []))[:3]
     job_registry.set_extranonce(extranonce1, extranonce2_size)
     stratum_listener.StratumProxyService._set_extranonce(extranonce1, extranonce2_size)
+    if args.custom_user:
+        stratum_listener.StratumProxyService._set_custom_user(args.custom_user, args.custom_password)
 
     defer.returnValue(f)