Example #1
0
 public TaskData(CancellationTokenSource cancellationTokenSource, AddressRange addressRange)
 {
     CancellationTokenSource = cancellationTokenSource;
     Result = new Result(addressRange);
 }
Example #2
0
        void AddScanTask(AddressRange addressRange)
        {
            var cancellationTokenSource = new CancellationTokenSource();

            Task <List <ulong> > task = Task.Run(() =>
            {
                try
                {
                    return(MemoryHelper.FindPatternAddresses(Process, addressRange, Pattern, StopAfterFirst));
                }
                catch (TaskCanceledException)
                {
                }
                catch (Exception ex)
                {
                    Log.WriteException(ex);
                }

                return(null as List <ulong>);
            }, cancellationTokenSource.Token);

            task.ContinueWith((continuingTask) =>
            {
                try
                {
                    TaskData data = null;
                    lock (m_Tasks)
                    {
                        m_Tasks.TryGetValue(continuingTask, out data);
                    }

                    if (data == null)
                    {
                        return;
                    }

                    data.Result.Complete();

                    if (HasCompleted)
                    {
                        return;
                    }

                    if (continuingTask.Result != null && continuingTask.Result.Any())
                    {
                        Pattern.Addresses.AddRange(continuingTask.Result.Select(address => address));
                        data.Result.Matches.AddRange(continuingTask.Result);

                        // Cancel all other threads and complete the scan because we needed only 1 result
                        if (StopAfterFirst)
                        {
                            TryCancel();
                        }
                    }

                    // Complete the scan if this is the last thread to finish
                    if (!HasCompleted)
                    {
                        int completedTaskCount = 0;
                        lock (m_Tasks)
                        {
                            completedTaskCount = m_Tasks.Select(t => t.Key).Where(thisTask => thisTask.IsCompleted).Count();
                        }

                        if (completedTaskCount == m_Tasks.Count)
                        {
                            TryComplete();
                        }
                    }
                }
                catch (TaskCanceledException)
                {
                }
                catch (Exception ex)
                {
                    Log.WriteException(ex);
                }
            });

            lock (m_Tasks)
            {
                m_Tasks.TryAdd(task, new TaskData(cancellationTokenSource, addressRange));
            }
        }