Skip to content

Latest commit

 

History

History
342 lines (275 loc) · 15 KB

File metadata and controls

342 lines (275 loc) · 15 KB

Najbolje sigurnosne prakse

Sigurnost je ključna za MCP implementacije, posebno u poslovnim okruženjima. Važno je osigurati da su alati i podaci zaštićeni od neovlaštenog pristupa, curenja podataka i drugih sigurnosnih prijetnji.

Uvod

U ovoj lekciji istražit ćemo najbolje sigurnosne prakse za MCP implementacije. Pokrit ćemo autentikaciju i autorizaciju, zaštitu podataka, sigurno izvođenje alata te usklađenost s propisima o privatnosti podataka.

Ciljevi učenja

Na kraju ove lekcije moći ćete:

  • Implementirati sigurne mehanizme autentikacije i autorizacije za MCP servere.
  • Zaštititi osjetljive podatke korištenjem enkripcije i sigurnog pohranjivanja.
  • Osigurati sigurno izvođenje alata s odgovarajućom kontrolom pristupa.
  • Primijeniti najbolje prakse za zaštitu podataka i usklađenost s pravilima privatnosti.

Autentikacija i autorizacija

Autentikacija i autorizacija su ključni za sigurnost MCP servera. Autentikacija odgovara na pitanje "Tko ste vi?", dok autorizacija odgovara na "Što smijete raditi?".

Pogledajmo primjere kako implementirati sigurnu autentikaciju i autorizaciju na MCP serverima koristeći .NET i Java.

Integracija .NET Identity

ASP .NET Core Identity pruža snažan okvir za upravljanje autentikacijom i autorizacijom korisnika. Možemo ga integrirati s MCP serverima kako bismo osigurali pristup alatima i resursima.

Postoje ključni koncepti koje trebamo razumjeti pri integraciji ASP.NET Core Identity s MCP serverima, a to su:

  • Konfiguracija Identity: Postavljanje ASP.NET Core Identity s korisničkim ulogama i tvrdnjama (claims). Tvrdnja je informacija o korisniku, poput njegove uloge ili dozvola, na primjer "Admin" ili "User".
  • JWT autentikacija: Korištenje JSON Web Tokena (JWT) za siguran pristup API-ju. JWT je standard za siguran prijenos informacija između strana kao JSON objekt, koji se može verificirati i kojem se može vjerovati jer je digitalno potpisan.
  • Politike autorizacije: Definiranje politika za kontrolu pristupa određenim alatima na temelju korisničkih uloga. MCP koristi politike autorizacije za određivanje koji korisnici mogu pristupiti kojim alatima prema njihovim ulogama i tvrdnjama.
public class SecureMcpStartup
{
    public void ConfigureServices(IServiceCollection services)
    {
        // Add ASP.NET Core Identity
        services.AddIdentity<ApplicationUser, IdentityRole>()
            .AddEntityFrameworkStores<ApplicationDbContext>()
            .AddDefaultTokenProviders();
        
        // Configure JWT authentication
        services.AddAuthentication(options =>
        {
            options.DefaultAuthenticateScheme = JwtBearerDefaults.AuthenticationScheme;
            options.DefaultChallengeScheme = JwtBearerDefaults.AuthenticationScheme;
        })
        .AddJwtBearer(options =>
        {
            options.TokenValidationParameters = new TokenValidationParameters
            {
                ValidateIssuer = true,
                ValidateAudience = true,
                ValidateLifetime = true,
                ValidateIssuerSigningKey = true,
                ValidIssuer = Configuration["Jwt:Issuer"],
                ValidAudience = Configuration["Jwt:Audience"],
                IssuerSigningKey = new SymmetricSecurityKey(
                    Encoding.UTF8.GetBytes(Configuration["Jwt:Key"]))
            };
        });
        
        // Add authorization policies
        services.AddAuthorization(options =>
        {
            options.AddPolicy("CanUseAdminTools", policy =>
                policy.RequireRole("Admin"));
                
            options.AddPolicy("CanUseBasicTools", policy =>
                policy.RequireAuthenticatedUser());
        });
        
        // Configure MCP server with security
        services.AddMcpServer(options =>
        {
            options.ServerName = "Secure MCP Server";
            options.ServerVersion = "1.0.0";
            options.RequireAuthentication = true;
        });
        
        // Register tools with authorization requirements
        services.AddMcpTool<BasicTool>(options => 
            options.RequirePolicy("CanUseBasicTools"));
            
        services.AddMcpTool<AdminTool>(options => 
            options.RequirePolicy("CanUseAdminTools"));
    }
    
    public void Configure(IApplicationBuilder app)
    {
        // Use authentication and authorization
        app.UseAuthentication();
        app.UseAuthorization();
        
        // Use MCP server middleware
        app.UseMcpServer();
    }
}

