mikrotik check cert
This commit is contained in:
		
							
								
								
									
										100
									
								
								certpusher.py
									
									
									
									
									
								
							
							
						
						
									
										100
									
								
								certpusher.py
									
									
									
									
									
								
							| @@ -269,8 +269,12 @@ class MikroTikManager(SSHManager): | |||||||
|         self.cert_name = "letsencrypt" |         self.cert_name = "letsencrypt" | ||||||
|         self.key_name = "letsencrypt-key" |         self.key_name = "letsencrypt-key" | ||||||
|      |      | ||||||
|     def check_certificate_expiry(self, source_cert: x509.Certificate) -> bool: |     def check_certificate_expiry(self, source_cert: x509.Certificate, services: List[str]) -> bool: | ||||||
|         """Check if certificate on MikroTik needs update""" |         """ | ||||||
|  |         Check if certificate on MikroTik needs update | ||||||
|  |         Also verifies that services are properly configured | ||||||
|  |         Returns True if upload needed, False if everything is OK | ||||||
|  |         """ | ||||||
|         try: |         try: | ||||||
|             logger.info("Checking MikroTik certificate") |             logger.info("Checking MikroTik certificate") | ||||||
|              |              | ||||||
| @@ -311,28 +315,86 @@ class MikroTikManager(SSHManager): | |||||||
|              |              | ||||||
|             time_diff = abs((source_expiry - mikrotik_expiry).total_seconds()) |             time_diff = abs((source_expiry - mikrotik_expiry).total_seconds()) | ||||||
|              |              | ||||||
|             if time_diff < 86400: |             if time_diff >= 86400: | ||||||
|                 logger.info("✓ Certificate is current. Skipping.") |  | ||||||
|                 return False |  | ||||||
|             else: |  | ||||||
|                 logger.info(f"Certificate differs. Upload needed.") |                 logger.info(f"Certificate differs. Upload needed.") | ||||||
|                 return True |                 return True | ||||||
|              |              | ||||||
|  |             # Certificate is current, but check if services are properly configured | ||||||
|  |             logger.info("Certificate is current. Verifying services configuration...") | ||||||
|  |              | ||||||
|  |             cert_name = "letsencrypt.pem_0" | ||||||
|  |             services_need_update = False | ||||||
|  |              | ||||||
|  |             for service in services: | ||||||
|  |                 success, stdout, stderr = self.execute_command( | ||||||
|  |                     f'/ip service print where name="{service}"', | ||||||
|  |                     ignore_error=True | ||||||
|  |                 ) | ||||||
|  |                  | ||||||
|  |                 if success and stdout: | ||||||
|  |                     # Check if certificate is set correctly | ||||||
|  |                     if f'certificate={cert_name}' not in stdout and 'certificate=letsencrypt' not in stdout: | ||||||
|  |                         logger.warning(f"Service {service} not using correct certificate") | ||||||
|  |                         services_need_update = True | ||||||
|  |                     else: | ||||||
|  |                         logger.info(f"✓ Service {service} properly configured") | ||||||
|  |                 else: | ||||||
|  |                     logger.warning(f"Could not check service {service}") | ||||||
|  |                     services_need_update = True | ||||||
|  |              | ||||||
|  |             if services_need_update: | ||||||
|  |                 logger.info("Services need reconfiguration. Updating...") | ||||||
|  |                 self.configure_services(services, cert_name) | ||||||
|  |                 return False  # Don't need to upload cert, just reconfigure | ||||||
|  |              | ||||||
|  |             logger.info("✓ Certificate and services are current. Skipping.") | ||||||
|  |             return False | ||||||
|  |                  | ||||||
|         except Exception as e: |         except Exception as e: | ||||||
|             logger.warning(f"Error checking: {e}") |             logger.warning(f"Error checking: {e}") | ||||||
|             return True |             return True | ||||||
|          |          | ||||||
|  |         def configure_services(self, services: List[str], cert_name: str): | ||||||
|  |             """Configure services to use certificate without re-uploading""" | ||||||
|  |             try: | ||||||
|  |                 for service in services: | ||||||
|  |                     logger.info(f"Configuring {service}") | ||||||
|  |                      | ||||||
|  |                     # Set certificate | ||||||
|  |                     success, _, stderr = self.execute_command( | ||||||
|  |                         f'/ip service set {service} certificate="{cert_name}"', | ||||||
|  |                         ignore_error=True | ||||||
|  |                     ) | ||||||
|  |                      | ||||||
|  |                     if success: | ||||||
|  |                         logger.info(f"✓ {service} configured") | ||||||
|  |                     else: | ||||||
|  |                         logger.warning(f"Failed to configure {service}: {stderr}") | ||||||
|  |                      | ||||||
|  |                     # Ensure service is enabled | ||||||
|  |                     success, stdout, _ = self.execute_command( | ||||||
|  |                         f'/ip service print where name="{service}"', | ||||||
|  |                         ignore_error=True | ||||||
|  |                     ) | ||||||
|  |                      | ||||||
|  |                     if 'disabled=yes' in stdout: | ||||||
|  |                         logger.info(f"Enabling {service}") | ||||||
|  |                         self.execute_command(f'/ip service enable {service}') | ||||||
|  |                          | ||||||
|  |             except Exception as e: | ||||||
|  |                 logger.error(f"Service configuration failed: {e}") | ||||||
|  |  | ||||||
|         def upload_certificate(self, cert_path: str, key_path: str, check_first: bool,  |         def upload_certificate(self, cert_path: str, key_path: str, check_first: bool,  | ||||||
|                             source_cert: x509.Certificate, services: List[str] = None) -> Tuple[bool, bool]: |                             source_cert: x509.Certificate, services: List[str] = None) -> Tuple[bool, bool]: | ||||||
|             """Upload certificate to MikroTik""" |             """Upload certificate to MikroTik""" | ||||||
|             try: |             try: | ||||||
|             if check_first and source_cert: |  | ||||||
|                 if not self.check_certificate_expiry(source_cert): |  | ||||||
|                     return True, False |  | ||||||
|              |  | ||||||
|                 if not services: |                 if not services: | ||||||
|                     services = ['www-ssl'] |                     services = ['www-ssl'] | ||||||
|                  |                  | ||||||
|  |                 if check_first and source_cert: | ||||||
|  |                     if not self.check_certificate_expiry(source_cert, services): | ||||||
|  |                         return True, False  # Certificate and services are OK | ||||||
|  |                  | ||||||
|                 logger.info(f"Deploying certificate for: {', '.join(services)}") |                 logger.info(f"Deploying certificate for: {', '.join(services)}") | ||||||
|                  |                  | ||||||
|                 # Disable services |                 # Disable services | ||||||
| @@ -409,22 +471,8 @@ class MikroTikManager(SSHManager): | |||||||
|                 imported_cert_name = "letsencrypt.pem_0" |                 imported_cert_name = "letsencrypt.pem_0" | ||||||
|                 logger.info(f"Using certificate: {imported_cert_name}") |                 logger.info(f"Using certificate: {imported_cert_name}") | ||||||
|                  |                  | ||||||
|             # Configure services |                 # Configure services using the new method | ||||||
|             for service in services: |                 self.configure_services(services, imported_cert_name) | ||||||
|                 logger.info(f"Configuring {service}") |  | ||||||
|                 success, _, stderr = self.execute_command( |  | ||||||
|                     f'/ip service set {service} certificate="{imported_cert_name}"' |  | ||||||
|                 ) |  | ||||||
|                  |  | ||||||
|                 if not success: |  | ||||||
|                     logger.error(f"Failed to configure {service}: {stderr}") |  | ||||||
|                 else: |  | ||||||
|                     logger.info(f"✓ {service} configured") |  | ||||||
|              |  | ||||||
|             # Enable services |  | ||||||
|             for service in services: |  | ||||||
|                 logger.info(f"Enabling {service}") |  | ||||||
|                 self.execute_command(f'/ip service enable {service}') |  | ||||||
|                  |                  | ||||||
|                 time.sleep(1) |                 time.sleep(1) | ||||||
|                  |                  | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user
	 Mateusz Gruszczyński
					Mateusz Gruszczyński