|
1 | 1 | package preflight
|
2 | 2 |
|
3 | 3 | import (
|
| 4 | + "errors" |
4 | 5 | "fmt"
|
| 6 | + "os/user" |
5 | 7 | "strconv"
|
6 | 8 | "strings"
|
7 | 9 |
|
@@ -209,3 +211,64 @@ func checkAdminHelperNamedPipeAccessible() error {
|
209 | 211 | }
|
210 | 212 | return nil
|
211 | 213 | }
|
| 214 | + |
| 215 | +func checkFileAndPrinterSharingIsEnabled() error { |
| 216 | + cmd := `(Get-NetFirewallRule -Group '@FirewallAPI.dll,-28502' | Where-Object {$_.Profile -eq 'Private, Public'}).Enabled` |
| 217 | + stdout, stderr, err := powershell.Execute(cmd) |
| 218 | + if err != nil { |
| 219 | + return fmt.Errorf("unable to check if Printer and File Sharing is enabled %v: %s", err, stderr) |
| 220 | + } |
| 221 | + if strings.Contains(stdout, "False") { |
| 222 | + return errors.New("Printer and File Sharing is disabled") |
| 223 | + } |
| 224 | + return nil |
| 225 | +} |
| 226 | + |
| 227 | +func fixFileAndPrinterSharing() error { |
| 228 | + cmd := `Set-NetFirewallRule -Group '@FirewallAPI.dll,-28502' -Enabled True -Profile 'Private,Public'` |
| 229 | + stdout, stderr, err := powershell.ExecuteAsAdmin("to enable Printer and File Sharing", cmd) |
| 230 | + if err != nil { |
| 231 | + return fmt.Errorf("unable to check if Printer and File Sharing is enabled %v: %s: %s", err, stdout, stderr) |
| 232 | + } |
| 233 | + return nil |
| 234 | +} |
| 235 | + |
| 236 | +func checkCRCSmbShareCreated() error { |
| 237 | + cmd := `Get-SmbShare -Name crc-dir0` |
| 238 | + stdout, stderr, err := powershell.Execute(cmd) |
| 239 | + if err != nil { |
| 240 | + return fmt.Errorf("unable to check if Printer and File Sharing is enabled %v: %s: %s", err, stdout, stderr) |
| 241 | + } |
| 242 | + return nil |
| 243 | +} |
| 244 | + |
| 245 | +func fixCRCSmbShareCreated() error { |
| 246 | + u, err := user.Current() |
| 247 | + if err != nil { |
| 248 | + return fmt.Errorf("unable to get user information for homedir and username: %v", err) |
| 249 | + } |
| 250 | + cmd := fmt.Sprintf(`New-SmbShare -Name 'crc-dir0' -Path '%s' -FullAccess '%s'`, u.HomeDir, username()) |
| 251 | + _, stderr, err := powershell.ExecuteAsAdmin("create new SMB share for home directory", cmd) |
| 252 | + if err != nil { |
| 253 | + return fmt.Errorf("unable to get create new SMB share %v: %s", err, stderr) |
| 254 | + } |
| 255 | + return nil |
| 256 | +} |
| 257 | + |
| 258 | +func removeSmbShare() error { |
| 259 | + cmd := `Remove-SmbShare -Name 'crc-dir0' -Force` |
| 260 | + _, stderr, err := powershell.ExecuteAsAdmin("remove SMB share for home directory", cmd) |
| 261 | + if err != nil { |
| 262 | + return fmt.Errorf("unable to get create new SMB share %v: %s", err, stderr) |
| 263 | + } |
| 264 | + return nil |
| 265 | +} |
| 266 | + |
| 267 | +func removeFirewallRuleAllowingPrinterAndFileSharing() error { |
| 268 | + cmd := `Set-NetFirewallRule -Group '@FirewallAPI.dll,-28502' -Enabled False -Profile 'Private,Public'` |
| 269 | + stdout, stderr, err := powershell.ExecuteAsAdmin("to disable Printer and File Sharing", cmd) |
| 270 | + if err != nil { |
| 271 | + logging.Warnf("unable to turn off Printer and File Sharing %v: %s: %s", err, stdout, stderr) |
| 272 | + } |
| 273 | + return nil |
| 274 | +} |
0 commit comments