U prethodnom kodu smo:

  • Konfigurirali ASP.NET Core Identity za upravljanje korisnicima.
  • Postavili JWT autentikaciju za siguran pristup API-ju. Naveli smo parametre za validaciju tokena, uključujući izdavača, publiku i ključ za potpisivanje.
  • Definirali politike autorizacije za kontrolu pristupa alatima na temelju korisničkih uloga. Na primjer, politika "CanUseAdminTools" zahtijeva da korisnik ima ulogu "Admin", dok politika "CanUseBasic" zahtijeva da korisnik bude autentificiran.
  • Registrirali MCP alate s posebnim zahtjevima za autorizaciju, osiguravajući da samo korisnici s odgovarajućim ulogama mogu pristupiti tim alatima.

Integracija Java Spring Security

Za Javu koristit ćemo Spring Security za implementaciju sigurne autentikacije i autorizacije MCP servera. Spring Security pruža sveobuhvatan sigurnosni okvir koji se besprijekorno integrira sa Spring aplikacijama.

Ključni koncepti ovdje su:

  • Konfiguracija Spring Security: Postavljanje sigurnosnih konfiguracija za autentikaciju i autorizaciju.
  • OAuth2 Resource Server: Korištenje OAuth2 za siguran pristup MCP alatima. OAuth2 je okvir za autorizaciju koji omogućava trećim stranama razmjenu pristupnih tokena za siguran pristup API-ju.
  • Sigurnosni presretači (Interceptors): Implementacija sigurnosnih presretača za provođenje kontrole pristupa pri izvođenju alata.
  • Kontrola pristupa temeljena na ulogama: Korištenje uloga za kontrolu pristupa određenim alatima i resursima.
  • Sigurnosne anotacije: Korištenje anotacija za osiguranje metoda i krajnjih točaka.
@Configuration
@EnableWebSecurity
public class SecurityConfig extends WebSecurityConfigurerAdapter {

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http
            .csrf().disable()
            .authorizeRequests()
                .antMatchers("/mcp/discovery").permitAll() // Allow tool discovery
                .antMatchers("/mcp/tools/**").hasAnyRole("USER", "ADMIN") // Require authentication for tools
                .antMatchers("/mcp/admin/**").hasRole("ADMIN") // Admin-only endpoints
                .anyRequest().authenticated()
            .and()
            .oauth2ResourceServer().jwt();
    }
    
    @Bean
    public McpSecurityInterceptor mcpSecurityInterceptor() {
        return new McpSecurityInterceptor();
    }
}

// MCP Security Interceptor for tool authorization
public class McpSecurityInterceptor implements ToolExecutionInterceptor {
    @Autowired
    private JwtDecoder jwtDecoder;
    
    @Override
    public void beforeToolExecution(ToolRequest request, Authentication authentication) {
        String toolName = request.getToolName();
        
        // Check if user has permissions for this tool
        if (toolName.startsWith("admin") && !authentication.getAuthorities().contains("ROLE_ADMIN")) {
            throw new AccessDeniedException("You don't have permission to use this tool");
        }
        
        // Additional security checks based on tool or parameters
        if ("sensitiveDataAccess".equals(toolName)) {
            validateDataAccessPermissions(request, authentication);
        }
    }
    
    private void validateDataAccessPermissions(ToolRequest request, Authentication auth) {
        // Implementation to check fine-grained data access permissions
    }
}

U prethodnom kodu smo:

  • Konfigurirali Spring Security za zaštitu MCP krajnjih točaka, dopuštajući javni pristup otkrivanju alata, dok se za izvođenje alata zahtijeva autentikacija.
  • Koristili OAuth2 kao resource server za upravljanje sigurnim pristupom MCP alatima.
  • Implementirali sigurnosni presretač za provođenje kontrole pristupa pri izvođenju alata, provjeravajući korisničke uloge i dozvole prije nego što se dozvoli pristup određenim alatima.
  • Definirali kontrolu pristupa temeljenu na ulogama za ograničavanje pristupa administratorskim alatima i pristupu osjetljivim podacima na temelju korisničkih uloga.

Zaštita podataka i privatnost

Zaštita podataka je ključna za osiguranje da se osjetljive informacije obrađuju sigurno. To uključuje zaštitu osobnih podataka (PII), financijskih podataka i drugih osjetljivih informacija od neovlaštenog pristupa i curenja.

Primjer zaštite podataka u Pythonu

Pogledajmo primjer kako implementirati zaštitu podataka u Pythonu koristeći enkripciju i detekciju PII.

from mcp_server import McpServer
from mcp_tools import Tool, ToolRequest, ToolResponse
from cryptography.fernet import Fernet
import os
import json
from functools import wraps

