예제 #1
0
파일: main.go 프로젝트: subutai-io/p2p
// StartProfiling will create a .prof file to analyze p2p app performance
func StartProfiling(profile string) {

	pwd, err := os.Getwd()
	if err != nil {
		ptp.Log(ptp.Error, "Getwd() error : %v", err)
		return
	}

	timeStr := "cpu"
	if profile == "cpu" {
		fileName := fmt.Sprintf("%s/%s.prof", pwd, timeStr)
		f, err := os.Create(fileName)
		if err != nil {
			ptp.Log(ptp.Error, "Create cpu_prof file failed. %v", err)
			return
		}
		ptp.Log(ptp.Info, "Start cpu profiling to file %s", fileName)
		pprof.StartCPUProfile(f)
	} else if profile == "memory" {
		_, err := os.Create(fmt.Sprintf("%s/%s.p2p_mem_prof", pwd, timeStr))
		if err != nil {
			ptp.Log(ptp.Error, "Create mem_prof file failed. %v", err)
			return
		}
	}
}
예제 #2
0
파일: main.go 프로젝트: subutai-io/p2p
// Daemon starts P2P daemon
func Daemon(port, sFile, profiling string) {
	StartProfiling(profiling)
	ptp.InitPlatform()
	instances = make(map[string]instance)
	ptp.InitErrors()

	if !ptp.CheckPermissions() {
		os.Exit(1)
	}

	proc := new(Procedures)
	rpc.Register(proc)
	rpc.HandleHTTP()
	listen, err := net.Listen("tcp", "localhost:"+port)
	if err != nil {
		ptp.Log(ptp.Error, "Cannot start RPC listener %v", err)
		os.Exit(1)
	}

	if sFile != "" {
		saveFile = sFile
		ptp.Log(ptp.Info, "Restore file provided")
		// Try to restore from provided file
		instances, err := loadInstances(saveFile)
		if err != nil {
			ptp.Log(ptp.Error, "Failed to load instances: %v", err)
		} else {
			ptp.Log(ptp.Info, "%d instances were loaded from file", len(instances))
			for _, inst := range instances {
				resp := new(Response)
				proc.Run(&inst, resp)
			}
		}
	}

	ptp.Log(ptp.Info, "Starting RPC Listener on %s port", port)
	go http.Serve(listen, nil)

	// Capture SIGINT
	// This is used for development purposes only, but later we should consider updating
	// this code to handle signals
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)

	go func() {
		for sig := range c {
			fmt.Println("Received signal: ", sig)
			pprof.StopCPUProfile()
			os.Exit(0)
		}
	}()
	select {}
}
예제 #3
0
파일: instance.go 프로젝트: subutai-io/p2p
// SetLog modifies specific option
func (p *Procedures) SetLog(args *NameValueArg, resp *Response) error {
	ptp.Log(ptp.Info, "Setting option %s to %s", args.Name, args.Value)
	resp.ExitCode = 0
	if args.Name == "log" {
		resp.Output = "Logging level has switched to " + args.Value + " level"
		if args.Value == "DEBUG" {
			ptp.SetMinLogLevel(ptp.Debug)
		} else if args.Value == "INFO" {
			ptp.SetMinLogLevel(ptp.Info)
		} else if args.Value == "TRACE" {
			ptp.SetMinLogLevel(ptp.Trace)
		} else if args.Value == "WARNING" {
			ptp.SetMinLogLevel(ptp.Warning)
		} else if args.Value == "ERROR" {
			ptp.SetMinLogLevel(ptp.Error)
		} else {
			resp.ExitCode = 1
			resp.Output = "Unknown log level was specified. Supported log levels is:\n"
			resp.Output = resp.Output + "TRACE\n"
			resp.Output = resp.Output + "DEBUG\n"
			resp.Output = resp.Output + "INFO\n"
			resp.Output = resp.Output + "WARNING\n"
			resp.Output = resp.Output + "ERROR\n"
		}
	}
	return nil
}
예제 #4
0
파일: main.go 프로젝트: subutai-io/p2p
// Dial connects to a local RPC server
func Dial(port string) *rpc.Client {
	client, err := rpc.DialHTTP("tcp", "localhost:"+port)
	if err != nil {
		ptp.Log(ptp.Error, "Failed to connect to RPC %v", err)
		os.Exit(1)
	}
	return client
}
예제 #5
0
파일: instance.go 프로젝트: subutai-io/p2p
func encodeInstances() ([]byte, error) {
	var savedInstances []RunArgs
	instances_mut.Lock()
	for _, inst := range instances {
		savedInstances = append(savedInstances, inst.Args)
	}
	instances_mut.Unlock()
	runtime.Gosched()
	b := bytes.Buffer{}
	e := gob.NewEncoder(&b)
	err := e.Encode(savedInstances)
	if err != nil {
		ptp.Log(ptp.Error, "Failed to encode instances: %v", err)
		return []byte(""), err
	}
	return b.Bytes(), nil
}
예제 #6
0
파일: instance.go 프로젝트: subutai-io/p2p
// Run starts a P2P instance
func (p *Procedures) Run(args *RunArgs, resp *Response) error {
	resp.ExitCode = 0
	resp.Output = "Running new P2P instance for " + args.Hash + "\n"

	// Validate if interface name is unique
	if args.Dev != "" {
		instances_mut.Lock()
		for _, inst := range instances {
			if inst.PTP.DeviceName == args.Dev {
				resp.ExitCode = 1
				resp.Output = "Device name is already in use"
				instances_mut.Unlock()
				runtime.Gosched()
				return errors.New(resp.Output)
			}
		}
		instances_mut.Unlock()
		runtime.Gosched()
	}

	var exists bool
	instances_mut.Lock()
	_, exists = instances[args.Hash]
	instances_mut.Unlock()
	runtime.Gosched()
	if !exists {
		resp.Output = resp.Output + "Lookup finished\n"
		if args.Key != "" {
			if len(args.Key) < 16 {
				args.Key += "0000000000000000"[:16-len(args.Key)]
			} else if len(args.Key) > 16 && len(args.Key) < 24 {
				args.Key += "000000000000000000000000"[:24-len(args.Key)]
			} else if len(args.Key) > 24 && len(args.Key) < 32 {
				args.Key += "00000000000000000000000000000000"[:32-len(args.Key)]
			} else if len(args.Key) > 32 {
				args.Key = args.Key[:32]
			}
		}

		var newInst instance
		newInst.ID = args.Hash
		newInst.Args = *args
		ptpInstance := ptp.StartP2PInstance(args.IP, args.Mac, args.Dev, "", args.Hash, args.Dht, args.Keyfile, args.Key, args.TTL, "", args.Fwd, args.Port)
		if ptpInstance == nil {
			resp.Output = resp.Output + "Failed to create P2P Instance"
			resp.ExitCode = 1
			return errors.New("Failed to create P2P Instance")
		}
		ptp.Log(ptp.Info, "Instance created")
		newInst.PTP = ptpInstance
		instances_mut.Lock()
		instances[args.Hash] = newInst
		instances_mut.Unlock()
		runtime.Gosched()
		go ptpInstance.Run()
		if saveFile != "" {
			resp.Output = resp.Output + "Saving instance into file"
			saveInstances(saveFile)
		}
	} else {
		resp.Output = resp.Output + "Hash already in use\n"
	}
	return nil
}