coderlv
diff --git a/‎mining_libs/jobs.py
Lines changed: 6 additions & 2 deletions b/‎mining_libs/jobs.py
Lines changed: 6 additions & 2 deletions
diff --git a/‎mining_proxy.py
Lines changed: 3 additions & 2 deletions b/‎mining_proxy.py
Lines changed: 3 additions & 2 deletions
@@ -80,9 +80,10 @@ def serialize_header(self, merkle_root, ntime, nonce):
         return r            
 
 class JobRegistry(object):   
-    def __init__(self, f, cmd, no_midstate, real_target, use_old_target=False):
+    def __init__(self, f, cmd, scrypt, no_midstate, real_target, use_old_target=False):
         self.f = f
         self.cmd = cmd # execute this command on new block
+        self.scrypt = scrypt # calculate target for scrypt algorithm instead of sha256
         self.no_midstate = no_midstate # Indicates if calculate midstate for getwork
         self.real_target = real_target # Indicates if real stratum target will be propagated to miners
         self.use_old_target = use_old_target # Use 00000000fffffff...f instead of correct 00000000ffffffff...0 target for really old miners
@@ -112,7 +113,10 @@ def set_extranonce(self, extranonce1, extranonce2_size):
         self.extranonce1_bin = binascii.unhexlify(extranonce1)
 
     def set_difficulty(self, new_difficulty):
-        dif1 = 0x00000000ffff0000000000000000000000000000000000000000000000000000 
+        if self.scrypt:
+            dif1 = 0x0000ffff00000000000000000000000000000000000000000000000000000000
+        else:
+            dif1 = 0x00000000ffff0000000000000000000000000000000000000000000000000000
         self.target = int(dif1 / new_difficulty)
         self.target_hex = binascii.hexlify(utils.uint256_to_str(self.target))
         self.difficulty = new_difficulty
 
@@ -43,7 +43,8 @@ def parse_args():
     parser.add_argument('-t', '--test', dest='test', action='store_true', help='Run performance test on startup')    
     parser.add_argument('-v', '--verbose', dest='verbose', action='store_true', help='Enable low-level debugging messages')
     parser.add_argument('-q', '--quiet', dest='quiet', action='store_true', help='Make output more quiet')
-    parser.add_argument('-i', '--pid-file', dest='pid_file', type=str, help='Store process pid to the file')    
+    parser.add_argument('-i', '--pid-file', dest='pid_file', type=str, help='Store process pid to the file')
+    parser.add_argument('-s', '--scrypt', dest='scrypt', action='store_true', help='Calculate targets for scrypt algorithm')
     return parser.parse_args()
 
 from stratum import settings
@@ -212,7 +213,7 @@ def main(args):
                 event_handler=client_service.ClientMiningService)
 
 
-    job_registry = jobs.JobRegistry(f, cmd=args.blocknotify_cmd,
+    job_registry = jobs.JobRegistry(f, cmd=args.blocknotify_cmd, scrypt=args.scrypt,
                    no_midstate=args.no_midstate, real_target=args.real_target, use_old_target=args.old_target)
     client_service.ClientMiningService.job_registry = job_registry
     client_service.ClientMiningService.reset_timeout()