# PII Detector - identifies and protects sensitive information
class PiiDetector:
    def __init__(self):
        # Load patterns for different types of PII
        with open("pii_patterns.json", "r") as f:
            self.patterns = json.load(f)
    
    def scan_text(self, text):
        """Scans text for PII and returns detected PII types"""
        detected_pii = []
        # Implementation to detect PII using regex or ML models
        return detected_pii
    
    def scan_parameters(self, parameters):
        """Scans request parameters for PII"""
        detected_pii = []
        for key, value in parameters.items():
            if isinstance(value, str):
                pii_in_value = self.scan_text(value)
                if pii_in_value:
                    detected_pii.append((key, pii_in_value))
        return detected_pii

# Encryption Service for protecting sensitive data
class EncryptionService:
    def __init__(self, key_path=None):
        if key_path and os.path.exists(key_path):
            with open(key_path, "rb") as key_file:
                self.key = key_file.read()
        else:
            self.key = Fernet.generate_key()
            if key_path:
                with open(key_path, "wb") as key_file:
                    key_file.write(self.key)
        
        self.cipher = Fernet(self.key)
    
    def encrypt(self, data):
        """Encrypt data"""
        if isinstance(data, str):
            return self.cipher.encrypt(data.encode()).decode()
        else:
            return self.cipher.encrypt(json.dumps(data).encode()).decode()
    
    def decrypt(self, encrypted_data):
        """Decrypt data"""
        if encrypted_data is None:
            return None
        
        decrypted = self.cipher.decrypt(encrypted_data.encode())
        try:
            return json.loads(decrypted)
        except:
            return decrypted.decode()

# Security decorator for tools
def secure_tool(requires_encryption=False, log_access=True):
    def decorator(cls):
        original_execute = cls.execute_async if hasattr(cls, 'execute_async') else cls.execute
        
        @wraps(original_execute)
        async def secure_execute(self, request):
            # Check for PII in request
            pii_detector = PiiDetector()
            pii_found = pii_detector.scan_parameters(request.parameters)
            
            # Log access if required
            if log_access:
                tool_name = self.get_name()
                user_id = request.context.get("user_id", "anonymous")
                log_entry = {
                    "timestamp": datetime.now().isoformat(),
                    "tool": tool_name,
                    "user": user_id,
                    "contains_pii": bool(pii_found),
                    "parameters": {k: "***" for k in request.parameters.keys()}  # Don't log actual values
                }
                logging.info(f"Tool access: {json.dumps(log_entry)}")
            
            # Handle detected PII
            if pii_found:
                # Either encrypt sensitive data or reject the request
                if requires_encryption:
                    encryption_service = EncryptionService("keys/tool_key.key")
                    for param_name, pii_types in pii_found:
                        # Encrypt the sensitive parameter
                        request.parameters[param_name] = encryption_service.encrypt(
                            request.parameters[param_name]
                        )
                else:
                    # If encryption not available but PII found, you might reject the request
                    raise ToolExecutionException(
                        "Request contains sensitive data that cannot be processed securely"
                    )
            
            # Execute the original method
            return await original_execute(self, request)
        
        # Replace the execute method
        if hasattr(cls, 'execute_async'):
            cls.execute_async = secure_execute
        else:
            cls.execute = secure_execute
        return cls
    
    return decorator

# Example of a secure tool with the decorator
@secure_tool(requires_encryption=True, log_access=True)
class SecureCustomerDataTool(Tool):
    def get_name(self):
        return "customerData"
    
    def get_description(self):
        return "Accesses customer data securely"
    
    def get_schema(self):
        # Schema definition
        return {}
    
    async def execute_async(self, request):
        # Implementation would access customer data securely
        # Since we used the decorator, PII is already detected and encrypted
        return ToolResponse(result={"status": "success"})

U prethodnom kodu smo:

  • Implementirali PiiDetector class to scan text and parameters for personally identifiable information (PII).
  • Created an EncryptionService class to handle encryption and decryption of sensitive data using the cryptography library.
  • Defined a secure_tool decorator that wraps tool execution to check for PII, log access, and encrypt sensitive data if required.
  • Applied the secure_tool decorator to a sample tool (SecureCustomerDataTool kako bismo osigurali da rukuje osjetljivim podacima na siguran način.

Što slijedi

Odricanje od odgovornosti:
Ovaj dokument je preveden korištenjem AI usluge za prevođenje Co-op Translator. Iako nastojimo postići točnost, imajte na umu da automatski prijevodi mogu sadržavati pogreške ili netočnosti. Izvorni dokument na izvornom jeziku treba se smatrati autoritativnim izvorom. Za kritične informacije preporučuje se profesionalni ljudski prijevod. Ne snosimo odgovornost za bilo kakva nesporazuma ili pogrešna tumačenja koja proizlaze iz korištenja ovog prijevoda.